Update app.py

app.py

@@ -3,20 +3,22 @@ import os
 import subprocess
 from mistralai.client import MistralClient
 
-# --- 1. 환경 설정 ---
+# --- 1. 환경 설정 (Codestral 전용 엔드포인트 지정) ---
 API_KEY = os.environ.get("MISTRAL_API_KEY")
 CLIENT = None
+CODESTRAL_ENDPOINT = "https://codestral.mistral.ai/v1" # <--- 이 부분이 핵심입니다.
+
 if API_KEY:
     try:
-        # requirements.txt에 mistralai==0.4.2 버전을 명시하여 경고를 제거합니다.
-        CLIENT = MistralClient(api_key=API_KEY)
-        print("Mistral API Client initialized successfully.")
+        # Codestral 전용 엔드포인트를 명시적으로 지정합니다.
+        CLIENT = MistralClient(api_key=API_KEY, endpoint=CODESTRAL_ENDPOINT)
+        print("Mistral API Client for Codestral 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. 백엔드 핵심 기능 ---
+# --- 2. 백엔드 핵심 기능 (변경 없음) ---
 def generate_c_code(description: str) -> str:
     if not CLIENT: return "Error: Mistral API key not configured."
     prompt = f"Generate a complete, compilable C code for this request: '{description}'. ONLY output the raw C code within a single ```c code block."
@@ -51,22 +53,25 @@ def compile_and_run_c_code(code: str) -> str:
     except Exception as e:
         return f"--- SYSTEM ERROR ---\nAn unexpected error occurred: {str(e)}"
 
-# --- 3. Gradio 챗봇 로직 (type='messages' 완벽 호환) ---
-def agent_chat(message: str, history: list[dict[str, str]]):
-    # 사용자 메시지를 history에 추가
-    history.append({"role": "user", "content": message})
-    
+# --- 3. Gradio 챗봇 로직 (변경 없음) ---
+def agent_chat(message: str, history: list[list[str | None]]):
+    if not CLIENT:
+        return "MISTRAL_API_KEY를 Space Secrets에 설정해주세요."
+
+    history_for_api = []
+    for human, assistant in history:
+        if human: history_for_api.append({"role": "user", "content": human})
+        if assistant: history_for_api.append({"role": "assistant", "content": assistant})
+    history_for_api.append({"role": "user", "content": message})
+
     lower_message = message.lower()
     bot_response = ""
 
-    if not CLIENT:
-        bot_response = "MISTRAL_API_KEY가 Space Secrets에 설정되지 않았습니다."
-    elif "compile" in lower_message or "run" in lower_message or "실행" in lower_message:
+    if "compile" in lower_message or "run" in lower_message or "실행" in lower_message:
         code_to_run = ""
-        # history (list of dicts)에서 코드 찾기
-        for item in reversed(history):
-            if item.get("role") == "assistant" and "```c" in item.get("content", ""):
-                code_to_run = item["content"].split("```c")[1].split("```")[0].strip()
+        for h_user, h_ai in reversed(history):
+            if h_ai and "```c" in h_ai:
+                code_to_run = h_ai.split("```c")[1].split("```")[0].strip()
                 break
         if not code_to_run:
             bot_response = "컴파일할 코드를 찾을 수 없습니다. 먼저 코드를 생성해주세요."
@@ -78,36 +83,32 @@ def agent_chat(message: str, history: list[dict[str, str]]):
         bot_response = f"코드가 생성되었습니다.\n\n```c\n{generated_code}\n```"
         
     else:
-        # Mistral API에는 전체 대화 기록을 전달
-        response = CLIENT.chat(model="codestral-latest", messages=history)
+        response = CLIENT.chat(model="codestral-latest", messages=history_for_api)
         bot_response = response.choices[0].message.content
 
-    # 봇 응답을 history에 추가
-    history.append({"role": "assistant", "content": bot_response})
-    
-    # 최종 업데이트된 history를 반환
+    history.append([message, bot_response])
     return history
 
-# --- 4. Gradio UI 구성 (type='messages' 명시) ---
+# --- 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"):
-            # 에러의 원인이었던 type을 'messages'로 명시적으로 지정
-            chatbot = gr.Chatbot(label="C-Agent", height=600, render_markdown=True, type="messages")
-            
+            chatbot = gr.Chatbot(label="C-Agent", height=600, render_markdown=True)
             with gr.Row():
                 msg = gr.Textbox(placeholder="Ask me to generate or run C code...", show_label=False, container=False, scale=10)
                 btn = gr.Button("Send", variant="primary", scale=1)
 
-            # 이벤트 리스너: 함수가 업데이트된 history 전체를 반환하고, 챗봇이 이를 받음
-            def handle_submit(message, history):
-                new_history = agent_chat(message, history)
-                return new_history, "" # 업데이트된 history와 빈 문��열을 반환하여 입력창 초기화
+            def user(user_message, history):
+                return "", history + [[user_message, None]]
 
-            btn.click(handle_submit, [msg, chatbot], [chatbot, msg])
-            msg.submit(handle_submit, [msg, chatbot], [chatbot, msg])
+            btn.click(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+                agent_chat, chatbot, chatbot
+            )
+            msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+                agent_chat, chatbot, chatbot
+            )
 
         with gr.TabItem("🛠️ MCP Tools API"):
             gr.Markdown("## Available MCP Tools\nThese APIs can be used by any MCP-compliant client.")