Update app.py

app.py

@@ -817,7 +817,7 @@ def execute_code(query: str):
 class Demo:
     def __init__(self):
         pass
-
+        
     async def generation_code(self, query: Optional[str], _setting: Dict[str, str], _history: Optional[History]):
         if not query or query.strip() == '':
             query = random.choice(DEMO_LIST)['description']
@@ -834,7 +834,6 @@ class Demo:
 3. 코드는 600줄을 넘지 않을 것
 4. 모든 코드는 하나의 HTML 파일에 통합할 것
 5. 핵심 기능만 구현하고 부가 기능은 생략할 것
-
 게임 요청: {query}
 """
         
@@ -865,7 +864,6 @@ class Demo:
                 gr.update(open=True)
             ]
             await asyncio.sleep(0)
-
             collected_content = None
             try:
                 # Claude API 시도
@@ -891,61 +889,54 @@ class Demo:
                     ]
                     await asyncio.sleep(0)
                 collected_content = content
-
             if collected_content:
                 # 코드 크기 확인하고 필요하면 경고 추가
                 clean_code = remove_code_block(collected_content)
                 code_lines = clean_code.count('\n') + 1
                 
-                if code_lines > 700:  # 여유 있게 250줄로 체크
+                if code_lines > 700:  # 여유 있게 700줄로 체크
                     warning_msg = f"""
 ⚠️ **경고: 생성된 코드가 너무 깁니다 ({code_lines}줄)**
-
 이로 인해 실행 시 오류가 발생할 수 있습니다. 다음과 같이 시도해 보세요:
 1. 더 간단한 게임을 요청하세요
 2. 특정 기능만 명시하여 요청하세요 (예: "간단한 Snake 게임, 점수 시스템 없이")
 3. "코드" 버튼을 사용하여 직접 실행해 보세요
-
 ```html
 {clean_code[:2000]}
-... (코드가 너무 깁니다) ...
-"""
+... (코드가 너무 깁니다) ... """
                     
-collected_content = warning_msg
-# 히스토리에는 추가하지 않음
-yield [
-collected_content,
-_history,
-None,
-gr.update(active_key="empty"),
-gr.update(open=True)
-]
-else:
-# 히스토리 갱신
-_history = messages_to_history([
-{'role': Role.SYSTEM, 'content': system_message}
-] + claude_messages + [{
-'role': Role.ASSISTANT,
-'content': collected_content
-}])
-
-# 최종 결과(코드) + 샌드박스 미리보기
-                yield [
-                    collected_content,
-                    _history,
-                    send_to_sandbox(clean_code),
-                    gr.update(active_key="render"),
-                    gr.update(open=True)
-                ]
-        else:
-            raise ValueError("No content was generated from either API")
-    except Exception as e:
-        raise ValueError(f'Error calling APIs: {str(e)}')
-
-def clear_history(self):
-    return []
-
-
+                    collected_content = warning_msg
+                    # 히스토리에는 추가하지 않음
+                    yield [
+                        collected_content,
+                        _history,
+                        None,
+                        gr.update(active_key="empty"),
+                        gr.update(open=True)
+                    ]
+                else:
+                    # 히스토리 갱신
+                    _history = messages_to_history([
+                        {'role': Role.SYSTEM, 'content': system_message}
+                    ] + claude_messages + [{
+                        'role': Role.ASSISTANT,
+                        'content': collected_content
+                    }])
+                    # 최종 결과(코드) + 샌드박스 미리보기
+                    yield [
+                        collected_content,
+                        _history,
+                        send_to_sandbox(clean_code),
+                        gr.update(active_key="render"),
+                        gr.update(open=True)
+                    ]
+            else:
+                raise ValueError("No content was generated from either API")
+        except Exception as e:
+            raise ValueError(f'Error calling APIs: {str(e)}')
+            
+    def clear_history(self):
+        return []
 
 # ------------------------
 # 7) Gradio / Modelscope UI 빌드