# /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:
print(f"Error initializing Mistral Client: {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:
"""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)
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 = """
C-Codestral Agent
"""
# --- 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}