Update app.py

app.py

@@ -17,104 +17,98 @@ if API_KEY:
 else:
     print("FATAL: MISTRAL_API_KEY is not set in Space Secrets.")
 
-# --- 2. 백엔드 핵심 기능 ---
+# --- 2. 백엔드 핵심 기능 함수 ---
 def generate_c_code(description: str) -> str:
     if not CLIENT: return "Error: Mistral API key not configured."
-    prompt = f"Generate a complete, compilable C code for this request: '{description}'. ONLY output the raw C code within a single ```c code block."
+    prompt = f"You are a C programming expert. Generate a complete, compilable C code for this request: '{description}'. ONLY output the raw C code, without any markdown formatting or explanations."
     messages = [{"role": "user", "content": prompt}]
     response = CLIENT.chat(model="codestral-latest", messages=messages)
-    code = response.choices[0].message.content
-    if "```c" in code:
-        code = code.split("```c")[1].split("```")[0].strip()
-    elif "```" in code:
-        code = code.split("```")[1].split("```")[0].strip()
-    return code
+    return response.choices[0].message.content.strip()
 
 def compile_and_run_c_code(code: str) -> str:
     try:
-        with open("main.c", "w", encoding='utf-8') as f:
-            f.write(code)
-        compile_proc = subprocess.run(
-            ["gcc", "main.c", "-o", "main.out", "-lm", "-w"],
-            capture_output=True, text=True, timeout=15
-        )
-        if compile_proc.returncode != 0:
-            return f"--- COMPILATION FAILED ---\n{compile_proc.stderr}"
-        run_proc = subprocess.run(
-            ["./main.out"], capture_output=True, text=True, timeout=15
-        )
-        if run_proc.returncode != 0:
-            return f"--- RUNTIME ERROR ---\n{run_proc.stderr}"
+        with open("main.c", "w", encoding='utf-8') as f: f.write(code)
+        compile_proc = subprocess.run(["gcc", "main.c", "-o", "main.out", "-lm", "-w"], capture_output=True, text=True, timeout=15)
+        if compile_proc.returncode != 0: return f"--- COMPILATION FAILED ---\n{compile_proc.stderr}"
+        run_proc = subprocess.run(["./main.out"], capture_output=True, text=True, timeout=15)
+        if run_proc.returncode != 0: return f"--- RUNTIME ERROR ---\n{run_proc.stderr}"
         output = run_proc.stdout
-        if not output.strip():
-            return "--- EXECUTION SUCCEEDED ---\n(No output was produced)"
-        return f"--- EXECUTION SUCCEEDED ---\n{output}"
-    except Exception as e:
-        return f"--- SYSTEM ERROR ---\nAn unexpected error occurred: {str(e)}"
-
-# --- 3. Gradio 챗봇 로직 (Gradio 5.x+ 호환) ---
-def agent_chat(history: list[dict[str, str]]):
-    if not CLIENT:
-        history.append({"role": "assistant", "content": "MISTRAL_API_KEY가 Space Secrets에 설정되지 않았습니다."})
-        return history
-
-    # 사용자의 마지막 메시지가 history의 마지막에 있습니다.
-    message = history[-1]["content"]
-    lower_message = message.lower()
-    bot_response = ""
-
-    # API에 보낼 메시지는 현재 대화 전체입니다.
-    messages_for_api = history
-
-    if "compile" in lower_message or "run" in lower_message or "실행" in lower_message:
-        code_to_run = ""
-        # 자신을 제외한 이전 대화에서 코드 찾기
-        for item in reversed(history[:-1]):
-            if item.get("role") == "assistant" and "```c" in item.get("content", ""):
-                code_to_run = item["content"].split("```c")[1].split("```")[0].strip()
-                break
-        if not code_to_run:
-            bot_response = "컴파일할 코드를 찾을 수 없습니다. 먼저 코드를 생성해주세요."
-        else:
-            bot_response = compile_and_run_c_code(code_to_run)
-            
-    elif "generate" in lower_message or "create" in lower_message or "만들어줘" in lower_message or "짜줘" in lower_message:
-        generated_code = generate_c_code(message)
-        bot_response = f"코드가 생성되었습니다.\n\n```c\n{generated_code}\n```"
-        
-    else:
-        response = CLIENT.chat(model="codestral-latest", messages=messages_for_api)
-        bot_response = response.choices[0].message.content
+        return f"--- EXECUTION SUCCEEDED ---\n{output}" if output.strip() else "--- EXECUTION SUCCEEDED ---\n(No output was produced)"
+    except Exception as e: return f"--- SYSTEM ERROR ---\nAn unexpected error occurred: {str(e)}"
+
+def analyze_and_refactor_code(code: str, instruction: str) -> str:
+    if not CLIENT: return "Error: Mistral API key not configured."
+    prompt = f"""You are a senior C code reviewer. Analyze the following C code based on the user's instruction.
+Provide a clear, concise response. If refactoring, provide the complete improved code in a C code block.
 
-    history.append({"role": "assistant", "content": bot_response})
-    return history
+Instruction: '{instruction}'
+
+C Code to Analyze:
+```c
+{code}
+```"""
+    messages = [{"role": "user", "content": prompt}]
+    response = CLIENT.chat(model="codestral-latest", messages=messages)
+    return response.choices[0].message.content
 
