Update app.py

app.py

@@ -23,7 +23,7 @@ def call_mistral_api(messages):
     except requests.exceptions.RequestException as e:
         return f"API Call Error: {e}"
 
-# --- 2. 백엔드 핵심 기능 (파서 추가) ---
+# --- 2. 백엔드 핵심 기능 함수 ---
 def parse_code_from_response(response_text: str) -> str | None:
     """LLM 응답에서 C 코드 블록을 추출합니다."""
     match = re.search(r'```c\n(.*?)\n```', response_text, re.DOTALL)
@@ -31,6 +31,10 @@ def parse_code_from_response(response_text: str) -> str | None:
         return match.group(1).strip()
     return None
 
+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."
+    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)
@@ -42,73 +46,55 @@ 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)}"
 
-# --- 3. 진짜 '지능형' 에이전트 로직 ---
-def intelligent_agent(code: str, instruction: str):
-    if not instruction:
-        yield code, "Error: Instruction cannot be empty."
-        return
-
-    lower_instruction = instruction.lower()
+def analyze_and_refactor_code(code: str, instruction: str):
+    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}])
     
-    # 1. 생성 (Generation)
-    if "generate" in lower_instruction or "create" in lower_instruction or "만들어줘" in lower_instruction:
-        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 or ask to 'generate' it."
-        return
+    # 지능형 파싱: 리팩토링된 코드가 있는지 확인
+    refactored_code = parse_code_from_response(analysis_result)
+    
+    if refactored_code:
+        # 리팩토링된 코드가 있으면, 그 코드를 에디터에, 전체 응답을 결과창에 반환
+        return refactored_code, analysis_result
+    else:
+        # 리팩토링된 코드가 없으면, 원래 코드는 그대로 두고, 분석 결과만 반환
+        return code, analysis_result
 
-    # 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. 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 (Stable Tabbed Version)")
 
-    # 3. 컴파일 및 실행 (Compile & Run)
-    if "compile" in lower_instruction or "run" in lower_instruction:
-        yield code, "1. Compiling and running..."
-        result = compile_and_run_c_code(code)
-        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')."
+    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=20, interactive=True, value='#include <stdio.h>\n\nint main() {\n    printf("Hello, World!\\n");\n    return 0;\n}')
+                
+                with gr.Column(scale=1):
+                    # 명령어 탭 구성
+                    with gr.Tabs():
+                        with gr.TabItem("Generate"):
+                            gen_instr = gr.Textbox(label="Generation Instruction", placeholder="e.g., 'a program that calculates factorial'")
+                            gen_btn = gr.Button("Generate Code", variant="primary")
+                        with gr.TabItem("Analyze / Refactor"):
+                            ana_instr = gr.Textbox(label="Analysis Instruction", placeholder="e.g., 'add comments', 'refactor for efficiency'")
+                            ana_btn = gr.Button("Analyze / Refactor", variant="primary")
+                        with gr.TabItem("Compile / Run"):
+                            run_btn = gr.Button("Compile and Run", variant="stop")
 
-# --- 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")
-    
-    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'.")
+            output_box = gr.Textbox(label="Output / Console", lines=8, interactive=False, show_copy_button=True)
 
-    execute_btn.click(
-        fn=intelligent_agent, 
-        inputs=[code_editor, instruction_box], 
-        outputs=[code_editor, output_box]
-    )
+            # 이벤트 핸들러 연결
+            gen_btn.click(fn=generate_c_code, inputs=[gen_instr], outputs=[code_editor])
+            ana_btn.click(fn=analyze_and_refactor_code, inputs=[code_editor, ana_instr], outputs=[code_editor, output_box])
+            run_btn.click(fn=compile_and_run_c_code, inputs=[code_editor], outputs=[output_box])
+            
+        with gr.TabItem("🛠️ MCP Tools"):
+            gr.Markdown("## Available MCP Tools for other Agents")
+            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"))
 
 if __name__ == "__main__":
     demo.queue().launch()