Update app.py

app.py

@@ -940,7 +940,7 @@ def deploy_to_huggingface(code: str):
         }
         
         required_packages = set(['gradio==5.5.0'])  # gradio는 기본으로 포함
-        
+
         # 코드에서 import 문 분석
         import_pattern = r'^(?:from\s+(\S+)|import\s+([^,\s]+)(?:\s*,\s*([^,\s]+))*)'
         
@@ -949,9 +949,9 @@ 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):  # from ... import ...
+                    if matches.group(1):
                         package = matches.group(1).split('.')[0]
-                    else:  # import ...
+                    else:
                         packages = [p.strip() for p in re.findall(r'[\w.]+', line[7:])]
                         for package in packages:
                             package = package.split('.')[0]
@@ -959,8 +959,12 @@ def deploy_to_huggingface(code: str):
                                 required_packages.add(import_mapping[package])
 
         # 7) 전체 애플리케이션 코드 생성
-        full_app_code = code
-        
+        # demo.launch() 호출이 있는지 확인하고 없으면 추가
+        if "demo.launch()" not in code:
+            full_app_code = code + "\n\nif __name__ == '__main__':\n    demo.launch()"
+        else:
+            full_app_code = code
+
         # 8) 파일 생성 및 업로드
         with open("app.py", "w", encoding="utf-8") as f:
             f.write(full_app_code)
@@ -985,6 +989,8 @@ def deploy_to_huggingface(code: str):
             repo_type="space"
         )
 
+
+
         # 9) Rate Limit 정보 업데이트
         deploy_count += 1
         with open(rate_limit_file, 'w') as f: