Update app.py

app.py

@@ -1,5 +1,4 @@
 import os
-import re
 import random
 import time
 import html
@@ -33,6 +32,17 @@ def get_logs():
     """StringIO에 쌓인 로그를 문자열로 반환"""
     return log_stream.getvalue()
 
+import re
+
+def get_deployment_update(code):
+    if not code or len(code.strip()) < 10:
+        return gr.update(visible=False)
+    clean_code = remove_code_block(code)
+    result = deploy_to_vercel(clean_code)
+    match = re.search(r'https?://[\\w\\-]+\\.vercel\\.app', result)
+    if match:
+        return gr.update(src=match.group(0), visible=True)
+    return gr.update(visible=False)
 
 # ------------------------
 # 1) DEMO_LIST 및 SystemPrompt
@@ -859,14 +869,15 @@ with gr.Blocks(css_paths=["app.css"], theme=theme) as demo:
                         )
                         gr.HTML('<div class="help-text">💡 원하는 게임의 설명을 입력하세요. 예: "테트리스 게임 제작해줘."</div>')
 
-
-
-                        deploy_result_container = gr.Markdown(
-                            value="아직 배포된 게임이 없습니다.",
-                            unsafe_allow_html=True,
+                        deploy_result_container = gr.IFrame(
+                            src="",
+                            visible=False,
+                            height=600,
+                            width="100%",
                             elem_id="deploy_result_container",
                         )
 
+
             
             # Code Drawer 열기/닫기
             codeBtn.click(lambda: gr.update(open=True), inputs=[], outputs=[code_drawer])
@@ -910,14 +921,13 @@ with gr.Blocks(css_paths=["app.css"], theme=theme) as demo:
             )
 
 
+
             deploy_btn.click(
-                # use handle_deploy_legacy to render the iframe/banner HTML
-                fn=lambda code: handle_deploy_legacy(code) if code else "<div style='color:red;'>배포할 코드가 없습니다.</div>",
+                fn=get_deployment_update,
                 inputs=[code_output],
                 outputs=[deploy_result_container]
             )
 
-
 # ------------------------
 # 9) 실행
 # ------------------------