-# --- 4. Gradio UI 구성 (Gradio 5.x+ 호환) ---
-with gr.Blocks(theme=gr.themes.Soft(primary_hue="green", secondary_hue="gray"), css="footer {visibility: hidden}") as demo:
-    gr.Markdown("# 🚀 C-Codestral Agent")
+# --- 3. 메인 에이전트 로직 ---
+def agent_logic(code: str, instruction: str):
+    if not instruction: return "Error: Instruction cannot be empty."
     
+    lower_instruction = instruction.lower()
+
+    if "compile" in lower_instruction or "run" in lower_instruction:
+        if not code: return "Error: Code editor is empty. Nothing to compile."
+        return compile_and_run_c_code(code)
+        
+    elif "generate" in lower_instruction or "create" in lower_instruction or "만들어줘" in lower_instruction:
+        # 새로운 코드를 생성하고, 그 결과를 코드 에디터에 직접 반환
+        new_code = generate_c_code(instruction)
+        # 출력은 코드 에디터로, 결과창에는 성공 메시지
+        return new_code, f"Code generation successful. The new code is now in the editor."
+
+    else: # Refactor, analyze, add comments, etc.
+        if not code: return "Error: Code editor is empty. Nothing to analyze."
+        # 분석/리팩토링 결과를 결과창에 반환
+        analysis_result = analyze_and_refactor_code(code, instruction)
+        # 코드 에디터는 그대로 두고, 결과창만 업데이트
+        return code, analysis_result
+
+# --- 4. Gradio UI 구성 (IDE 형태) ---
+with gr.Blocks(theme=gr.themes.Monochrome(primary_hue="indigo", secondary_hue="blue"), css="footer {visibility: hidden}") as demo:
+    gr.Markdown("# 💻 The C-Codestral IDE Agent")
+
     with gr.Tabs():
-        with gr.TabItem("🤖 Agent Chat"):
-            # Gradio 5.x 에서는 'type' 파라미터가 없습니다.
-            chatbot = gr.Chatbot(label="C-Agent", height=600, render_markdown=True)
-            msg = gr.Textbox(placeholder="Ask me to generate or run C code...", show_label=False, container=False)
-            
-            def add_user_message(message, history):
-                # 사용자 메시지를 먼저 history에 추가하고, 입력창을 비웁니다.
-                return history + [{"role": "user", "content": message}], ""
-
-            # 1. 사용자가 메시지를 보내면, add_user_message가 먼저 실행됩니다.
-            # 2. .then()을 통해, 업데이트된 history를 agent_chat으로 넘겨 봇 응답을 받습니다.
-            msg.submit(add_user_message, [msg, chatbot], [chatbot, msg], queue=False).then(
-                agent_chat, chatbot, chatbot
-            )
+        with gr.TabItem("👨‍💻 IDE"):
+            with gr.Row():
+                with gr.Column(scale=2):
+                    code_editor = gr.Code(label="C Code Editor", language="c", lines=25, interactive=True)
+                with gr.Column(scale=1):
+                    instruction_box = gr.Textbox(label="Instruction", placeholder="e.g., 'Compile and run', 'Find bugs', 'Add comments', 'Generate a factorial function'...")
+                    execute_btn = gr.Button("Execute Instruction", variant="primary")
+                    output_box = gr.Markdown(label="Output / Console")
+
+            def handle_ide_submit(code, instruction):
+                # 생성 요청 시 동작을 분기
+                if "generate" in instruction.lower() or "create" in instruction.lower():
+                    new_code, message = agent_logic(code, instruction)
+                    return new_code, message
+                else:
+                    # 컴파일 또는 분석
+                    result = agent_logic(code, instruction)
+                    # 이 경우, agent_logic은 하나의 값만 반환하므로, 코드 에디터는 그대로 유지
+                    return code, result
+
+            execute_btn.click(handle_ide_submit, inputs=[code_editor, instruction_box], outputs=[code_editor, output_box])
 
         with gr.TabItem("🛠️ MCP Tools API"):
             gr.Markdown("## Available MCP Tools\nThese APIs can be used by any MCP-compliant client.")
             with gr.Accordion("Tool: Generate C Code", open=False):
                 gr.Interface(fn=generate_c_code, inputs="text", outputs=gr.Code(language="c", label="Generated C Code"))
             with gr.Accordion("Tool: Compile & Run C Code", open=False):
-                gr.Interface(fn=compile_and_run_c_code, inputs=gr.Code(language="c", label="C Code to Run"), outputs=gr.Textbox(label="Output/Error"))
+                gr.Interface(fn=compile_and_run_c_code, inputs=gr.Code(language="c"), outputs=gr.Textbox(label="Output"))
+            with gr.Accordion("Tool: Analyze & Refactor C Code", open=False):
+                gr.Interface(fn=analyze_and_refactor_code, inputs=[gr.Code(language="c"), "text"], outputs=gr.Markdown())
 
 if __name__ == "__main__":
-    demo.queue().launch()
+    demo.launch()