Spaces:

openfree
/

Vibe-Game

Paused

App Files Files Community

openfree commited on Apr 24

Commit

89d6b7d

verified ·

1 Parent(s): 3fc84bc

Update app.py

Browse files

Files changed (1) hide show

app.py +68 -21

app.py CHANGED Viewed

@@ -525,22 +525,48 @@ def deploy_to_vercel(code: str):
 def remove_code_block(text):
     """
-    메시지 내의 ```html ... ``` 부분만 추출하여 반환
     """
-    pattern = r'```html\n(.+?)\n```'
     match = re.search(pattern, text, re.DOTALL)
     if match:
         return match.group(1).strip()
-    else:
-        return text.strip()
 def send_to_sandbox(code):
     """
-    HTML 코드를 iframe으로 렌더링하기 위한 data URI 생성
     """
-    encoded_html = base64.b64encode(code.encode('utf-8')).decode('utf-8')
     data_uri = f"data:text/html;charset=utf-8;base64,{encoded_html}"
-    return f"<iframe src=\"{data_uri}\" width=\"100%\" height=\"920px\" style=\"border:none;\"></iframe>"
 def boost_prompt(prompt: str) -> str:
     """
@@ -605,6 +631,36 @@ def history_render(history: History):
     """
     return gr.update(open=True), history
 # ------------------------
 # 6) 데모 클래스
@@ -685,10 +741,11 @@ class Demo:
                 }])
                 # 최종 결과(코드) + 샌드박스 미리보기
                 yield [
                     collected_content,
                     _history,
-                    send_to_sandbox(remove_code_block(collected_content)),
                     gr.update(active_key="render"),
                     gr.update(open=True)
                 ]
@@ -912,7 +969,9 @@ with gr.Blocks(css_paths=["app.css"], theme=theme) as demo:
                         input_text = antd.InputTextarea(
                             size="large",
                             allow_clear=True,
-                            placeholder=random.choice(DEMO_LIST)['description']
                         )
                         gr.HTML('<div class="help-text">💡 원하는 게임의 설명을 입력하세요. 예: "Create a platformer game with jumping mechanics"</div>')
@@ -991,18 +1050,6 @@ with gr.Blocks(css_paths=["app.css"], theme=theme) as demo:
             )
             # (G) 'Code실행' 버튼 => 미리보기 iframe 로드
-            def execute_code(query: str):
-                if not query or query.strip() == '':
-                    return None, gr.update(active_key="empty")
-                try:
-                    if '```html' in query and '```' in query:
-                        code = remove_code_block(query)
-                    else:
-                        code = query.strip()
-                    return send_to_sandbox(code), gr.update(active_key="render")
-                except Exception:
-                    return None, gr.update(active_key="empty")
             execute_btn.click(
                 fn=execute_code,
                 inputs=[input_text],

 def remove_code_block(text):
     """
+    More robust function to extract code from markdown code blocks
     """
+    # First try to find ```html blocks
+    pattern = r'```html\s*([\s\S]+?)\s*```'
     match = re.search(pattern, text, re.DOTALL)
     if match:
         return match.group(1).strip()
+    # If no html block found, try any code block
+    pattern = r'```(?:\w+)?\s*([\s\S]+?)\s*```'
+    match = re.search(pattern, text, re.DOTALL)
+    if match:
+        return match.group(1).strip()
+    # If no code block, return the original text (might be raw code)
+    return text.strip()
 def send_to_sandbox(code):
     """
+    Improved function to create iframe with proper code cleaning
     """
+    # Make sure we have clean code without markdown delimiters
+    clean_code = remove_code_block(code)
+    # Ensure the code has basic HTML structure if it's missing
+    if not clean_code.strip().startswith('<!DOCTYPE') and not clean_code.strip().startswith('<html'):
+        clean_code = f"""<!DOCTYPE html>
+<html>
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Game Preview</title>
+</head>
+<body>
+{clean_code}
+</body>
+</html>"""
+    # Create data URI for iframe
+    encoded_html = base64.b64encode(clean_code.encode('utf-8')).decode('utf-8')
     data_uri = f"data:text/html;charset=utf-8;base64,{encoded_html}"
+    return f'<iframe src="{data_uri}" width="100%" height="920px" style="border:none;"></iframe>'
 def boost_prompt(prompt: str) -> str:
     """
     """
     return gr.update(open=True), history
+def execute_code(query: str):
+    """
+    Improved function to execute code directly from input
+    """
+    if not query or query.strip() == '':
+        return None, gr.update(active_key="empty")
+    try:
+        # Handle markdown code blocks or raw code
+        clean_code = remove_code_block(query)
+        # Add HTML structure if it's just a fragment
+        if not clean_code.strip().startswith('<!DOCTYPE') and not clean_code.strip().startswith('<html'):
+            if not ('<body' in clean_code and '</body>' in clean_code):
+                clean_code = f"""<!DOCTYPE html>
+<html>
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Game Preview</title>
+</head>
+<body>
+{clean_code}
+</body>
+</html>"""
+        return send_to_sandbox(clean_code), gr.update(active_key="render")
+    except Exception as e:
+        print(f"Execute code error: {str(e)}")
+        return None, gr.update(active_key="empty")
 # ------------------------
 # 6) 데모 클래스
                 }])
                 # 최종 결과(코드) + 샌드박스 미리보기
+                clean_code = remove_code_block(collected_content)
                 yield [
                     collected_content,
                     _history,
+                    send_to_sandbox(clean_code),
                     gr.update(active_key="render"),
                     gr.update(open=True)
                 ]
                         input_text = antd.InputTextarea(
                             size="large",
                             allow_clear=True,
+                            placeholder=random.choice(DEMO_LIST)['description'],
+                            rows=10,  # 입력창 높이 증가
+                            max_length=100000  # 입력 최대 길이 증가
                         )
                         gr.HTML('<div class="help-text">💡 원하는 게임의 설명을 입력하세요. 예: "Create a platformer game with jumping mechanics"</div>')
             )
             # (G) 'Code실행' 버튼 => 미리보기 iframe 로드
             execute_btn.click(
                 fn=execute_code,
                 inputs=[input_text],