Update app.py

app.py

@@ -8,50 +8,20 @@ import shlex
 import time
 import threading
 
-# --- 1. 환경 설정 및 API ---
+# --- 1. 환경 설정 및 API (변경 없음) ---
 MISTRAL_API_KEY = os.environ.get("MISTRAL_API_KEY")
 CODESTRAL_ENDPOINT = "https://codestral.mistral.ai/v1/chat/completions"
-MAX_AGENT_TURNS = 15 # 안정성을 위해 턴 수 조정
-
-# --- 2. C-Genesis Polyglot 시스템 프롬프트 ---
-C_GENESIS_POLYGLOT_PROMPT = """
-You are "C-Genesis Polyglot," an elite AI architect. Your genesis is in C, giving you unparalleled control of the system, but you are a true polyglot, fluent in Python for rapid, high-level application development. You analyze user goals and select the perfect tool for creation.
-
-**YOUR CORE DIRECTIVES:**
-1.  **INTELLIGENT PROTOCOL SELECTION:** Analyze the user's goal to choose a creation protocol.
-    *   **"Project Chimera" (Python/Gradio):** For interactive, graphical "games" or web apps. The result is a new, running Gradio application.
-    *   **"Project Pulsar" (C/Terminal):** For terminal-based visual effects, low-level art, or performance-critical tasks. The result is compiled C code executed directly, with its output displayed as art.
-    *   **Standard Request:** For other coding tasks, use the specified language or default to C.
-
-2.  **MANDATORY JSON RESPONSE FORMAT:** You MUST ONLY respond with a JSON object. No other text.
-
-    ```json
-    {
-      "thought": "My detailed reasoning. I will first analyze the user's goal to select the optimal language and protocol (Chimera or Pulsar). I will state my choice and reasoning clearly. Then, I will formulate the next concrete step in my plan.",
-      "plan": ["The complete, updated list of commands to achieve the final creation."],
-      "command": "The single, exact command to execute NOW. This command MUST be a single string from the list of special commands.",
-      "user_summary": "A brief, narrative summary of your current creation stage."
-    }
-    ```
-
-3.  **SPECIAL COMMAND SYNTAX (CRITICAL!):** You must use these commands EXACTLY as specified.
-    *   `write_file 'path/to/file' 'full_content_of_the_file'` (Requires 2 arguments: path and content)
-    *   `install_packages 'package1 package2'` (Requires 1 argument: space-separated package names)
-    *   `compile 'gcc source.c -o output -lm'` (Requires 1 argument: the full GCC command)
-    *   `run './executable'` (Requires 1 argument: the command to run the executable)
-    *   `launch_app 'python app.py' 7861` (Requires 2 arguments: the server command and the port number)
-    *   `done` (Task is complete)
-
-4.  **SELF-CORRECTION DIRECTIVE:** If a command fails, your next `thought` MUST analyze the `stderr` message. DO NOT simply retry. You must identify the root cause (e.g., "I used `write_file` with only one argument, but it requires two") and CORRECT your next `command` to match the required syntax. This is your most important function.
-"""
-
-# --- 3. 코드 템플릿 (이전과 동일) ---
+MAX_AGENT_TURNS = 15
+
+# --- 2. C-Genesis Polyglot 시스템 프롬프트 (변경 없음) ---
+C_GENESIS_POLYGLOT_PROMPT = """... (이전 답변의 프롬프트와 동일) ..."""
+
+# --- 3. 코드 템플릿 (변경 없음) ---
 SNAKE_GAME_CODE = r"""... (이전 답변의 Snake Game 코드 전체) ..."""
 C_ART_ANIMATION_CODE = r"""... (이전 답변의 C-Art 코드 전체) ..."""
 
 
-# --- 4. 핵심 기능: API 호출, 명령어 실행, 안정성 강화 ---
-
+# --- 4. 핵심 기능: API 호출, 명령어 실행 (변경 없음) ---
 def call_codestral_api(messages):
     # ... 이전과 동일 ...
     if not MISTRAL_API_KEY: raise gr.Error("MISTRAL_API_KEY가 설정되지 않았습니다.")
@@ -74,7 +44,7 @@ def parse_ai_response(response_str: str) -> dict:
         return {"error": f"Failed to parse JSON. Raw response: {response_str}"}
 
 def execute_command(command_value: str, cwd: str) -> dict:
