Update app.py

app.py

@@ -17,6 +17,10 @@ import modelscope_studio.components.antd as antd
 
 import html
 import urllib.parse
+from huggingface_hub import HfApi, create_repo
+import string
+import random
+
 
 # SystemPrompt 부분을 직접 정의
 SystemPrompt = """너의 이름은 'MOUSE'이다. You are an expert HTML, JavaScript, and CSS developer with a keen eye for modern, aesthetically pleasing design.
@@ -708,6 +712,52 @@ def load_session_history(template_type="best"):
         return gr.HTML("Error loading templates")
 
 
+
+
+
+# 배포 관련 함수 추가
+def generate_space_name():
+    """6자리 랜덤 영문 이름 생성"""
+    letters = string.ascii_lowercase
+    return ''.join(random.choice(letters) for i in range(6))
+
+def deploy_to_huggingface(code: str):
+    """HuggingFace Spaces에 코드 배포"""
+    try:
+        token = os.getenv("HF_TOKEN")
+        if not token:
+            return "HuggingFace 토큰이 설정되지 않았습니다."
+
+        api = HfApi(token=token)
+        space_name = generate_space_name()
+        
+        # Space 생성
+        repo_id = f"{api.whoami()['name']}/{space_name}"
+        create_repo(
+            repo_id,
+            repo_type="space",
+            space_sdk="static",
+            token=token
+        )
+
+        # index.html 생성 및 업로드
+        with open("index.html", "w", encoding="utf-8") as f:
+            f.write(code)
+        
+        api.upload_file(
+            path_or_fileobj="index.html",
+            path_in_repo="index.html",
+            repo_id=repo_id,
+            repo_type="space"
+        )
+
+        # 배포된 URL 반환
+        space_url = f"https://huggingface.co/spaces/{repo_id}"
+        return f"배포 완료! 다음 URL에서 확인하세요: {space_url}"
+
+    except Exception as e:
+        return f"배포 중 오류 발생: {str(e)}"
+        
 # Demo 인스턴스 생성
 demo_instance = Demo()
 
@@ -765,13 +815,17 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
                             placeholder=random.choice(DEMO_LIST)['description']
                         )
 
-        # 버튼들을 가로로 배치하기 위한 Flex 컨테이너 (한 번만 출력)
+
+# UI 수정 부분 - antd.Col(span=24, md=8) 내부의 버튼 컨테이너에 배포 버튼 추가:
                         with antd.Flex(gap="small", justify="space-between"):
                             btn = antd.Button("Send", type="primary", size="large")
                             execute_btn = antd.Button("Code 실행", type="default", size="large")
+                            deploy_btn = antd.Button("배포하기", type="default", size="large")  # 추가
                             clear_btn = antd.Button("Clear", type="default", size="large")
 
-                
+# 배포 결과를 표시할 텍스트 영역 추가
+                        deploy_result = gr.Textbox(label="배포 결과", interactive=False)
+ 
 
                 with antd.Col(span=24, md=16):
                     with ms.Div(elem_classes="right_panel"):
@@ -885,6 +939,12 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
                 outputs=[history]
             )
 
+# 이벤트 핸들러 추가
+            deploy_btn.click(
+                fn=lambda code: deploy_to_huggingface(remove_code_block(code)) if code else "코드가 없습니다.",
+                inputs=[code_output],
+                outputs=[deploy_result]
+            )
 
 
 if __name__ == "__main__":