c

Running

App Files Files Community

kimhyunwoo commited on Jun 8

Commit

6a21a24

verified ·

1 Parent(s): 07d3ed8

Update app.py

Browse files

Files changed (1) hide show

app.py +138 -81

app.py CHANGED Viewed

@@ -1,16 +1,19 @@
 # /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:
     try:
-        # NOTE: requirements.txt에 mistralai==0.4.2 버전을 명시해야 합니다.
         CLIENT = MistralClient(api_key=API_KEY)
         print("Mistral API Client initialized successfully.")
     except Exception as e:
@@ -18,27 +21,18 @@ if API_KEY:
 else:
     print("FATAL: MISTRAL_API_KEY is not set in Space Secrets.")
-# --- 2. 핵심 기능 함수 (MCP 도구로 사용될 함수들) ---
 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. Your task is to generate a complete, compilable C code based on the user's request.
-The code should be clean, efficient, and well-commented.
-Please only output the raw C code inside a single C code block and nothing else.
-User's request: {description}"""
     messages = [{"role": "user", "content": prompt}]
     response = CLIENT.chat(model="codestral-latest", messages=messages)
     code = response.choices[0].message.content
-    # 코드 블록에서 순수 코드만 추출
     if "```c" in code:
         code = code.split("```c")[1].split("```")[0].strip()
     elif "```" in code:
         code = code.split("```")[1].split("```")[0].strip()
     return code
 def compile_and_run_c_code(code: str) -> str:
@@ -46,99 +40,162 @@ 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
         )
         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:
         return f"--- SYSTEM ERROR ---\nAn unexpected error occurred: {str(e)}"
-# --- 3. Gradio 챗봇 로직 (Agent 역할) ---
-def agent_chat(message, history):
-    if not CLIENT:
-        yield "MISTRAL_API_KEY를 Space Secrets에 설정해주세요."
-        return
-    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})
     lower_message = message.lower()
     if "compile" in lower_message or "run" in lower_message or "실행" in lower_message:
         code_to_run = ""
-        # 대화 기록에서 마지막 C 코드 블록 찾기
-        for h_user, h_ai in reversed(history + [[message, ""]]):
-            content_to_check = h_ai or h_user
-            if "```c" in content_to_check:
-                code_to_run = content_to_check.split("```c")[1].split("```")[0].strip()
                 break
         if not code_to_run:
-            yield "컴파일할 코드를 찾을 수 없습니다. 먼저 코드를 생성하거나, 메시지에 코드를 포함시켜 주세요."
-            return
-        yield f"코드를 컴파일하고 실행합니다...\n\n```c\n{code_to_run}\n```"
-        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"코드가 생성되었습니다. 'run this code'라고 입력���여 실행할 수 있습니다.\n\n```c\n{generated_code}\n```"
     else:
-        # 일반적인 질문은 LLM에게 직접 전달
-        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(), css=".gradio-container{max-width: 800px !important; margin: auto;}") 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...", label="Your Request", scale=4)
-            def clear_and_submit(message, history):
-                # 메시지 전송 후 입력창을 비우는 효과를 위해 래퍼 함수 사용
-                yield from agent_chat(message, history)
-            msg.submit(clear_and_submit, [msg, chatbot], chatbot)
-            msg.submit(lambda: "", None, msg) # 입력창 클리어
-            gr.Examples(
-                ["Generate a C program to calculate the factorial of 10.", "run the code"],
-                inputs=msg
-            )
-        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):
-                # 최신 Gradio에서는 language='c'를 지원합니다.
-                gr.Interface(fn=generate_c_code, inputs="text", outputs=gr.Code(language="c"), allow_flagging="never")
-            with gr.Accordion("Tool: Compile & Run C Code", open=False):
-                gr.Interface(fn=compile_and_run_c_code, inputs=gr.Code(language="c"), outputs="text", allow_flagging="never")
-if __name__ == "__main__":
-    demo.queue().launch()

 # /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:
     try:
         CLIENT = MistralClient(api_key=API_KEY)
         print("Mistral API Client initialized successfully.")
     except Exception as e:
 else:
     print("FATAL: MISTRAL_API_KEY is not set in Space Secrets.")
+# --- 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}]
     response = CLIENT.chat(model="codestral-latest", messages=messages)
     code = response.choices[0].message.content
     if "```c" in code:
         code = code.split("```c")[1].split("```")[0].strip()
     elif "```" in code:
         code = code.split("```")[1].split("```")[0].strip()
     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)
         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
         )
         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 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}