Update app.py

app.py

@@ -6,204 +6,187 @@ import json
 import re
 import shlex
 import time
-import threading
 
-# --- 1. 환경 설정 및 API (변경 없음) ---
 MISTRAL_API_KEY = os.environ.get("MISTRAL_API_KEY")
 CODESTRAL_ENDPOINT = "https://codestral.mistral.ai/v1/chat/completions"
-MAX_AGENT_TURNS = 15
+MAX_AGENT_TURNS = 10
+
+C_GENESIS_PROMPT = """
+You are the "C-Genesis Weaver," a master AI artisan who creates digital art using only the C language and standard Linux commands. Your mission is to bring a user's creative request to life.
+
+**YOUR DIRECTIVE:**
+When the user asks for something "fun," "cool," or "impressive," you must execute your masterpiece protocol: **The "Cosmic Pulsar" Animation.**
+
+**MASTER PLAN (Follow these steps EXACTLY):**
+1.  **Step 1: Write the C code.** Your command must be: `echo "..." > pulsar.c` (The content of "..." is a placeholder, the system will inject the real code).
+2.  **Step 2: Compile the C code.** Your command must be: `gcc pulsar.c -o pulsar -lm -O2`
+3.  **Step 3: Run the executable.** Your command must be: `./pulsar`
+4.  **Step 4: Finish.** Your command must be: `done`
+
+**RESPONSE FORMAT (CRITICAL!):**
+You MUST ONLY respond with a JSON object in the following format. Do not add any text before or after the JSON.
+{
+  "thought": "Your step-by-step reasoning. Explain which step of the Master Plan you are on and why.",
+  "command": "The single, exact Linux command for the current step from the Master Plan."
+}
+
+**SELF-CORRECTION:**
+If a command like gcc fails, analyze the `stderr` message. In your next thought, explain the error. Then, for your next command, try to fix it. For example, by re-running the `echo` command to "rewrite" the (supposedly) faulty code.
+"""
+
+C_ART_ANIMATION_CODE = r"""
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <math.h>
+#include <unistd.h>
+#include <time.h>
+#define WIDTH 80
+#define HEIGHT 40
+#define NUM_FRAMES 150
+#define C_RESET   "\x1B[0m"
+#define C_RED     "\x1B[31m"
+#define C_YELLOW  "\x1B[33m"
+#define C_BLUE    "\x1B[34m"
+#define C_CYAN    "\x1B[36m"
+#define C_WHITE   "\x1B[37m"
+const char* colors[] = {C_RED, C_YELLOW, C_BLUE, C_CYAN, C_WHITE};
+const int num_colors = sizeof(colors) / sizeof(colors[0]);
+void render_frame(float t) {
+    char screen[HEIGHT][WIDTH+1];
+    for(int y=0; y<HEIGHT; y++) {
+        for(int x=0; x<WIDTH; x++) screen[y][x] = ' ';
+        screen[y][WIDTH] = '\0';
+    }
+    for(int i=0; i<1000; i++) {
+        float angle = (float)i * 0.1 + t*2.0;
+        float radius = 0.1 + (float)i / 1000.0 * 0.4;
+        float x = cos(angle) * radius * 2.0 * (WIDTH/HEIGHT);
+        float y = sin(angle) * radius;
+        int screen_x = (int)(x * WIDTH/2 + WIDTH/2);
+        int screen_y = (int)(y * HEIGHT/2 + HEIGHT/2);
+        if(screen_x >= 0 && screen_x < WIDTH && screen_y >=0 && screen_y < HEIGHT) {
+            int color_index = (i + (int)(t*20)) % num_colors;
+            char L = ".,-~:;=!*#$@"[i%13];
+            sprintf(&screen[screen_y][screen_x], "%s%c", colors[color_index], L);
+        }
+    }
+    for(int y=0; y<HEIGHT; y++) printf("%s\n", screen[y]);
+}
+int main() {
+    printf("\e[?25l");
+    for (int i=0; i<NUM_FRAMES; i++) {
+        printf("\x1B[H\x1B[J");
+        float t = (float)i / NUM_FRAMES;
+        render_frame(t * M_PI * 2.0);
+        usleep(50000);
+    }
+    printf("\e[?25h"); printf(C_RESET);
+    return 0;
+}
+"""
 
-# --- 2. C-Genesis Polyglot 시스템 프롬프트 (변경 없음) ---
-C_GENESIS_POLYGLOT_PROMPT = """... (이전 답변의 프롬프트와 동일) ..."""
-
-# --- 3. 코드 템플릿 (변경 없음) ---
-SNAKE_GAME_CODE = r"""... (이전 답변의 Snake Game 코드 전체) ..."""
-C_ART_ANIMATION_CODE = r"""... (이전 답변의 C-Art 코드 전체) ..."""
-
-
-# --- 4. 핵심 기능: API 호출, 명령어 실행 (변경 없음) ---
 def call_codestral_api(messages):
