Update app.py

app.py

@@ -1015,13 +1015,13 @@ def deploy_to_huggingface(code: str):
         token = os.getenv("HF_TOKEN")
         if not token:
             return "HuggingFace 토큰이 설정되지 않았습니다."
-
+        
         # 2) Space 생성 준비
         api = HfApi(token=token)
         space_name = generate_space_name()
         username = api.whoami()['name']
         repo_id = f"{username}/{space_name}"
-
+        
         # 3) Space 생성 (private로 설정)
         try:
             create_repo(
@@ -1033,16 +1033,16 @@ def deploy_to_huggingface(code: str):
             )
         except Exception as e:
             raise e
-
+        
         # 4) 코드 정리
         code = code.replace("```python", "").replace("```", "").strip()
-
+        
         # 5) 전체 애플리케이션 코드 생성
         if "demo.launch()" not in code:
             full_app_code = code + "\n\nif __name__ == '__main__':\n    demo.launch()"
         else:
             full_app_code = code
-
+        
         # 6) 파일 생성 및 업로드
         with open("app.py", "w", encoding="utf-8") as f:
             f.write(full_app_code)
@@ -1053,24 +1053,57 @@ def deploy_to_huggingface(code: str):
             repo_id=repo_id,
             repo_type="space"
         )
-
+        
         # 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'
-
+        
+        # 기본 requirements 설정
+        default_requirements = [
+            'gradio==5.5.0',
+            'huggingface_hub',
+            'transformers',
+            'torch',
+            'torchvision',
+            'diffusers',
+            'accelerate',
+            'safetensors',
+            'scipy',
+            'pillow',
+            'numpy'
+        ]
+        
+        # 기존 requirements가 있으면 파싱하여 중복 제거
+        existing_requirements = set()
+        if requirements:
+            existing_requirements = {line.split('==')[0] if '==' in line else line 
+                                  for line in requirements.split('\n') if line.strip()}
+        
+        # 최종 requirements 생성
+        final_requirements = set()
+        
+        # 기존 requirements 추가 (버전 정보 유지)
+        if requirements:
+            for req in requirements.split('\n'):
+                if req.strip():
+                    pkg_name = req.split('==')[0] if '==' in req else req
+                    final_requirements.add(req.strip())
+        
+        # 기본 requirements 추가 (없는 것만)
+        for req in default_requirements:
+            pkg_name = req.split('==')[0] if '==' in req else req
+            if pkg_name not in {r.split('==')[0] if '==' in r else r for r in final_requirements}:
+                final_requirements.add(req)
+        
+        # requirements.txt 작성
         with open("requirements.txt", "w") as f:
-            f.write(requirements)
+            f.write('\n'.join(sorted(final_requirements)))
             
         api.upload_file(
             path_or_fileobj="requirements.txt",
@@ -1078,15 +1111,12 @@ def deploy_to_huggingface(code: str):
             repo_id=repo_id,
             repo_type="space"
         )
-
+        
         # 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>'
-
     except Exception as e:
         return f"배포 중 오류 발생: {str(e)}"
-
-
         
 # Demo 인스턴스 생성
 demo_instance = Demo()