Update app.py

app.py

@@ -547,6 +547,40 @@ def execute_code(query: str):
 # ------------------------
 # 6) 데모 클래스
 # ------------------------
+# ── deploy 버튼용 함수 새로 작성 ─────────────────────────────
+def deploy_and_show(code_md: str):
+    # 1) 코드 정리 & 배포
+    clean = remove_code_block(code_md)
+    result = deploy_to_vercel(clean)
+
+    url_match = re.search(r"https?://[\\w-]+\\.vercel\\.app", result)
+    if url_match:                         # ── 성공 ──
+        url = url_match.group(0)
+
+        # ① 오른쪽 메시지(마크다운)
+        md = f"✅ **배포 완료!**\n\n➡️ [열기]({url})"
+
+        # ② 왼쪽 iframe
+        iframe = (
+            f"<iframe src='{url}' width='100%' height='920px' "
+            "style='border:none;'></iframe>"
+        )
+
+        # state_tab : 'render' 로 전환
+        return (
+            gr.update(value=md, visible=True),  # deploy_result_container
+            iframe,                             # sandbox
+            gr.update(active_key='render')      # state_tab
+        )
+    # ── 실패 ──
+    md = f"❌ **배포 실패**\n\n```\n{result}\n```"
+    return (
+        gr.update(value=md, visible=True),
+        None,
+        gr.update(active_key='empty')
+    )
+
+
 
 class Demo:
     def __init__(self):
@@ -880,11 +914,8 @@ with gr.Blocks(css_paths=["app.css"], theme=theme) as demo:
                         )
                         gr.HTML('<div class="help-text">💡 원하는 게임의 설명을 입력하세요. 예: "테트리스 게임 제작해줘."</div>')
 
-   
-                        deploy_result_container = gr.Markdown(
-                            value="",                # 첫 화면은 비워 두기
-                            visible=False                 
-                        )
+                        deploy_result_container = gr.Markdown(value="", visible=False)
+
 
             
             # Code Drawer 열기/닫기
@@ -934,7 +965,11 @@ with gr.Blocks(css_paths=["app.css"], theme=theme) as demo:
                 outputs=[deploy_result_container]
             )
 
-
+            deploy_btn.click(
+                fn=deploy_and_show,
+                inputs=[code_output],
+                outputs=[deploy_result_container, sandbox, state_tab]
+            )
 
 # ------------------------
 # 9) 실행