Update app.py

app.py

@@ -31,6 +31,11 @@ def parse_code_from_response(response_text: str) -> str | None:
         return match.group(1).strip()
     return None
 
+# --- MCP 탭과 에이전트가 모두 사용할 전역 함수로 정의 ---
+def generate_c_code(description: str) -> str:
+    prompt = f"Generate a complete, compilable C code for this request: '{description}'. ONLY output the raw C code, without markdown or explanations."
+    return call_mistral_api([{"role": "user", "content": prompt}])
+
 def compile_and_run_c_code(code: str) -> str:
     try:
         with open("main.c", "w", encoding='utf-8') as f: f.write(code)
@@ -41,6 +46,10 @@ def compile_and_run_c_code(code: str) -> str:
         output = run_proc.stdout
         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:
+    prompt = f"You are a senior C code reviewer. Fulfill this instruction: '{instruction}'. If you refactor or change the code, YOU MUST provide the complete, new code in a ```c code block. \n\nC Code to Analyze:\n```c\n{code}\n```"
+    return call_mistral_api([{"role": "user", "content": prompt}])
 
 # --- 3. 진짜 '지능형' 에이전트 로직 ---
 def intelligent_agent_ide(initial_code: str, full_instruction: str):
@@ -48,7 +57,6 @@ def intelligent_agent_ide(initial_code: str, full_instruction: str):
         yield initial_code, "Error: Instruction cannot be empty."
         return
 
-    # 'and', 'then', ',' 등을 기준으로 명령어를 여러 단계로 분리
     tasks = [task.strip() for task in re.split(r'\s+and\s+|\s*,\s*then\s*|\s*그리고\s*', full_instruction, flags=re.IGNORECASE)]
     
     current_code = initial_code
@@ -61,15 +69,12 @@ def intelligent_agent_ide(initial_code: str, full_instruction: str):
         yield current_code, "\n".join(output_log)
         time.sleep(0.5)
 
-        # 1. 생성 (Generation)
         if "generate" in lower_task or "create" in lower_task or "만들어줘" in lower_task:
-            prompt = f"Generate a complete, compilable C code for this request: '{task}'. ONLY output the raw C code, without markdown or explanations."
-            new_code = call_mistral_api([{"role": "user", "content": prompt}])
+            new_code = generate_c_code(task)
             current_code = new_code if new_code and "Error" not in new_code else current_code
             output_log.append(f"✅ Code generated and updated in the editor.")
             yield current_code, "\n".join(output_log)
         
-        # 2. 컴파일 및 실행 (Compile & Run)
         elif "compile" in lower_task or "run" in lower_task or "실행" in lower_task:
             if not current_code:
                 output_log.append("❌ Error: Code is empty. Cannot compile.")
@@ -79,14 +84,12 @@ def intelligent_agent_ide(initial_code: str, full_instruction: str):
             output_log.append(result)
             yield current_code, "\n".join(output_log)
 
-        # 3. 분석/리팩토링 (Analyze/Refactor)
         else:
             if not current_code:
                 output_log.append("❌ Error: Code is empty. Cannot analyze.")
                 yield current_code, "\n".join(output_log)
                 break
-            prompt = f"You are a senior C code reviewer. Fulfill this instruction: '{task}'. If you refactor or change the code, YOU MUST provide the complete, new code in a ```c code block. \n\nC Code to Analyze:\n```c\n{current_code}\n```"
-            analysis_result = call_mistral_api([{"role": "user", "content": prompt}])
+            analysis_result = analyze_and_refactor_code(current_code, task)
             
             refactored_code = parse_code_from_response(analysis_result)
             if refactored_code:
@@ -100,7 +103,6 @@ def intelligent_agent_ide(initial_code: str, full_instruction: str):
     output_log.append("\n--- All tasks complete. ---")
     yield current_code, "\n".join(output_log)
 
-
 # --- 4. 통합된 Gradio UI ---
 with gr.Blocks(theme=gr.themes.Monochrome(primary_hue="indigo", secondary_hue="blue"), css="footer {visibility: hidden}") as demo:
     gr.Markdown("# 💻 The True C-Codestral IDE Agent")
@@ -125,14 +127,11 @@ with gr.Blocks(theme=gr.themes.Monochrome(primary_hue="indigo", secondary_hue="b
         with gr.TabItem("🛠️ MCP Tools API"):
             gr.Markdown("## Available MCP Tools for other Agents\nThese APIs are the building blocks of our IDE agent.")
             with gr.Accordion("Tool: Generate C Code", open=False):
-                # MCP용 툴은 'generate_c_code' 함수를 직접 노출
-                prompt_input = gr.Textbox(label="Description")
-                code_output = gr.Code(language="c", label="Generated C Code")
-                gr.Interface(fn=generate_c_code, inputs=prompt_input, outputs=code_output)
+                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):
-                code_input_run = gr.Code(language="c", label="C Code to Run")
-                text_output_run = gr.Textbox(label="Output")
-                gr.Interface(fn=compile_and_run_c_code, inputs=code_input_run, outputs=text_output_run)
+                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()