Spaces:

openfree
/

Vibe-Game

Paused

App Files Files Community

openfree commited on Apr 24

Commit

c549e3b

verified ·

1 Parent(s): 1a919ed

Update app.py

Browse files

Files changed (1) hide show

app.py +39 -8

app.py CHANGED Viewed

@@ -526,30 +526,46 @@ def deploy_to_vercel(code: str):
 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>
@@ -563,7 +579,7 @@ def send_to_sandbox(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>'
@@ -634,14 +650,21 @@ def history_render(history: 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>
@@ -741,7 +764,15 @@ class Demo:
                 }])
                 # 최종 결과(코드) + 샌드박스 미리보기
                 clean_code = remove_code_block(collected_content)
                 yield [
                     collected_content,
                     _history,

 def remove_code_block(text):
     """
     More robust function to extract code from markdown code blocks
+    텍스트에서 ```html 및 ``` 태그를 완전히 제거하는 함수
     """
+    # ```html 태그로 둘러싸인 코드 블록 찾기
     pattern = r'```html\s*([\s\S]+?)\s*```'
     match = re.search(pattern, text, re.DOTALL)
     if match:
         return match.group(1).strip()
+    # 일반 코드 블록 처리
     pattern = r'```(?:\w+)?\s*([\s\S]+?)\s*```'
     match = re.search(pattern, text, re.DOTALL)
     if match:
         return match.group(1).strip()
+    # 텍스트에 ```html과 ```가 포함된 경우 명시적으로 제거
+    text = re.sub(r'```html\s*', '', text)
+    text = re.sub(r'\s*```', '', text)
+    # 코드 블록이 없는 경우 원본 텍스트 반환
     return text.strip()
 def send_to_sandbox(code):
     """
     Improved function to create iframe with proper code cleaning
+    ```html 태그가 확실히 제거된 코드를 iframe으로 렌더링
     """
+    # 코드에서 마크다운 표기 제거
     clean_code = remove_code_block(code)
+    # 디버깅: 코드 앞부분 확인
+    code_start = clean_code[:50] if len(clean_code) > 50 else clean_code
+    print(f"Code start: {code_start}")
+    # ```html 태그가 여전히 있으면 명시적으로 제거
+    if clean_code.startswith('```html'):
+        clean_code = clean_code[7:].strip()
+    if clean_code.endswith('```'):
+        clean_code = clean_code[:-3].strip()
+    # 기본 HTML 구조 추가
     if not clean_code.strip().startswith('<!DOCTYPE') and not clean_code.strip().startswith('<html'):
         clean_code = f"""<!DOCTYPE html>
 <html>
 </body>
 </html>"""
+    # 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 execute_code(query: str):
     """
     Improved function to execute code directly from input
+    코드 실행 시 ```html 태그를 확실히 제거
     """
     if not query or query.strip() == '':
         return None, gr.update(active_key="empty")
     try:
+        # 코드 정제 - 마크다운 태그 철저히 제거
         clean_code = remove_code_block(query)
+        # ```html 태그가 여전히 있으면 명시적으로 제거
+        if clean_code.startswith('```html'):
+            clean_code = clean_code[7:].strip()
+        if clean_code.endswith('```'):
+            clean_code = clean_code[:-3].strip()
+        # HTML 구조 추가
         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 제거
                 clean_code = remove_code_block(collected_content)
+                # 디버그 출력
+                print(f"Original content start: {collected_content[:30]}")
+                print(f"Cleaned code start: {clean_code[:30]}")
+                # 마크다운 형식의 collected_content는 그대로 출력
+                # 하지만 샌드박스에는 정제된 clean_code 전달
                 yield [
                     collected_content,
                     _history,