c

Running

App Files Files Community

kimhyunwoo commited on Jun 8

Commit

e0c0af8

verified ·

1 Parent(s): 963f5f3

Update app.py

Browse files

Files changed (1) hide show

app.py +54 -138

app.py CHANGED Viewed

@@ -1,15 +1,11 @@
 # /app.py
 import os
 import subprocess
-from fastapi import FastAPI, Request
-from fastapi.responses import HTMLResponse
-from pydantic import BaseModel
 from mistralai.client import MistralClient
-# --- 1. FastAPI 앱 및 Mistral 클라이언트 초기화 ---
-app = FastAPI()
 API_KEY = os.environ.get("MISTRAL_API_KEY")
 CLIENT = None
 if API_KEY:
@@ -23,7 +19,6 @@ else:
 # --- 2. 백엔드 핵심 기능 함수 ---
 def generate_c_code(description: str) -> str:
-    """Generates C code based on a natural language description."""
     if not CLIENT: return "Error: Mistral API key not configured."
     prompt = f"You are a C programming expert. Generate a complete, compilable C code for this request: '{description}'. ONLY output the raw C code within a single ```c code block."
     messages = [{"role": "user", "content": prompt}]
@@ -36,7 +31,6 @@ def generate_c_code(description: str) -> str:
     return code
 def compile_and_run_c_code(code: str) -> str:
-    """Compiles and runs a given C code snippet and returns its output or any errors."""
     try:
         with open("main.c", "w", encoding='utf-8') as f:
             f.write(code)
@@ -58,144 +52,66 @@ def compile_and_run_c_code(code: str) -> str:
     except Exception as e:
         return f"--- SYSTEM ERROR ---\nAn unexpected error occurred: {str(e)}"
-# --- 3. HTML, CSS, JavaScript 프론트엔드 ---
-HTML_TEMPLATE = """
-<!DOCTYPE html>
-<html lang="en">
-<head>
-    <meta charset="UTF-8">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>C-Codestral Agent</title>
-    <style>
-        body { font-family: sans-serif; background-color: #f5f5f5; display: flex; justify-content: center; align-items: center; height: 100vh; margin: 0; }
-        #app-container { width: 100%; max-width: 800px; height: 90vh; background: white; border-radius: 8px; box-shadow: 0 0 20px rgba(0,0,0,0.1); display: flex; flex-direction: column; }
-        h1 { text-align: center; color: #333; padding: 20px; margin: 0; border-bottom: 1px solid #eee; }
-        #chat-box { flex-grow: 1; padding: 20px; overflow-y: auto; }
-        .message { margin-bottom: 15px; }
-        .user-msg { text-align: right; }
-        .bot-msg { text-align: left; }
-        .msg-bubble { display: inline-block; padding: 10px 15px; border-radius: 18px; max-width: 80%; }
-        .user-msg .msg-bubble { background-color: #007bff; color: white; }
-        .bot-msg .msg-bubble { background-color: #e9e9eb; color: black; }
-        pre { background-color: #2d2d2d; color: #f8f8f2; padding: 15px; border-radius: 5px; white-space: pre-wrap; word-wrap: break-word; }
-        #input-area { display: flex; padding: 20px; border-top: 1px solid #eee; }
-        #user-input { flex-grow: 1; padding: 10px; border: 1px solid #ccc; border-radius: 20px; outline: none; }
-        #send-btn { background: #007bff; color: white; border: none; padding: 10px 20px; margin-left: 10px; border-radius: 20px; cursor: pointer; }
-    </style>
-</head>
-<body>
-    <div id="app-container">
-        <h1>🚀 C-Codestral Agent</h1>
-        <div id="chat-box"></div>
-        <div id="input-area">
-            <input type="text" id="user-input" placeholder="Ask me to generate or run C code...">
-            <button id="send-btn">Send</button>
-        </div>
-    </div>
-    <script>
-        const chatBox = document.getElementById('chat-box');
-        const userInput = document.getElementById('user-input');
-        const sendBtn = document.getElementById('send-btn');
-        let chatHistory = [];
-        function addMessage(sender, text) {
-            const msgDiv = document.createElement('div');
-            msgDiv.classList.add('message', sender === 'user' ? 'user-msg' : 'bot-msg');
-            const bubble = document.createElement('div');
-            bubble.classList.add('msg-bubble');
-            // Convert ```c ... ``` to <pre><code>...</code>
-            if (text.includes('```')) {
-                const parts = text.split(/```c|```/);
-                parts.forEach((part, index) => {
-                    if (index % 2 === 1) { // Code block
-                        const pre = document.createElement('pre');
-                        const code = document.createElement('code');
-                        code.textContent = part.trim();
-                        pre.appendChild(code);
-                        bubble.appendChild(pre);
-                    } else { // Regular text
-                        const textNode = document.createTextNode(part);
-                        bubble.appendChild(textNode);
-                    }
-                });
-            } else {
-                bubble.textContent = text;
-            }
-            msgDiv.appendChild(bubble);
-            chatBox.appendChild(msgDiv);
-            chatBox.scrollTop = chatBox.scrollHeight;
-        }
-        async function sendMessage() {
-            const message = userInput.value.trim();
-            if (!message) return;
-            addMessage('user', message);
-            chatHistory.push({ "role": "user", "content": message });
-            userInput.value = '';
-            const response = await fetch('/chat', {
-                method: 'POST',
-                headers: { 'Content-Type': 'application/json' },
-                body: JSON.stringify({ message: message, history: chatHistory })
-            });
-            const data = await response.json();
-            const botResponse = data.response;
-            addMessage('bot', botResponse);
-            chatHistory.push({ "role": "assistant", "content": botResponse });
-        }
-        sendBtn.addEventListener('click', sendMessage);
-        userInput.addEventListener('keypress', (e) => {
-            if (e.key === 'Enter') sendMessage();
-        });
-    </script>
-</body>
-</html>
-"""
-# --- 4. FastAPI 엔드포인트 ---
-class ChatRequest(BaseModel):
-    message: str
-    history: list
-@app.get("/", response_class=HTMLResponse)
-async def get_home():
-    return HTML_TEMPLATE
-@app.post("/chat")
-async def handle_chat(request: ChatRequest):
-    message = request.message
-    history = request.history
     lower_message = message.lower()