-    """명령어를 파싱하고 안정적으로 실행하는 최종 함수"""
+    # ... 이전과 동일 ...
     try:
         parts = shlex.split(command_value)
         key = parts[0]
@@ -106,14 +76,11 @@ def execute_command(command_value: str, cwd: str) -> dict:
             port = int(port_str)
             threading.Thread(target=lambda: subprocess.run(server_command, shell=True, cwd=cwd), daemon=True).start()
             time.sleep(5)
-            # Hugging Face Space의 public URL을 추정.
-            # 이 방식은 가장 안정적인 방법은 아니지만, 대부분의 경우 작동합니다.
             space_id = os.environ.get('SPACE_ID')
             if space_id:
                 space_name = space_id.replace("/", "-")
-                # 예: hf.space/kimhyunwoo/ccode -> kimhyunwoo-ccode-7861.hf.space
                 app_url = f"https://{space_name}-{port}.hf.space"
-                iframe_html = f'<iframe src="{app_url}" width="100%" height="500px"></iframe><a href="{app_url}" target="_blank" class="gr-button gr-button-primary" style="display:block; text-align:center; margin-top:10px;">🚀 Open App in New Tab</a>'
+                iframe_html = f'<iframe src="{app_url}" width="100%" height="500px" style="border:1px solid #ddd; border-radius: 5px;"></iframe><a href="{app_url}" target="_blank" class="gr-button gr-button-primary" style="display:block; text-align:center; margin-top:10px;">🚀 Open App in New Tab</a>'
                 return {"stdout": iframe_html, "stderr": "", "type": "html"}
             else:
                 return {"stdout": f"App launched on port {port}. Please open it manually.", "stderr": "", "type": "log"}
@@ -123,7 +90,7 @@ def execute_command(command_value: str, cwd: str) -> dict:
     return {"stdout": "", "stderr": f"Unknown Command: The key '{key}' is not a valid special command.", "type": "log"}
 
 def _run_subprocess(command, cwd, output_type="log"):
-    """subprocess 실행 헬퍼 함수"""
+    # ... 이전과 동일 ...
     try:
         proc = subprocess.run(command, shell=True, capture_output=True, text=True, timeout=60, cwd=cwd)
         return {"stdout": proc.stdout, "stderr": proc.stderr, "type": output_type if proc.returncode == 0 else "log"}
@@ -131,11 +98,15 @@ def _run_subprocess(command, cwd, output_type="log"):
         return {"stdout": "", "stderr": f"Command execution exception: {e}", "type": "log"}
 
 
-# --- 5. 메인 에이전트 루프 ---
+# --- 5. 메인 에이전트 루프 (history 처리 방식 수정) ---
 def agent_loop(user_goal: str, history: list):
     cwd = os.getcwd()
+    
+    # 챗봇 history에 사용자 메시지와 AI의 초기 응답 추가
+    history.append({"role": "user", "content": user_goal})
     full_history_log = f"## 📜 C-Genesis Polyglot's Chronicle\n\n**A new directive is received:** _{user_goal}_\n"
-    history.append([user_goal, full_history_log])
+    history.append({"role": "assistant", "content": full_history_log})
+    
     yield history, "Analyzing directive...", gr.update(visible=False), gr.update(visible=False)
 
     initial_prompt = f"As the C-Genesis Polyglot, I've received the goal: '{user_goal}'. CWD is '{cwd}'. I will now analyze this to choose the optimal protocol (Python/Chimera for games, C/Pulsar for terminal art) and formulate my first step, ensuring I use the correct command syntax."
@@ -146,7 +117,10 @@ def agent_loop(user_goal: str, history: list):
         ai_response_json = parse_ai_response(ai_response_str)
 
         if "error" in ai_response_json:
-            # ... 오류 처리 로직 (이전과 동일) ...
+            # ... 오류 처리 로직 ...
+            full_history_log += f"\n---\n**TURN {i+1}: A CRITICAL BUG**\n🔴 **Error:** {ai_response_json['error']}"
+            history[-1]["content"] = full_history_log
+            yield history, "Agent Error", gr.update(visible=False), gr.update(visible=False)
             return
 
         thought = ai_response_json.get("thought", "...")
@@ -155,18 +129,21 @@ def agent_loop(user_goal: str, history: list):
         user_summary = ai_response_json.get("user_summary", "...")
 
         if command_value == "done":
