Update app.py

app.py

@@ -51,28 +51,34 @@ C Code to Analyze:
     response = CLIENT.chat(model="codestral-latest", messages=messages)
     return response.choices[0].message.content
 
-# --- 3. 메인 에이전트 로직 ---
-def agent_logic(code: str, instruction: str):
-    if not instruction: return "Error: Instruction cannot be empty."
-    
+# --- 3. 메인 이벤트 핸들러 ---
+def handle_instruction(code: str, instruction: str):
+    if not instruction:
+        return code, "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:
-        # 새로운 코드를 생성하고, 그 결과를 코드 에디터에 직접 반환
+    # 'generate' 또는 'create'가 포함된 경우, 코드 에디터가 비어있어도 무조건 생성
+    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)
-        # 출력은 코드 에디터로, 결과창에는 성공 메시지
-        return new_code, f"Code generation successful. The new code is now in the editor."
+        yield new_code, "Code generation successful. New code is 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."
+        return
 
+    if "compile" in lower_instruction or "run" in lower_instruction:
+        yield code, "Compiling and running..."
+        result = compile_and_run_c_code(code)
+        yield code, result
+    
     else: # Refactor, analyze, add comments, etc.
-        if not code: return "Error: Code editor is empty. Nothing to analyze."
-        # 분석/리팩토링 결과를 결과창에 반환
+        yield code, f"Analyzing code with instruction: '{instruction}'..."
         analysis_result = analyze_and_refactor_code(code, instruction)
-        # 코드 에디터는 그대로 두고, 결과창만 업데이트
-        return code, analysis_result
+        yield 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:
@@ -84,22 +90,19 @@ with gr.Blocks(theme=gr.themes.Monochrome(primary_hue="indigo", secondary_hue="b
                 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'...")
+                    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.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
+                    output_box = gr.Textbox(label="Output / Console", lines=15, interactive=False, show_copy_button=True)
 
-            execute_btn.click(handle_ide_submit, inputs=[code_editor, instruction_box], outputs=[code_editor, output_box])
+            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.")
@@ -111,4 +114,4 @@ with gr.Blocks(theme=gr.themes.Monochrome(primary_hue="indigo", secondary_hue="b
                 gr.Interface(fn=analyze_and_refactor_code, inputs=[gr.Code(language="c"), "text"], outputs=gr.Markdown())
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.queue().launch()