Import library automatically

Question

Is there any extension of Visual Studio Code that when using a library in the code, it is automatically imported?

Ex:

from PySimpleGUI import PySimpleGUI as sg
layout = [
    [sg.Text('Teste')]
]

I would like to enter only the layout part and import automatically without having to give the import