-    # ... 이전과 동일 ...
-    if not MISTRAL_API_KEY: raise gr.Error("MISTRAL_API_KEY가 설정되지 않았습니다.")
+    if not MISTRAL_API_KEY:
+        raise gr.Error("MISTRAL_API_KEY가 설정되지 않았습니다.")
     headers = {"Authorization": f"Bearer {MISTRAL_API_KEY}", "Content-Type": "application/json"}
     data = {"model": "codestral-latest", "messages": messages, "response_format": {"type": "json_object"}}
     try:
-        response = requests.post(CODESTRAL_ENDPOINT, headers=headers, data=json.dumps(data), timeout=120)
+        response = requests.post(CODESTRAL_ENDPOINT, headers=headers, data=json.dumps(data), timeout=60)
         response.raise_for_status()
         return response.json()["choices"][0]["message"]["content"]
     except Exception as e:
         return json.dumps({"error": f"API Call Error: {e}"})
 
 def parse_ai_response(response_str: str) -> dict:
-    # ... 이전과 동일 ...
     try:
-        match = re.search(r'\{.*\}', response_str, re.DOTALL)
-        if match: return json.loads(match.group(0))
         return json.loads(response_str)
     except Exception as e:
         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]
-        args = parts[1:]
-    except (ValueError, IndexError):
-        return {"stdout": "", "stderr": f"Command Parsing Error: Invalid command format '{command_value}'", "type": "log"}
-
-    if key == "write_file":
-        if len(args) != 2: return {"stdout": "", "stderr": f"Syntax Error: `write_file` requires 2 arguments (path, content), but {len(args)} were given.", "type": "log"}
-        path, content = args[0], args[1]
-        try:
-            with open(os.path.join(cwd, path), "w", encoding='utf-8') as f: f.write(content)
-            return {"stdout": f"File '{path}' written successfully.", "stderr": "", "type": "log"}
-        except Exception as e: return {"stdout": "", "stderr": f"Error writing file: {e}", "type": "log"}
-
-    if key == "install_packages":
-        if len(args) != 1: return {"stdout": "", "stderr": f"Syntax Error: `install_packages` requires 1 argument (packages string), but {len(args)} were given.", "type": "log"}
-        return _run_subprocess(f"pip install {args[0]}", cwd)
-
-    if key == "compile" or key == "run":
-        if len(args) != 1: return {"stdout": "", "stderr": f"Syntax Error: `{key}` requires 1 argument (the command), but {len(args)} were given.", "type": "log"}
-        output_type = "animation" if key == "run" else "log"
-        return _run_subprocess(args[0], cwd, output_type)
-
-    if key == "launch_app":
-        if len(args) != 2: return {"stdout": "", "stderr": f"Syntax Error: `launch_app` requires 2 arguments (command, port), but {len(args)} were given.", "type": "log"}
-        server_command, port_str = args[0], args[1]
-        try:
-            port = int(port_str)
-            threading.Thread(target=lambda: subprocess.run(server_command, shell=True, cwd=cwd), daemon=True).start()
-            time.sleep(5)
-            space_id = os.environ.get('SPACE_ID')
-            if space_id:
-                space_name = space_id.replace("/", "-")
-                app_url = f"https://{space_name}-{port}.hf.space"
-                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"}
-        except (ValueError, Exception) as e:
-            return {"stdout": "", "stderr": f"Error launching app: {e}", "type": "log"}
-
-    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"):
-    # ... 이전과 동일 ...
+def execute_command(command: str, cwd: str) -> dict:
+    command = command.strip()
+    if command.startswith('echo'):
+        command = f"echo '{C_ART_ANIMATION_CODE}' > pulsar.c"
+
     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"}
+        output_type = "animation" if command.strip().startswith('./pulsar') and proc.returncode == 0 else "log"
+        return {"stdout": proc.stdout, "stderr": proc.stderr, "type": output_type}
     except Exception as e:
         return {"stdout": "", "stderr": f"Command execution exception: {e}", "type": "log"}
 
-
-# --- 5. 메인 에이전트 루프 (history 처리 방식 수정) ---
-def agent_loop(user_goal: str, history: list):
+def agent_loop(user_goal: str):
     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({"role": "assistant", "content": full_history_log})
-    
-    yield history, "Analyzing directive...", gr.update(visible=False), gr.update(visible=False)
+    full_history_log = f"## 📜 C-Genesis Weaver's Chronicle\n\n**A directive is received:** _{user_goal}_\n"
+    yield full_history_log, "Analyzing directive...", 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."
-    message_context = [{"role": "system", "content": C_GENESIS_POLYGLOT_PROMPT}, {"role": "user", "content": initial_prompt}]
+    message_context = [{"role": "system", "content": C_GENESIS_PROMPT}, {"role": "user", "content": f"My goal is: '{user_goal}'. Begin your Master Plan."}]
     
     for i in range(MAX_AGENT_TURNS):
         ai_response_str = call_codestral_api(message_context)
         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)
