Update app.py

app.py

@@ -1,3 +1,5 @@
+
+
 import os
 import re
 import random
@@ -10,6 +12,8 @@ import asyncio
 import requests
 import anthropic
 import openai
+import io
+import logging
 
 from http import HTTPStatus
 from typing import Dict, List, Optional, Tuple
@@ -20,11 +24,23 @@ import modelscope_studio.components.base as ms
 import modelscope_studio.components.legacy as legacy
 import modelscope_studio.components.antd as antd
 
+# === [1] 로거 설정 ===
+log_stream = io.StringIO()
+handler = logging.StreamHandler(log_stream)
+logger = logging.getLogger()
+logger.setLevel(logging.DEBUG)   # 원하는 레벨로 설정
+logger.addHandler(handler)
+
+def get_logs():
+    """StringIO에 쌓인 로그를 문자열로 반환"""
+    return log_stream.getvalue()
+
 
 # ------------------------
 # 1) DEMO_LIST 및 SystemPrompt
 # ------------------------
 
+
 DEMO_LIST = [
     {"description": "블록이 위에서 떨어지는 클래식 테트리스 게임을 개발해주세요. 화살표 키로 조작하며, 가로줄이 채워지면 해당 줄이 제거되고 점수가 올라가는 메커니즘이 필요합니다. 난이도는 시간이 지날수록 블록이 빨라지도록 구현하고, 게임오버 조건과 점수 표시 기능을 포함해주세요."},
     {"description": "두 명이 번갈아가며 플레이할 수 있는 체스 게임을 만들어주세요. 기본적인 체스 규칙(킹, 퀸, 룩, 비숍, 나이트, 폰의 이동 규칙)을 구현하고, 체크와 체크메이트 감지 기능이 필요합니다. 드래그 앤 드롭으로 말을 움직일 수 있게 하며, 이동 기록도 표시해주세요."},
@@ -62,6 +78,7 @@ DEMO_LIST = [
     {"description": "탑다운 뷰의 간단한 액션 RPG 게임을 개발해주세요. WASD로 이동하고, 마우스 클릭으로 기본 공격, 1-4 키로 특수 스킬을 사용합니다. 플레이어는 몬스터를 처치하며 경험치와 아이템을 획득하고, 레벨업 시 능력치(공격력, 체력, 속도 등)를 향상시킵니다. 다양한 무기와 방어구를 착용할 수 있으며, 스킬 트리 시스템으로 캐릭터를 특화시킬 수 있습니다. 여러 지역과 보스 몬스터, 간단한 퀘스트 시스템도 구현해주세요."},
 ]
 
+
 SystemPrompt = """
 # GameCraft 시스템 프롬프트
 
@@ -203,11 +220,10 @@ openai_client = openai.OpenAI(api_key=YOUR_OPENAI_TOKEN)
 
 async def try_claude_api(system_message, claude_messages, timeout=15):
     """
-    Claude API 호출 (스트리밍) - 간결한 코드 생성 요청 추가
+    Claude API 호출 (스트리밍)
     """
     try:
-        # 시스템 프롬프트에 코드 길이 제한 강화
-        system_message_with_limit = system_message + "\n\n추가 중요 지침: 생성하는 코드는 절대로 200줄을 넘지 마세요. 코드 간결성이 최우선입니다. 주석을 최소화하고, 핵심 기능만 구현하세요. 불필요한 UI 요소나 장식은 제외하세요. 모든 코드는 하나의 HTML 파일에 포함되어야 합니다."
+        system_message_with_limit = system_message + "\n\n추가 중요 지침: 생성하는 코드는 절대로 200줄을 넘지 마세요. 코드 간결성이 최우선입니다. 주석을 최소화하고, 핵심 기능만 구현하세요."
         
         start_time = time.time()
         with claude_client.messages.stream(
@@ -215,7 +231,7 @@ async def try_claude_api(system_message, claude_messages, timeout=15):
             max_tokens=19800,
             system=system_message_with_limit,
             messages=claude_messages,
-            temperature=0.3,  # 온도 낮게 설정하여 일관된 결과 유도
+            temperature=0.3,
         ) as stream:
             collected_content = ""
             for chunk in stream:
@@ -235,16 +251,15 @@ async def try_openai_api(openai_messages):
     OpenAI API 호출 (스트리밍) - 코드 길이 제한 강화
     """
     try:
-        # 첫 번째 시스템 메시지에 코드 길이 제한 추가
         if openai_messages and openai_messages[0]["role"] == "system":
-            openai_messages[0]["content"] += "\n\n추가 중요 지침: 생성하는 코드는 절대로 200줄을 넘지 마세요. 코드 간결성이 최우선입니다. 주석을 최소화하고, 핵심 기능만 구현하세요. 불필요한 UI 요소나 장식은 제외하세요. 모든 코드는 하나의 HTML 파일에 포함되어야 합니다."
-        
+            openai_messages[0]["content"] += "\n\n추가 중요 지침: 생성하는 코드는 절대로 200줄을 넘지 마세요. 코드 간결성이 최우선입니다. 주석은 최소화하고, 핵심 기능만 구현하세요."
+
         stream = openai_client.chat.completions.create(
             model="o3",
             messages=openai_messages,
             stream=True,
             max_tokens=19800,
-            temperature=0.2  # 온도 낮게 설정
+            temperature=0.2
         )
         collected_content = ""
         for chunk in stream:
@@ -253,14 +268,13 @@ async def try_openai_api(openai_messages):
                 yield collected_content
     except Exception as e:
         raise e
-        
+
 
 # ------------------------
 # 4) 템플릿(하나로 통합)
 # ------------------------
 
 def load_json_data():
-    # 실제 DEMO_LIST의 모든 항목을 반환하도록 수정
     data_list = []
     for item in DEMO_LIST:
         data_list.append({
@@ -343,9 +357,6 @@ function copyToInput(card) {
     return gr.HTML(value=html_content)
 
 def load_all_templates():
-    """
-    모든 템플릿을 하나로 보여주는 함수
-    """
     return create_template_html("🎮 모든 게임 템플릿", load_json_data())
 
 
@@ -354,10 +365,6 @@ def load_all_templates():
 # ------------------------
 
 def remove_code_block(text):
-    """
-    More robust function to extract code from markdown code blocks
-    텍스트에서 ```html 및 ``` 태그를 완전히 제거하는 함수
-    """
     pattern = r'```html\s*([\s\S]+?)\s*```'
     match = re.search(pattern, text, re.DOTALL)
     if match:
@@ -373,10 +380,6 @@ def remove_code_block(text):
     return text.strip()
 
 def optimize_code(code: str) -> str:
-    """
-    AI가 생성한 코드를 최적화하여 크기를 줄이는 함수
-    불필요한 주석, 공백, 장황한 코드 등을 제거
-    """
     if not code or len(code.strip()) == 0:
         return code
     
@@ -410,9 +413,6 @@ def optimize_code(code: str) -> str:
     return cleaned_code
 
 def send_to_sandbox(code):
-    """
-    iframe으로 렌더링
-    """
     clean_code = remove_code_block(code)
     clean_code = optimize_code(clean_code)
     
@@ -457,7 +457,6 @@ def boost_prompt(prompt: str) -> str:
 - 최소한의 필수 게임 요소만 포함
 """
     try:
-        # Claude API
         try:
             response = claude_client.messages.create(
                 model="claude-3-7-sonnet-20250219",
@@ -645,20 +644,23 @@ class Demo:
     def clear_history(self):
         return []
 
+
 ####################################################
 # 1) deploy_to_vercel 함수
 ####################################################
 def deploy_to_vercel(code: str):
-    print("[DEBUG] deploy_to_vercel() 시작. code 길이:", len(code) if code else 0)
+    logger.debug(f"[deploy_to_vercel] code 길이: {len(code) if code else 0}")
     try:
         if not code or len(code.strip()) < 10:
+            logger.info("[deploy_to_vercel] code가 짧아서 배포 불가")
             return "No code to deploy. (Code is empty)"
 
         token = "A8IFZmgW2cqA4yUNlLPnci0N"
         if not token:
+            logger.error("[deploy_to_vercel] Vercel 토큰이 없습니다.")
             return "Vercel token is not set."
 
-        project_name = ''.join(random.choice(string.ascii_lowercase) for i in range(6))
+        project_name = ''.join(random.choice(string.ascii_lowercase) for _ in range(6))
         deploy_url = "https://api.vercel.com/v13/deployments"
         headers = {
             "Authorization": f"Bearer {token}",
@@ -693,18 +695,18 @@ def deploy_to_vercel(code: str):
             "projectSettings": project_settings
         }
 
-        print("[DEBUG] Vercel API 요청 전송중...")
+        logger.debug("[deploy_to_vercel] Vercel API 요청 전송중...")
         deploy_response = requests.post(deploy_url, headers=headers, json=deploy_data)
-        print("[DEBUG] 응답 status_code:", deploy_response.status_code)
+        logger.debug(f"[deploy_to_vercel] 응답 status_code: {deploy_response.status_code}")
 
         if deploy_response.status_code != 200:
+            logger.warning(f"[deploy_to_vercel] 배포 실패: {deploy_response.text}")
             return f"Deployment failed: {deploy_response.text}"
 
         deployment_url = f"https://{project_name}.vercel.app"
-        print("[DEBUG] 배포 성공, 배포 URL:", deployment_url)
+        logger.info(f"[deploy_to_vercel] 배포 성공, URL={deployment_url}")
         time.sleep(3)
 
-        # ★ 원본 코드 스타일: 심플 HTML (스크립트/iframe 없이)
         return f"""
         <div style="border:1px solid #34c759; padding:15px; border-radius:8px;">
             <h3 style="margin:0; color:#34c759;">✅ Deployment complete!</h3>
@@ -718,44 +720,38 @@ def deploy_to_vercel(code: str):
         """
 
     except Exception as e:
-        print("[ERROR] deploy_to_vercel() 예외:", e)
+        logger.exception("[deploy_to_vercel] 예외 발생")
         return f"Error during deployment: {str(e)}"
 
-####################################################
-# 2) handle_deploy_legacy: iframe으로 결과 표시
-####################################################
+
+# ------------------------
+# (3) handle_deploy_legacy
+# ------------------------
 def handle_deploy_legacy(code):
-    print("[DEBUG] handle_deploy_legacy() 호출됨. code 길이:", len(code) if code else 0)
+    logger.debug(f"[handle_deploy_legacy] code 길이: {len(code) if code else 0}")
     if not code or len(code.strip()) < 10:
-        print("[DEBUG] 유효하지 않은 코드 (None or 너무 짧음).")
+        logger.info("[handle_deploy_legacy] 코드가 짧음.")
         return "<div style='color:red;'>배포할 코드가 없습니다.</div>"
 
-    # remove_code_block 처리
     clean_code = remove_code_block(code)
-    print("[DEBUG] remove_code_block 후 clean_code 길이:", len(clean_code))
+    logger.debug(f"[handle_deploy_legacy] remove_code_block 후 길이: {len(clean_code)}")
 
-    # 실제 배포
     result_html = deploy_to_vercel(clean_code)
-    print("[DEBUG] 배포 결과 HTML 길이:", len(result_html))
+    logger.debug(f"[handle_deploy_legacy] 배포 결과 HTML 길이: {len(result_html)}")
 
-    # iframe에 표시하기 위해 Base64로 인코딩
     encoded_html = base64.b64encode(result_html.encode('utf-8')).decode()
     data_uri = f"data:text/html;charset=utf-8;base64,{encoded_html}"
 
-    # Gradio 3.29 이하 버전에서는 unsafe_allow_html 미지원이므로,
-    # iframe 태그만 반환하여 script 실행.
     iframe_html = f"""
     <iframe src="{data_uri}"
             style="width:100%; height:600px; border:none;"
             sandbox="allow-scripts allow-same-origin allow-popups">
     </iframe>
     """
-    print("[DEBUG] iframe_html 반환")
+    logger.debug("[handle_deploy_legacy] iframe_html 반환")
     return iframe_html
 
 
-
-
 # ------------------------
 # 8) Gradio / Modelscope UI 빌드
 # ------------------------
@@ -1023,10 +1019,6 @@ with gr.Blocks(css_paths=["app.css"], theme=theme) as demo:
                         gr.HTML('<div class="help-text">💡 원하는 게임의 설명을 입력하세요. 예: "테트리스 게임 제작해줘."</div>')
 
 
-##################################################################
-# 3) deploy_result_container = gr.HTML() (unsafe_allow_html = False)
-##################################################################
-# unsafe_allow_html 없이도 iframe 자체는 표시됨
                         deploy_result_container = gr.HTML(
                             """
                             <div class="deploy-section">
@@ -1039,6 +1031,7 @@ with gr.Blocks(css_paths=["app.css"], theme=theme) as demo:
                             label="Deployment Result"
                         )
 
+
             # 이벤트 / 콜백
             # Code Drawer
             codeBtn.click(
@@ -1107,7 +1100,7 @@ with gr.Blocks(css_paths=["app.css"], theme=theme) as demo:
                 outputs=[sandbox, state_tab]
             )
 
-            # ★ 수정된 부분: outputs=[deploy_result_container]
+            # 여기서 outputs=[deploy_result_container]가 핵심 수정
             deploy_btn.click(
                 fn=lambda code: deploy_to_vercel(remove_code_block(code)) if code else "No code generated.",
                 inputs=[code_output],