Update app.py

app.py

@@ -1,46 +1,35 @@
 import gradio as gr
 import os
 import subprocess
-import requests  # mistralai 대신 requests 사용
+import requests
 import json
+import re
 
-# --- 1. 환경 설정 ---
+# --- 1. 환경 설정 및 API 호출 함수 ---
 API_KEY = os.environ.get("MISTRAL_API_KEY")
-CODESTRAL_ENDPOINT = "https://codestral.mistral.ai/v1/chat/completions" # 직접 호출할 엔드포인트
-HEADERS = {
-    "Authorization": f"Bearer {API_KEY}",
-    "Content-Type": "application/json"
-}
+CODESTRAL_ENDPOINT = "https://codestral.mistral.ai/v1/chat/completions"
+HEADERS = {"Authorization": f"Bearer {API_KEY}", "Content-Type": "application/json"}
 
 if not API_KEY:
     print("FATAL: MISTRAL_API_KEY is not set in Space Secrets.")
 
-# --- 2. 백엔드 핵심 기능 함수 (requests 사용) ---
 def call_mistral_api(messages):
     if not API_KEY: return "Error: MISTRAL_API_KEY is not configured."
-    
-    data = {
-        "model": "codestral-latest",
-        "messages": messages
-    }
-    
+    data = {"model": "codestral-latest", "messages": messages}
     try:
         response = requests.post(CODESTRAL_ENDPOINT, headers=HEADERS, data=json.dumps(data), timeout=30)
-        response.raise_for_status()  # 2xx가 아닌 상태 코드에 대해 에러 발생
+        response.raise_for_status()
         return response.json()["choices"][0]["message"]["content"]
     except requests.exceptions.RequestException as e:
         return f"API Call Error: {e}"
 
-def generate_c_code(description: str) -> str:
-    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}]
-    code = call_mistral_api(messages)
-    
-    if "```c" in code:
-        code = code.split("```c")[1].split("```")[0].strip()
-    elif "```" in code:
-        code = code.split("```")[1].split("```")[0].strip()
-    return code
+# --- 2. 백엔드 핵심 기능 (파서 추가) ---
+def parse_code_from_response(response_text: str) -> str | None:
+    """LLM 응답에서 C 코드 블록을 추출합니다."""
+    match = re.search(r'```c\n(.*?)\n```', response_text, re.DOTALL)
+    if match:
+        return match.group(1).strip()
+    return None
 
 def compile_and_run_c_code(code: str) -> str:
     try:
@@ -53,70 +42,73 @@ def compile_and_run_c_code(code: str) -> str:
         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. 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.
-
-Instruction: '{instruction}'
-
-C Code to Analyze:
-```c
-{code}
-```"""
-    messages = [{"role": "user", "content": prompt}]
-    return call_mistral_api(messages)
-
-# --- 3. 메인 이벤트 핸들러 ---
-def handle_instruction(code: str, instruction: str):
+# --- 3. 진짜 '지능형' 에이전트 로직 ---
+def intelligent_agent(code: str, instruction: str):
     if not instruction:
-        return code, "Error: Instruction cannot be empty."
+        yield code, "Error: Instruction cannot be empty."
+        return
 
     lower_instruction = instruction.lower()
-
+    
+    # 1. 생성 (Generation)
     if "generate" in lower_instruction or "create" in lower_instruction or "만들어줘" in lower_instruction:
-        yield code, "Generating new code..."
-        new_code = generate_c_code(instruction)
-        yield new_code, "Code generation successful. New code is in the editor."
+        yield code, "1. Generating new code..."
+        prompt = f"You are a C programming expert. Generate a complete, compilable C code for this request: '{instruction}'. ONLY output the raw C code, without any markdown formatting or explanations."
+        new_code = call_mistral_api([{"role": "user", "content": prompt}])
+        yield new_code, "2. Code generation successful. The new code is now in the editor."
         return
 
     if not code:
-        yield code, "Error: Code editor is empty. Please provide code to work with, or ask to 'generate' new code."
+        yield code, "Error: Code editor is empty. Please provide code or ask to 'generate' it."
         return
 
+    # 2. 분석/리팩토링 (Analyze/Refactor)
+    if any(keyword in lower_instruction for keyword in ["analyze", "refactor", "explain", "comment", "review", "개선", "설명", "버그"]):
+        yield code, f"1. Analyzing code with instruction: '{instruction}'..."
+        prompt = f"You are a senior C code reviewer. Fulfill this instruction: '{instruction}'. If you refactor the code, YOU MUST provide the complete, improved code in a ```c code block. \n\nC Code to Analyze:\n```c\n{code}\n```"
+        analysis_result = call_mistral_api([{"role": "user", "content": prompt}])
+        
+        # 지능형 파싱: 리팩토링된 코드가 있는지 확인
+        refactored_code = parse_code_from_response(analysis_result)
+        if refactored_code:
+            yield refactored_code, f"2. Refactoring successful. Code editor updated.\n\n{analysis_result}"
+            # 'and compile' 같은 후속 명령이 있는지 확인
+            if "compile" in lower_instruction or "run" in lower_instruction:
+                yield refactored_code, "3. Compiling the new code..."
+                compile_result = compile_and_run_c_code(refactored_code)
+                yield refactored_code, f"4. Compilation finished.\n\n{compile_result}"
+        else:
+            # 리팩토링된 코드가 없으면, 분석 결과만 보여줌
+            yield code, f"2. Analysis complete.\n\n{analysis_result}"
+        return
+
+    # 3. 컴파일 및 실행 (Compile & Run)
     if "compile" in lower_instruction or "run" in lower_instruction:
-        yield code, "Compiling and running..."
+        yield code, "1. Compiling and running..."
         result = compile_and_run_c_code(code)
-        yield code, result
-    
-    else: 
-        yield code, f"Analyzing code with instruction: '{instruction}'..."
-        analysis_result = analyze_and_refactor_code(code, instruction)
-        yield code, analysis_result
+        yield code, f"2. Process finished.\n\n{result}"
+        return
+        
+    # 4. 그 외
+    yield code, "Error: Could not understand the instruction. Please be more specific (e.g., 'generate', 'compile', 'refactor')."
 
-# --- 4. Gradio UI 구성 ---
+# --- 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 C-Codestral IDE Agent (Library-Free)")
-
-    with gr.Tabs():
-        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', 'Generate a factorial function'...", lines=3)
-                    execute_btn = gr.Button("Execute Instruction", variant="primary")
-                    output_box = gr.Textbox(label="Output / Console", lines=15, interactive=False, show_copy_button=True)
-
-            execute_btn.click(fn=handle_instruction, 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"), 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())
+    gr.Markdown("# 💻 The True C-Codestral IDE Agent")
+    
+    with gr.Row():
+        with gr.Column(scale=2):
+            code_editor = gr.Code(label="C Code Editor", language="c", lines=25, interactive=True, value='#include <stdio.h>\n\nint main() {\n    printf("Hello, World!\\n");\n    return 0;\n}')
+        with gr.Column(scale=1):
+            instruction_box = gr.Textbox(label="Instruction", placeholder="e.g., 'Refactor this code and run it', 'Generate a factorial function', 'Find bugs'...", lines=4)
+            execute_btn = gr.Button("Execute", variant="primary", size="lg")
+            output_box = gr.Markdown(label="Console / Output", value="Welcome! Enter an instruction and press 'Execute'.")
+
+    execute_btn.click(
+        fn=intelligent_agent, 
+        inputs=[code_editor, instruction_box], 
+        outputs=[code_editor, output_box]
+    )
 
 if __name__ == "__main__":
     demo.queue().launch()