+            full_history_log += f"\n---\n**CRITICAL FAILURE:**\n🔴 **Error:** {ai_response_json['error']}"
+            yield full_history_log, "Agent Error", gr.update(visible=False)
             return
 
         thought = ai_response_json.get("thought", "...")
-        plan = ai_response_json.get("plan", [])
-        command_value = ai_response_json.get("command", "done")
-        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
+        command = ai_response_json.get("command", "done")
 
-        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]["content"] = full_history_log
-        yield history, f"Stage {i+1}: {user_summary}", gr.update(visible=False), gr.update(visible=False)
+        if command == "done":
+            full_history_log += "\n\n---\n## ✨ The Creation is Complete. ✨"
+            yield full_history_log, "Done", gr.update(visible=False)
+            return
+        
+        user_summary = f"Stage {i+1}: " + ("Writing C code..." if "echo" in command else "Compiling..." if "gcc" in command else "Running the art..." if "./pulsar" in command else "Executing...")
+        full_history_log += f"\n---\n### **{user_summary}**\n\n**🧠 Artisan's Thought:** *{thought}*\n\n**⚡ Action:** `{command}`\n"
+        yield full_history_log, user_summary, gr.update(visible=False)
         time.sleep(1)
 
-        exec_result = execute_command(command_value, cwd)
+        exec_result = execute_command(command, cwd)
         stdout, stderr, output_type = exec_result["stdout"], exec_result["stderr"], exec_result["type"]
         
-        c_code_output = gr.update(visible=False)
-        html_output = gr.update(visible=False)
-
+        output_display_update = gr.update(visible=False)
         full_history_log += f"\n**Result:**\n"
         if output_type == "animation":
-            full_history_log += "*(A C-based animation unfolds below...)*"
-            c_code_output = gr.Code(value=stdout if stdout else stderr, language=None, visible=True, label="C/Terminal Output")
-        elif output_type == "html":
-            full_history_log += "*(A Python/Gradio app has been launched. See below or open in a new tab...)*"
-            html_output = gr.HTML(stdout, visible=True)
+            full_history_log += "*(The animated tapestry unfolds below...)*"
+            output_display_update = gr.Code(value=stdout if stdout else stderr, language=None, visible=True, label="Live Terminal Output")
         else:
-            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"
+            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]["content"] = full_history_log
-        yield history, f"Stage {i+1}: {user_summary}", c_code_output, html_output
+        yield full_history_log, user_summary, output_display_update
         
-        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."
+        user_prompt_for_next_turn = f"The last action was `{command}`. The result was:\nSTDOUT: {stdout}\nSTDERR: {stderr}\n\nBased on this, continue to the next step of the Master Plan. If an error occurred, analyze it and correct your course."
         message_context.append({"role": "assistant", "content": json.dumps(ai_response_json)})
         message_context.append({"role": "user", "content": user_prompt_for_next_turn})
 
-
-# --- 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.")
+with gr.Blocks(theme=gr.themes.Monochrome(primary_hue="teal", secondary_hue="green"), css="footer {visibility: hidden}") as demo:
+    gr.Markdown("# 🧬 The C-Genesis Weaver 🧬")
+    gr.Markdown("An AI artisan who weaves digital tapestries with C code. Ask for something `fun` or `impressive` to witness the creation.")
     
     with gr.Row():
         with gr.Column(scale=1):
             status_box = gr.Textbox(label="Current Stage", interactive=False)
-            user_input = gr.Textbox(label="State Your Directive", placeholder="e.g., 'Make a fun game' or 'Show me cool terminal art'")
-            submit_btn = gr.Button("▶️ Create", variant="primary")
-            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.`")
-        
+            user_input = gr.Textbox(label="State Your Directive", placeholder="e.g., 'Make something fun for me'")
+            submit_btn = gr.Button("▶️ Weave", variant="primary")
         with gr.Column(scale=2):
-            # `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 []
-        # 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
+            chronicle_display = gr.Markdown()
+    
+    output_display = gr.Code(label="Live Terminal Output", language=None, visible=False, interactive=False, lines=20)
+
+    def on_submit(user_goal):
+        for chronicle, status, output in agent_loop(user_goal):
+            yield chronicle, status, "", output
 
     submit_btn.click(
         on_submit,
-        inputs=[user_input, chatbot],
-        outputs=[chatbot, status_box, user_input, c_output_display, py_output_display]
+        inputs=[user_input],
+        outputs=[chronicle_display, status_box, user_input, output_display]
     )
 
 if __name__ == "__main__":