Update app.py

app.py

@@ -4,24 +4,28 @@ import gradio as gr
 import os
 import subprocess
 from langchain_mistralai.chat_models import ChatMistralAI
-from langchain_core.messages import HumanMessage, SystemMessage, AIMessage
+from langchain_core.messages import HumanMessage, AIMessage
 from langchain.agents import AgentExecutor, create_tool_calling_agent
 from langchain_core.prompts import ChatPromptTemplate
 from langchain.tools import tool
 
 # --- 1. 환경 설정 및 LLM 초기화 ---
 API_KEY = os.environ.get("MISTRAL_API_KEY", None)
-if API_KEY is None:
-    print("FATAL: MISTRAL_API_KEY is not set in Space Secrets.")
-    # API 키가 없으면 사용자에게 경고 메시지를 표시하는 UI를 로드
-    llm = None
+llm = None
+if API_KEY:
+    try:
+        llm = ChatMistralAI(model="codestral-latest", temperature=0, api_key=API_KEY, streaming=True)
+    except Exception as e:
+        print(f"Failed to initialize LLM: {e}")
 else:
-    llm = ChatMistralAI(model="codestral-latest", temperature=0, api_key=API_KEY, streaming=True)
+    print("FATAL: MISTRAL_API_KEY is not set in Space Secrets.")
 
-# --- 2. 핵심 도구(Tools) 정의: @tool 데코레이터 사용 ---
+# --- 2. 핵심 도구(Tools) 정의 ---
+# LangChain 에이전트가 내부적으로 사용할 @tool 데코레이터
 @tool
 def generate_c_code(description: str) -> str:
     """Generates complete, compilable C code based on a natural language description. Use this when the user wants to create new code."""
+    if not llm: return "Error: LLM not initialized."
     code_generation_prompt = f"""You are a world-class C programming expert.
 Generate a single, complete, and compilable C code file based on this request: '{description}'.
 The code must be clean, efficient, and include necessary headers.
@@ -38,26 +42,21 @@ 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
+            ["./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 subprocess.TimeoutExpired:
         return "--- ERROR ---\nProcess timed out. Check for infinite loops."
     except Exception as e:
@@ -66,11 +65,10 @@ def compile_and_run_c_code(code: str) -> str:
 @tool
 def refactor_or_analyze_c_code(code: str, request: str) -> str:
     """Analyzes, refactors, or explains a given C code based on a user's specific request. Use this for code improvement or understanding."""
