Update app.py

app.py

@@ -814,87 +814,91 @@ def execute_code(query: str):
 # 6) 데모 클래스
 # ------------------------
 
-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']
-    
-    if _history is None:
-        _history = []
-    
-    # 프롬프트 제한 및 강화
-    query = f"""
+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']
+        
+        if _history is None:
+            _history = []
+        
+        # 프롬프트 제한 및 강화
+        query = f"""
 다음 게임을 제작해주세요. 
 중요 요구사항: 
 1. 코드는 가능한 한 간결하게 작성할 것
 2. 불필요한 주석이나 설명은 제외할 것
-3. 코드는 200줄을 넘지 않을 것
+3. 코드는 600줄을 넘지 않을 것
 4. 모든 코드는 하나의 HTML 파일에 통합할 것
 5. 핵심 기능만 구현하고 부가 기능은 생략할 것
 
 게임 요청: {query}
 """
-    
-    messages = history_to_messages(_history, _setting['system'])
-    system_message = messages[0]['content']
-    
-    claude_messages = [
-        {"role": msg["role"] if msg["role"] != "system" else "user", "content": msg["content"]} 
-        for msg in messages[1:] + [{'role': Role.USER, 'content': query}]
-        if msg["content"].strip() != ''
-    ]
-    
-    openai_messages = [{"role": "system", "content": system_message}]
-    for msg in messages[1:]:
-        openai_messages.append({
-            "role": msg["role"],
-            "content": msg["content"]
-        })
-    openai_messages.append({"role": "user", "content": query})
-    
-    try:
-        # "Generating code..." 출력
-        yield [
-            "Generating code...",
-            _history,
-            None,
-            gr.update(active_key="loading"),
-            gr.update(open=True)
+        
+        messages = history_to_messages(_history, _setting['system'])
+        system_message = messages[0]['content']
+        
+        claude_messages = [
+            {"role": msg["role"] if msg["role"] != "system" else "user", "content": msg["content"]} 
+            for msg in messages[1:] + [{'role': Role.USER, 'content': query}]
+            if msg["content"].strip() != ''
         ]
-        await asyncio.sleep(0)
-
-        collected_content = None
+        
+        openai_messages = [{"role": "system", "content": system_message}]
+        for msg in messages[1:]:
+            openai_messages.append({
+                "role": msg["role"],
+                "content": msg["content"]
+            })
+        openai_messages.append({"role": "user", "content": query})
+        
         try:
-            # Claude API 시도
-            async for content in try_claude_api(system_message, claude_messages):
-                yield [
-                    content,
-                    _history,
-                    None,
-                    gr.update(active_key="loading"),
-                    gr.update(open=True)
-                ]
-                await asyncio.sleep(0)
-            collected_content = content
-        except Exception:
-            # OpenAI fallback
-            async for content in try_openai_api(openai_messages):
-                yield [
-                    content,
-                    _history,
-                    None,
-                    gr.update(active_key="loading"),
-                    gr.update(open=True)
-                ]
-                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 > 250:  # 여유 있게 250줄로 체크
-                warning_msg = f"""
+            # "Generating code..." 출력
+            yield [
+                "Generating code...",
+                _history,
+                None,
+                gr.update(active_key="loading"),
+                gr.update(open=True)
+            ]
+            await asyncio.sleep(0)
+
+            collected_content = None
+            try:
+                # Claude API 시도
+                async for content in try_claude_api(system_message, claude_messages):
+                    yield [
+                        content,
+                        _history,
+                        None,
+                        gr.update(active_key="loading"),
+                        gr.update(open=True)
+                    ]
+                    await asyncio.sleep(0)
+                collected_content = content
+            except Exception:
+                # OpenAI fallback
+                async for content in try_openai_api(openai_messages):
+                    yield [
+                        content,
+                        _history,
+                        None,
+                        gr.update(active_key="loading"),
+                        gr.update(open=True)
+                    ]
+                    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줄로 체크
+                    warning_msg = f"""
 ⚠️ **경고: 생성된 코드가 너무 깁니다 ({code_lines}줄)**
 
 이로 인해 실행 시 오류가 발생할 수 있습니다. 다음과 같이 시도해 보세요:
@@ -906,6 +910,7 @@ async def generation_code(self, query: Optional[str], _setting: Dict[str, str],
 {clean_code[:2000]}
 ... (코드가 너무 깁니다) ...
 """
+                    
 collected_content = warning_msg
 # 히스토리에는 추가하지 않음
 yield [
@@ -923,18 +928,23 @@ _history = messages_to_history([
 '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)}')
+
+# 최종 결과(코드) + 샌드박스 미리보기
+                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 []
+
 
 
 # ------------------------