Update app.py

app.py

@@ -173,6 +173,130 @@ async def try_openai_api(openai_messages):
         raise e
 
 
+# Import to Package Name Mapping Dictionary
+IMPORT_TO_PACKAGE = {
+    'PIL': 'pillow',
+    'cv2': 'opencv-python',
+    'sklearn': 'scikit-learn',
+    'bs4': 'beautifulsoup4',
+    'yaml': 'pyyaml',
+    'tensorflow': 'tensorflow-cpu',
+    'tf': 'tensorflow-cpu',
+    'wx': 'wxPython',
+    'cairo': 'pycairo',
+    'MySQLdb': 'mysqlclient',
+    'psycopg2': 'psycopg2-binary',
+    'magic': 'python-magic',
+    'Image': 'pillow',
+    'skimage': 'scikit-image',
+    'matplotlib.pyplot': 'matplotlib',
+    'plt': 'matplotlib',
+    'np': 'numpy',
+    'pd': 'pandas',
+    'torch': 'pytorch',
+    'transformers': 'transformers',
+    'librosa': 'librosa',
+    'nltk': 'nltk',
+    'spacy': 'spacy',
+    'requests': 'requests',
+    'flask': 'flask',
+    'fastapi': 'fastapi',
+    'aiohttp': 'aiohttp',
+    'imageio': 'imageio',
+    'moviepy': 'moviepy',
+    'networkx': 'networkx',
+    'statsmodels': 'statsmodels',
+    'seaborn': 'seaborn',
+    'bokeh': 'bokeh',
+    'plotly': 'plotly',
+    'dash': 'dash',
+    'streamlit': 'streamlit',
+    'altair': 'altair',
+    'geopandas': 'geopandas',
+    'folium': 'folium',
+    'shapely': 'shapely',
+    'fiona': 'fiona',
+    'rasterio': 'rasterio',
+    'gdal': 'gdal',
+    'pyproj': 'pyproj',
+    'rtree': 'rtree',
+    'cartopy': 'cartopy',
+    'geoplot': 'geoplot',
+    'descartes': 'descartes',
+    'pysal': 'pysal',
+    'geopy': 'geopy',
+    'osmnx': 'osmnx',
+    'contextily': 'contextily',
+    'xarray': 'xarray',
+    'netCDF4': 'netCDF4',
+    'h5py': 'h5py',
+    'tables': 'pytables',
+    'zarr': 'zarr',
+    'dask': 'dask',
+    'numba': 'numba',
+    'sympy': 'sympy',
+    'scipy': 'scipy',
+    'scikit-image': 'scikit-image',
+    'scikit-learn': 'scikit-learn',
+    'keras': 'keras',
+    'theano': 'theano',
+    'caffe': 'caffe',
+    'mxnet': 'mxnet',
+    'chainer': 'chainer',
+    'pytorch': 'torch',
+    'tensorflow-gpu': 'tensorflow-gpu',
+    'cupy': 'cupy',
+    'pycuda': 'pycuda',
+    'pyopencl': 'pyopencl',
+    'pyvista': 'pyvista',
+    'mayavi': 'mayavi',
+    'vtk': 'vtk',
+    'trimesh': 'trimesh',
+    'open3d': 'open3d-python',
+    'pyqt5': 'PyQt5',
+    'pyside2': 'PySide2',
+    'tkinter': 'tk',
+    'kivy': 'kivy',
+    'pygame': 'pygame',
+    'arcade': 'arcade',
+    'pyglet': 'pyglet',
+    'panda3d': 'panda3d',
+    'ursina': 'ursina',
+    'moderngl': 'moderngl',
+    'glfw': 'glfw',
+    'pyopengl': 'PyOpenGL',
+    'pysdl2': 'PySDL2',
+    'pybullet': 'pybullet',
+    'box2d': 'box2d-py',
+    'pymunk': 'pymunk',
+    'pyode': 'pyode',
+    'pyrr': 'pyrr',
+    'noise': 'noise',
+    'wave': 'wave',
+    'sounddevice': 'sounddevice',
+    'pyaudio': 'PyAudio',
+    'simpleaudio': 'simpleaudio',
+    'pygame.mixer': 'pygame',
+    'pydub': 'pydub',
+    'aubio': 'aubio',
+    'music21': 'music21',
+    'pretty_midi': 'pretty_midi',
+    'mido': 'mido',
+    'fluidsynth': 'fluidsynth',
+    'mingus': 'mingus',
+    'pyfluidsynth': 'pyfluidsynth',
+    'python-rtmidi': 'python-rtmidi',
+    'pygame.midi': 'pygame',
+    'soundfile': 'soundfile',
+    'resampy': 'resampy'
+}
+
+def get_package_name(import_name):
+    """임포트명으로부터 실제 패키지명을 반환"""
+    # 점이 있는 경우 첫 부분만 사용 (예: matplotlib.pyplot -> matplotlib)
+    base_import = import_name.split('.')[0]
+    return IMPORT_TO_PACKAGE.get(base_import, base_import)
+
 def analyze_code(code: str) -> str:
     """코드 분석 결과를 HTML 형식으로 반환"""
     analysis = []
@@ -195,13 +319,14 @@ def analyze_code(code: str) -> str:
     for line in code.split('\n'):
         if line.startswith('import ') or line.startswith('from '):
             imports.append(line.strip())
-            # 패키지 이름 추출
+            # 패키지 이름 추출 및 변환
             if line.startswith('import '):
                 package = line.split('import ')[1].split()[0].split('.')[0]
             else:
                 package = line.split('from ')[1].split()[0].split('.')[0]
-            required_packages.add(package)
+            required_packages.add(get_package_name(package))
     
+   
     if imports:
         analysis.append("<h2>📚 필요한 라이브러리</h2>")
         analysis.append("<ul>")