+    if not llm: return "Error: LLM not initialized."
     analysis_prompt = f"""You are a senior C code reviewer. Analyze the following C code based on the user's request.
 Provide a clear, concise, and helpful response. If refactoring, provide the complete improved code in a C code block.
-
 User's Request: '{request}'
-
 C Code to Analyze:
 ```c
 {code}
@@ -78,82 +76,66 @@ C Code to Analyze:
     response = llm.invoke([HumanMessage(content=analysis_prompt)])
     return response.content
 
-# --- 3. LangChain 에이전트 설정 ---
-tools = [generate_c_code, compile_and_run_c_code, refactor_or_analyze_c_code]
+# --- 3. MCP 도구 API 생성 ---
+# 각 도구를 gr.Interface를 사용해 API로 만듭니다. 이것이 외부 MCP 클라이언트에 노출됩니다.
+generate_api = gr.Interface(fn=generate_c_code, inputs="text", outputs="code", title="C Code Generator", description="Generates C code from a description.")
+compile_api = gr.Interface(fn=compile_and_run_c_code, inputs="code", outputs="text", title="C Code Compiler & Runner", description="Compiles and runs C code.")
+refactor_api = gr.Interface(fn=refactor_or_analyze_c_code, inputs=["code", "text"], outputs="markdown", title="C Code Refactor & Analyzer", description="Refactors or analyzes C code.")
 
-agent_prompt = ChatPromptTemplate.from_messages(
-    [
+# --- 4. LangChain 에이전트 설정 ---
+tools = [generate_c_code, compile_and_run_c_code, refactor_or_analyze_c_code]
+agent_executor = None
+if llm:
+    agent_prompt = ChatPromptTemplate.from_messages([
         ("system", "You are a powerful C-language assistant agent. You can generate, compile, and analyze C code. Think step-by-step and use your tools to fulfill the user's request. If you generate code, you should almost always offer to compile and run it for the user as the next step."),
         ("placeholder", "{chat_history}"),
         ("human", "{input}"),
         ("placeholder", "{agent_scratchpad}"),
-    ]
-)
-
-if llm:
+    ])
     agent = create_tool_calling_agent(llm, tools, agent_prompt)
-    agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True)
-else:
-    agent_executor = None
+    agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True, handle_parsing_errors=True)
 
 def agent_chat(user_input, history):
     if agent_executor is None:
         yield "Error: MISTRAL_API_KEY is not configured. Please set it in the Space secrets."
         return
-
     chat_history = []
     for human, ai in history:
         chat_history.append(HumanMessage(content=human))
         chat_history.append(AIMessage(content=ai))
-    
-    response_stream = agent_executor.stream({
-        "input": user_input,
-        "chat_history": chat_history
-    })
-    
+    response_stream = agent_executor.stream({"input": user_input, "chat_history": chat_history})
     full_response = ""
     for chunk in response_stream:
         if "output" in chunk:
             full_response += chunk["output"]
             yield full_response
 
-# --- 4. Gradio UI/UX 디자인 ---
+# --- 5. Gradio UI/UX 디자인 ---
 with gr.Blocks(theme=gr.themes.Monochrome(), css=".gradio-container{max-width: 800px !important; margin: auto;}") as demo:
     gr.Markdown("# 🚀 C-Codestral Agent")
-    
-    chatbot = gr.Chatbot(
-        label="C-Agent",
-        bubble_full_width=False,
-        height=600,
-        avatar_images=(None, "https://huggingface.co/datasets/huggingface/brand-assets/resolve/main/hf-logo-with-title.png")
-    )
-    
-    with gr.Row():
-        txt = gr.Textbox(
-            show_label=False,
-            placeholder="Enter your request and press enter...",
-            container=False,
-            scale=7
-        )
-        submit_btn = gr.Button("Submit", variant="primary", scale=1)
-
-    txt.submit(agent_chat, [txt, chatbot], chatbot)
-    submit_btn.click(agent_chat, [txt, chatbot], chatbot)
-    txt.submit(lambda: "", None, txt) # Clear textbox on submit
-    submit_btn.click(lambda: "", None, txt) # Clear textbox on submit
-
-    gr.Examples(
-        [
-            "Generate a C program to calculate the factorial of a number.",
-            "Now, compile and run the code you just created.",
-            "Refactor the factorial code to use recursion instead of a loop.",
-        ],
-        inputs=txt,
-        label="Example Prompts"
-    )
-
-    # --- 5. MCP 도구 노출 ---
-    gr.load_tools(tools)
+    with gr.Tabs():
+        with gr.TabItem("🤖 Agent Chat"):
+            chatbot = gr.Chatbot(label="C-Agent", bubble_full_width=False, height=600, avatar_images=(None, "https://huggingface.co/datasets/huggingface/brand-assets/resolve/main/hf-logo-with-title.png"))
+            with gr.Row():
+                txt = gr.Textbox(show_label=False, placeholder="Enter your request and press enter...", container=False, scale=7)
+                submit_btn = gr.Button("Submit", variant="primary", scale=1)
+            txt.submit(agent_chat, [txt, chatbot], chatbot)
+            submit_btn.click(agent_chat, [txt, chatbot], chatbot)
+            txt.submit(lambda: "", None, txt)
+            submit_btn.click(lambda: "", None, txt)
+            gr.Examples(
+                ["Generate a C program to calculate the factorial of a number.", "Now, compile and run the code you just created.", "Refactor the factorial code to use recursion instead of a loop."],
+                inputs=txt, label="Example Prompts"
+            )
+        # MCP 도구들을 별도의 탭에 API로 노출
+        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):
+                generate_api.render()
+            with gr.Accordion("Tool: Compile & Run C Code", open=False):
+                compile_api.render()
+            with gr.Accordion("Tool: Refactor & Analyze C Code", open=False):
+                refactor_api.render()
 
 if __name__ == "__main__":
     demo.queue().launch(debug=True)