Update app.py

app.py

@@ -974,134 +974,8 @@ def deploy_to_huggingface(code: str):
         except Exception as e:
             raise e
 
-        # 4) 코드 정리 및 import 분석
+        # 4) 코드 정리
         code = code.replace("```python", "").replace("```", "").strip()
-        
-        # Import 문 추출 및 requirements.txt 생성을 위한 매핑
-        import_mapping = {
-            'numpy': 'numpy',
-            'pandas': 'pandas',
-            'torch': 'torch',
-            'matplotlib': 'matplotlib',
-            'plotly': 'plotly',
-            'transformers': 'transformers',
-            'PIL': 'Pillow',
-            'cv2': 'opencv-python',
-            'sklearn': 'scikit-learn',
-            'tensorflow': 'tensorflow',
-            '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',
-            'dlib': 'dlib',
-            'pygraphviz': 'pygraphviz'
-        }
-        
-        required_packages = set()
-
-        # 코드에서 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 '):
-                matches = re.match(import_pattern, line)
-                if matches:
-                    if matches.group(1):  # from ... import ...
-                        package = matches.group(1).split('.')[0]
-                        if package in import_mapping:
-                            required_packages.add(import_mapping[package])
-                    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])
-
-        # gradio는 항상 포함
-        required_packages.add('gradio==5.5.0')
 
         # 5) 전체 애플리케이션 코드 생성
         if "demo.launch()" not in code:
@@ -1120,8 +994,21 @@ def deploy_to_huggingface(code: str):
             repo_type="space"
         )
 
-        # requirements.txt 생성 및 업로드
-        requirements = '\n'.join(sorted(required_packages))
+        # 7) requirements.txt 생성 및 업로드
+        analysis_result = analyze_code(code)
+        requirements = ""
+
+        # HTML에서 requirements.txt 섹션 찾기
+        if "<h3>📋 Requirements.txt</h3>" in analysis_result:
+            start_idx = analysis_result.find("<pre>") + 5
+            end_idx = analysis_result.find("</pre>")
+            if start_idx > 4 and end_idx > 0:
+                requirements = analysis_result[start_idx:end_idx].strip()
+
+        # requirements가 비어있으면 기본값 설정
+        if not requirements:
+            requirements = 'gradio==5.5.0'
+
         with open("requirements.txt", "w") as f:
             f.write(requirements)
             
@@ -1132,7 +1019,7 @@ def deploy_to_huggingface(code: str):
             repo_type="space"
         )
 
-        # 7) 결과 반환
+        # 8) 결과 반환
         space_url = f"https://huggingface.co/spaces/{username}/{space_name}"
         return f'배포 완료! Private Space로 생성되었습니다. <a href="{space_url}" target="_blank" style="color: #1890ff; text-decoration: underline; cursor: pointer;">여기를 클릭하여 Space 열기</a>'