Update app.py

app.py

@@ -822,7 +822,7 @@ def deploy_to_huggingface(code: str):
         
         # Import 문 추출 및 requirements.txt 생성을 위한 매핑
         import_mapping = {
-            'gradio': 'gradio>=4.0.0',
+            'gradio': 'gradio==5.6.0',
             'numpy': 'numpy',
             'pandas': 'pandas',
             'torch': 'torch',
@@ -833,23 +833,114 @@ def deploy_to_huggingface(code: str):
             'cv2': 'opencv-python',
             'sklearn': 'scikit-learn',
             'tensorflow': 'tensorflow',
-            'scipy': 'scipy'
+            'scipy': 'scipy',
+            'librosa': 'librosa',
+            'soundfile': 'soundfile',
+            'nltk': 'nltk',
+            'spacy': 'spacy',
+            'gensim': 'gensim',
+            'keras': 'keras',
+            'seaborn': 'seaborn',
+            'bokeh': 'bokeh',
+            'requests': 'requests',
+            'beautifulsoup4': 'beautifulsoup4',
+            'scikit-image': 'scikit-image',
+            'opencv-python': 'opencv-python',
+            'tqdm': 'tqdm',
+            'streamlit': 'streamlit',
+            'dash': 'dash',
+            'flask': 'flask',
+            'django': 'django',
+            'sqlalchemy': 'sqlalchemy',
+            'pymongo': 'pymongo',
+            'redis': 'redis',
+            'psycopg2': 'psycopg2-binary',
+            'mysql-connector': 'mysql-connector-python',
+            'pytest': 'pytest',
+            'unittest': 'unittest2',
+            'sphinx': 'sphinx',
+            'jupyter': 'jupyter',
+            'ipython': 'ipython',
+            'fastapi': 'fastapi',
+            'uvicorn': 'uvicorn',
+            'aiohttp': 'aiohttp',
+            'websockets': 'websockets',
+            'pyyaml': 'pyyaml',
+            'json': 'json5',
+            'xml': 'lxml',
+            'html': 'html5lib',
+            'urllib3': 'urllib3',
+            'cryptography': 'cryptography',
+            'bcrypt': 'bcrypt',
+            'jwt': 'pyjwt',
+            'pillow': 'Pillow',
+            'imageio': 'imageio',
+            'moviepy': 'moviepy',
+            'ffmpeg': 'ffmpeg-python',
+            'pydub': 'pydub',
+            'wave': 'wave',
+            'pyaudio': 'pyaudio',
+            'sounddevice': 'sounddevice',
+            'pygame': 'pygame',
+            'kivy': 'kivy',
+            'pyqt5': 'PyQt5',
+            'tkinter': 'tk',
+            'wx': 'wxPython',
+            'pyside2': 'PySide2',
+            'deap': 'deap',
+            'gym': 'gym',
+            'stable-baselines3': 'stable-baselines3',
+            'optuna': 'optuna',
+            'hyperopt': 'hyperopt',
+            'ray': 'ray',
+            'dask': 'dask',
+            'vaex': 'vaex',
+            'modin': 'modin',
+            'cupy': 'cupy',
+            'jax': 'jax',
+            'numba': 'numba',
+            'cython': 'cython',
+            'sympy': 'sympy',
+            'statsmodels': 'statsmodels',
+            'prophet': 'prophet',
+            'lightgbm': 'lightgbm',
+            'xgboost': 'xgboost',
+            'catboost': 'catboost',
+            'shap': 'shap',
+            'lime': 'lime',
+            'eli5': 'eli5',
+            'yellowbrick': 'yellowbrick',
+            'altair': 'altair',
+            'plotnine': 'plotnine',
+            'folium': 'folium',
+            'geopandas': 'geopandas',
+            'shapely': 'shapely',
+            'rasterio': 'rasterio',
+            'networkx': 'networkx',
+            'graphviz': 'graphviz',
+            'pydot': 'pydot',
+            'pygraphviz': 'pygraphviz'
         }
         
         required_packages = set(['gradio>=4.0.0'])  # gradio는 기본으로 포함
         
         # 코드에서 import 문 분석
+        import_pattern = r'^(?:from\s+(\S+)|import\s+([^,\s]+)(?:\s*,\s*([^,\s]+))*)'
+        
         for line in code.split('\n'):
+            line = line.strip()
             if line.startswith('import ') or line.startswith('from '):
-                # 'import' 또는 'from' 문에서 패키지 이름 추출
-                if line.startswith('import '):
-                    package = line.split('import ')[1].split()[0].split('.')[0]
-                else:
-                    package = line.split('from ')[1].split()[0].split('.')[0]
-                
-                if package in import_mapping:
-                    required_packages.add(import_mapping[package])
-                    
+                matches = re.match(import_pattern, line)
+                if matches:
+                    if matches.group(1):  # from ... import ...
+                        package = matches.group(1).split('.')[0]
+                    else:  # import ...
+                        packages = [p.strip() for p in re.findall(r'[\w.]+', line[7:])]
+                        for package in packages:
+                            package = package.split('.')[0]
+                            if package in import_mapping:
+                                required_packages.add(import_mapping[package])
+
         # 7) 전체 애플리케이션 코드 생성
         full_app_code = code