Update app.py

app.py

@@ -835,21 +835,41 @@ def deploy_to_huggingface(code: str):
                 return "현재 HuggingFace API 요청이 제한되었습니다. 잠시 후 다시 시도해주세요."
             raise e
 
-        # 6) 코드 정리 - 중복 제거
+        # 6) 코드 정리 및 import 분석
         code = code.replace("```python", "").replace("```", "").strip()
         
-        # 필수 imports만 포함
-        imports = "import gradio as gr\nimport numpy as np\n"
+        # Import 문 추출 및 requirements.txt 생성을 위한 매핑
+        import_mapping = {
+            'gradio': 'gradio>=4.0.0',
+            'numpy': 'numpy',
+            'pandas': 'pandas',
+            'torch': 'torch',
+            'matplotlib': 'matplotlib',
+            'plotly': 'plotly',
+            'transformers': 'transformers',
+            'PIL': 'Pillow',
+            'cv2': 'opencv-python',
+            'sklearn': 'scikit-learn',
+            'tensorflow': 'tensorflow',
+            'scipy': 'scipy'
+        }
+        
+        required_packages = set(['gradio>=4.0.0'])  # gradio는 기본으로 포함
         
-        # 메인 코드에서 중복 imports 제거
-        code_lines = code.split('\n')
-        main_code = []
-        for line in code_lines:
-            if not line.startswith('import ') and not line.startswith('from '):
-                main_code.append(line)
+        # 코드에서 import 문 분석
+        for line in code.split('\n'):
+            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])
+                    
         # 7) 전체 애플리케이션 코드 생성
-        full_app_code = imports + "\n".join(main_code)
+        full_app_code = code
         
         # 8) 파일 생성 및 업로드
         with open("app.py", "w", encoding="utf-8") as f:
@@ -863,7 +883,7 @@ def deploy_to_huggingface(code: str):
         )
 
         # requirements.txt 생성 및 업로드
-        requirements = "gradio>=4.0.0\nnumpy"
+        requirements = '\n'.join(sorted(required_packages))
 
         with open("requirements.txt", "w") as f:
             f.write(requirements)