Update app.py

app.py

@@ -793,35 +793,13 @@ def deploy_to_huggingface(code: str):
         if not token:
             return "HuggingFace 토큰이 설정되지 않았습니다."
 
-        # 2) Rate Limit 관리
-        rate_limit_file = "rate_limit.json"
-        current_time = time.time()
-
-        try:
-            with open(rate_limit_file, 'r') as f:
-                rate_limit_data = json.load(f)
-                last_deploy_time = rate_limit_data.get('last_deploy_time', 0)
-                deploy_count = rate_limit_data.get('deploy_count', 0)
-        except FileNotFoundError:
-            last_deploy_time = 0
-            deploy_count = 0
-
-        # 3) Rate Limit 체크
-        if current_time - last_deploy_time < 54000 and deploy_count >= 5:
-            remaining_time = int((54000 - (current_time - last_deploy_time)) / 3600)
-            return f"배포 횟수 제한에 도달했습니다. {remaining_time}시간 후에 다시 시도해주세요."
-
-        if current_time - last_deploy_time >= 54000:
-            deploy_count = 0
-            last_deploy_time = current_time
-
-        # 4) Space 생성 준비
+        # 2) Space 생성 준비
         api = HfApi(token=token)
         space_name = generate_space_name()
         username = api.whoami()['name']
         repo_id = f"{username}/{space_name}"
 
-        # 5) Space 생성 (private로 설정)
+        # 3) Space 생성 (private로 설정)
         try:
             create_repo(
                 repo_id,
@@ -831,11 +809,9 @@ def deploy_to_huggingface(code: str):
                 private=True
             )
         except Exception as e:
-            if "Too Many Requests" in str(e):
-                return "현재 HuggingFace API 요청이 제한되었습니다. 잠시 후 다시 시도해주세요."
             raise e
 
-        # 6) 코드 정리 및 import 분석
+        # 4) 코드 정리 및 import 분석
         code = code.replace("```python", "").replace("```", "").strip()
         
         # Import 문 추출 및 requirements.txt 생성을 위한 매핑
@@ -939,7 +915,7 @@ def deploy_to_huggingface(code: str):
             'pygraphviz': 'pygraphviz'
         }
         
-        required_packages = set(['gradio==5.5.0'])  # gradio는 기본으로 포함
+        required_packages = set()
 
         # 코드에서 import 문 분석
         import_pattern = r'^(?:from\s+(\S+)|import\s+([^,\s]+)(?:\s*,\s*([^,\s]+))*)'
@@ -949,23 +925,27 @@ def deploy_to_huggingface(code: str):
             if line.startswith('import ') or line.startswith('from '):
                 matches = re.match(import_pattern, line)
                 if matches:
-                    if matches.group(1):
+                    if matches.group(1):  # from ... import ...
                         package = matches.group(1).split('.')[0]
-                    else:
+                        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])
 
-        # 7) 전체 애플리케이션 코드 생성
-        # demo.launch() 호출이 있는지 확인하고 없으면 추가
+        # gradio는 항상 포함
+        required_packages.add('gradio==5.5.0')
+
+        # 5) 전체 애플리케이션 코드 생성
         if "demo.launch()" not in code:
             full_app_code = code + "\n\nif __name__ == '__main__':\n    demo.launch()"
         else:
             full_app_code = code
 
-        # 8) 파일 생성 및 업로드
+        # 6) 파일 생성 및 업로드
         with open("app.py", "w", encoding="utf-8") as f:
             f.write(full_app_code)
         
@@ -978,7 +958,6 @@ def deploy_to_huggingface(code: str):
 
         # requirements.txt 생성 및 업로드
         requirements = '\n'.join(sorted(required_packages))
-
         with open("requirements.txt", "w") as f:
             f.write(requirements)
             
@@ -989,25 +968,12 @@ def deploy_to_huggingface(code: str):
             repo_type="space"
         )
 
-
-
-        # 9) Rate Limit 정보 업데이트
-        deploy_count += 1
-        with open(rate_limit_file, 'w') as f:
-            json.dump({
-                'last_deploy_time': current_time,
-                'deploy_count': deploy_count
-            }, f)
-
-        # 10) 결과 반환
+        # 7) 결과 반환
         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:
-        error_msg = str(e)
-        if "Too Many Requests" in error_msg:
-            return "HuggingFace API 요청 한도를 초과했습니다. 15시간 후에 다시 시도해주세요."
-        return f"배포 중 오류 발생: {error_msg}"
+        return f"배포 중 오류 발생: {str(e)}"
         
 # Demo 인스턴스 생성
 demo_instance = Demo()