-    response_text = ""
     if "compile" in lower_message or "run" in lower_message or "실행" in lower_message:
         code_to_run = ""
-        for item in reversed(history):
-            if item['role'] == 'assistant' and "```c" in item['content']:
-                code_to_run = item['content'].split("```c")[1].split("```")[0].strip()
                 break
         if not code_to_run:
-            response_text = "컴파일할 코드를 찾을 수 없습니다. 먼저 코드를 생성해주세요."
-        else:
-            response_text = compile_and_run_c_code(code_to_run)
     elif "generate" in lower_message or "create" in lower_message or "만들어줘" in lower_message or "짜줘" in lower_message:
         generated_code = generate_c_code(message)
-        response_text = f"코드가 생성되었습니다. 'run this code'라고 입력하여 실행할 수 있습니다.\n\n```c\n{generated_code}\n```"
     else:
-        if not CLIENT:
-            response_text = "MISTRAL_API_KEY가 설정되지 않았습니다."
-        else:
-            messages = history + [{"role": "user", "content": message}]
-            response = CLIENT.chat(model="codestral-latest", messages=messages)
-            response_text = response.choices[0].message.content
-    return {"response": response_text}

 # /app.py
+import gradio as gr
 import os
 import subprocess
 from mistralai.client import MistralClient
+# --- 1. 환경 설정 및 클라이언트 초기화 ---
 API_KEY = os.environ.get("MISTRAL_API_KEY")
 CLIENT = None
 if API_KEY:
 # --- 2. 백엔드 핵심 기능 함수 ---
 def generate_c_code(description: str) -> str:
     if not CLIENT: return "Error: Mistral API key not configured."
     prompt = f"You are a C programming expert. Generate a complete, compilable C code for this request: '{description}'. ONLY output the raw C code within a single ```c code block."
     messages = [{"role": "user", "content": prompt}]
     return code
 def compile_and_run_c_code(code: str) -> str:
     try:
         with open("main.c", "w", encoding='utf-8') as f:
             f.write(code)
     except Exception as e:
         return f"--- SYSTEM ERROR ---\nAn unexpected error occurred: {str(e)}"
+# --- 3. Gradio 챗봇 로직 ---
+def agent_chat(message, history):
+    if not CLIENT:
+        yield "MISTRAL_API_KEY를 Space Secrets에 설정해주세요."
+        return
     lower_message = message.lower()
     if "compile" in lower_message or "run" in lower_message or "실행" in lower_message:
         code_to_run = ""
+        for h_user, h_ai in reversed(history):
+            if "```c" in h_ai:
+                code_to_run = h_ai.split("```c")[1].split("```")[0].strip()
                 break
         if not code_to_run:
+            yield "컴파일할 코드를 찾을 수 없습니다. 먼저 코드를 생성해주세요."
+            return
+        yield f"```c\n{code_to_run}\n```\n\n--- COMPILING & RUNNING ---"
+        result = compile_and_run_c_code(code_to_run)
+        yield result
     elif "generate" in lower_message or "create" in lower_message or "만들어줘" in lower_message or "짜줘" in lower_message:
+        yield "C 코드를 생성하는 중..."
         generated_code = generate_c_code(message)
+        yield f"코드가 생성되었습니다.\n\n```c\n{generated_code}\n```"
     else:
+        history_messages = []
+        for human, ai in history:
+            history_messages.append({"role": "user", "content": human})
+            history_messages.append({"role": "assistant", "content": ai})
+        history_messages.append({"role": "user", "content": message})
+        response = CLIENT.chat(model="codestral-latest", messages=history_messages)
+        yield response.choices[0].message.content
+# --- 4. Gradio UI 구성 ---
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="green", secondary_hue="gray"), css="footer {visibility: hidden}") as demo:
+    gr.Markdown("# 🚀 C-Codestral Agent")
+    with gr.Tabs():
+        with gr.TabItem("🤖 Agent Chat"):
+            chatbot = gr.Chatbot(label="C-Agent", height=600, bubble_full_width=False, render_markdown=True)
+            msg = gr.Textbox(placeholder="Ask me to generate or run C code...", show_label=False, container=False, scale=10)
+            btn = gr.Button("Send", scale=1)
+            btn.click(agent_chat, [msg, chatbot], chatbot)
+            msg.submit(agent_chat, [msg, chatbot], chatbot)
+            # Clear input after submit
+            clear_event = btn.click(lambda: None, None, msg, queue=False)
+            clear_event2 = msg.submit(lambda: None, None, msg, queue=False)
+        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", label="C Code to Run"), outputs=gr.Textbox(label="Output/Error"))
+if __name__ == "__main__":
+    demo.queue().launch()