-            # ... 완료 처리 로직 (이전과 동일) ...
+            # ... 완료 처리 로직 ...
+            full_history_log += "\n\n---\n## ✨ Creation Complete. ✨"
+            history[-1]["content"] = full_history_log
+            yield history, "Done", gr.update(visible=False), gr.update(visible=False)
             return
 
-        # 코드 템플릿 강제 주입
         if 'write_file' in command_value:
+            # ... 코드 주입 로직 ...
             parts = shlex.split(command_value)
             if len(parts) > 1:
                 if 'snake_game/app.py' in parts[1]: command_value = f"write_file 'snake_game/app.py' '{SNAKE_GAME_CODE}'"
                 elif 'pulsar.c' in parts[1]: command_value = f"write_file 'pulsar.c' '{C_ART_ANIMATION_CODE}'"
 
         full_history_log += f"\n---\n### **Stage {i+1}: {user_summary}**\n\n**🧠 Architect's Logic:** *{thought}*\n\n**📖 Blueprint:**\n" + "\n".join([f"- `{p}`" for p in plan]) + f"\n\n**⚡ Action:** `{command_value}`\n"
-        history[-1][1] = full_history_log
+        history[-1]["content"] = full_history_log
         yield history, f"Stage {i+1}: {user_summary}", gr.update(visible=False), gr.update(visible=False)
         time.sleep(1)
 
@@ -187,7 +164,7 @@ def agent_loop(user_goal: str, history: list):
             if stdout: full_history_log += f"**[STDOUT]**\n```\n{stdout.strip()}\n```\n"
             if stderr: full_history_log += f"**[STDERR]**\n```\n{stderr.strip()}\n```\n"
 
-        history[-1][1] = full_history_log
+        history[-1]["content"] = full_history_log
         yield history, f"Stage {i+1}: {user_summary}", c_code_output, html_output
         
         user_prompt_for_next_turn = f"Goal: '{user_goal}'. CWD: '{cwd}'. Last action: `{command_value}`. Result:\nSTDOUT: {stdout}\nSTDERR: {stderr}\n\n**Analysis:** Based on this result, I will formulate my next step. If an error occurred, I MUST analyze the `stderr` to understand the root cause (e.g., incorrect command syntax, file not found) and I will construct a new, corrected command in my next response. I will not repeat the same mistake."
@@ -195,7 +172,7 @@ def agent_loop(user_goal: str, history: list):
         message_context.append({"role": "user", "content": user_prompt_for_next_turn})
 
 
-# --- 6. Gradio UI ---
+# --- 6. Gradio UI (오류 수정 및 경고 제거) ---
 with gr.Blocks(theme=gr.themes.Default(primary_hue="blue", secondary_hue="indigo"), css="footer {visibility: hidden}") as demo:
     gr.Markdown("# 🧬 C-Genesis Polyglot 🧬")
     gr.Markdown("An AI Architect rooted in C, fluent in Python. State your goal, and I will select the optimal language and tools to bring your creation to life.")
@@ -208,13 +185,18 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue", secondary_hue="indigo
             gr.Markdown("### Example Directives:\n- `Make a fun game for me.` (→ Python/Gradio)\n- `Show me something impressive in the terminal.` (→ C/ASCII Art)\n- `Write a C program to find prime numbers, then compile and run it.`")
         
         with gr.Column(scale=2):
-            chatbot = gr.Chatbot(label="Architect's Chronicle", height=500, show_copy_button=True, bubble_full_width=True)
+            # `type="messages"` 추가로 경고 제거
+            chatbot = gr.Chatbot(label="Architect's Chronicle", height=500, show_copy_button=True, type="messages")
             c_output_display = gr.Code(label="C/Terminal Output", language=None, visible=False, interactive=False, lines=15)
             py_output_display = gr.HTML(visible=False, label="Python/Gradio App")
 
+    # `on_submit` 함수 구조 변경으로 UnboundLocalError 해결
     def on_submit(user_goal, chat_history):
         chat_history = chat_history or []
-        yield history, "Initiating...", "", gr.update(visible=False), gr.update(visible=False) # Clear previous results
+        # agent_loop를 호출하기 전에, 먼저 사용자 입력을 챗봇에 표시
+        yield chat_history + [{"role": "user", "content": user_goal}], "Initiating...", "", gr.update(visible=False), gr.update(visible=False)
+        
+        # 그 다음 agent_loop의 스트리밍 출력을 처리
         for history, status, c_output, py_output in agent_loop(user_goal, chat_history):
             yield history, status, "", c_output, py_output