Update app.py

app.py

@@ -1,29 +1,46 @@
 import gradio as gr
 import os
 import subprocess
-from mistralai.client import MistralClient
+import requests  # mistralai 대신 requests 사용
+import json
 
 # --- 1. 환경 설정 ---
 API_KEY = os.environ.get("MISTRAL_API_KEY")
-CLIENT = None
-CODESTRAL_ENDPOINT = "https://codestral.mistral.ai/v1"
+CODESTRAL_ENDPOINT = "https://codestral.mistral.ai/v1/chat/completions" # 직접 호출할 엔드포인트
+HEADERS = {
+    "Authorization": f"Bearer {API_KEY}",
+    "Content-Type": "application/json"
+}
 
-if API_KEY:
-    try:
-        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:
+if not API_KEY:
     print("FATAL: MISTRAL_API_KEY is not set in Space Secrets.")
 
-# --- 2. 백엔드 핵심 기능 함수 ---
+# --- 2. 백엔드 핵심 기능 함수 (requests 사용) ---
+def call_mistral_api(messages):
+    if not API_KEY: return "Error: MISTRAL_API_KEY is not configured."
+    
+    data = {
+        "model": "codestral-latest",
+        "messages": messages
+    }
+    
+    try:
+        response = requests.post(CODESTRAL_ENDPOINT, headers=HEADERS, data=json.dumps(data), timeout=30)
+        response.raise_for_status()  # 2xx가 아닌 상태 코드에 대해 에러 발생
+        return response.json()["choices"][0]["message"]["content"]
+    except requests.exceptions.RequestException as e:
+        return f"API Call Error: {e}"
+
 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, without any markdown formatting or explanations."
     messages = [{"role": "user", "content": prompt}]
-    response = CLIENT.chat(model="codestral-latest", messages=messages)
-    return response.choices[0].message.content.strip()
+    code = call_mistral_api(messages)
+    
+    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:
@@ -37,7 +54,6 @@ def compile_and_run_c_code(code: str) -> str:
     except Exception as e: return f"--- SYSTEM ERROR ---\nAn unexpected error occurred: {str(e)}"
 
 def analyze_and_refactor_code(code: str, instruction: str) -> str:
-    if not CLIENT: return "Error: Mistral API key not configured."
     prompt = f"""You are a senior C code reviewer. Analyze the following C code based on the user's instruction.
 Provide a clear, concise response. If refactoring, provide the complete improved code in a C code block.
 
@@ -48,8 +64,7 @@ C Code to Analyze:
 {code}
 ```"""
     messages = [{"role": "user", "content": prompt}]
-    response = CLIENT.chat(model="codestral-latest", messages=messages)
-    return response.choices[0].message.content
+    return call_mistral_api(messages)
 
 # --- 3. 메인 이벤트 핸들러 ---
 def handle_instruction(code: str, instruction: str):
@@ -58,14 +73,12 @@ def handle_instruction(code: str, instruction: str):
 
     lower_instruction = instruction.lower()
 
-    # 'generate' 또는 'create'가 포함된 경우, 코드 에디터가 비어있어도 무조건 생성
     if "generate" in lower_instruction or "create" in lower_instruction or "만들어줘" in lower_instruction:
         yield code, "Generating new code..."
         new_code = generate_c_code(instruction)
         yield new_code, "Code generation successful. New code is in the editor."
         return
 
-    # 그 외의 경우, 코드 에디터가 비어있으면 에러
     if not code:
         yield code, "Error: Code editor is empty. Please provide code to work with, or ask to 'generate' new code."
         return
@@ -75,14 +88,14 @@ def handle_instruction(code: str, instruction: str):
         result = compile_and_run_c_code(code)
         yield code, result
     
-    else: # Refactor, analyze, add comments, etc.
+    else: 
         yield code, f"Analyzing code with instruction: '{instruction}'..."
         analysis_result = analyze_and_refactor_code(code, instruction)
         yield code, analysis_result
 
-# --- 4. Gradio UI 구성 (IDE 형태) ---
+# --- 4. Gradio UI 구성 ---
 with gr.Blocks(theme=gr.themes.Monochrome(primary_hue="indigo", secondary_hue="blue"), css="footer {visibility: hidden}") as demo:
-    gr.Markdown("# 💻 The C-Codestral IDE Agent")
+    gr.Markdown("# 💻 The C-Codestral IDE Agent (Library-Free)")
 
     with gr.Tabs():
         with gr.TabItem("👨‍💻 IDE"):
@@ -90,19 +103,11 @@ with gr.Blocks(theme=gr.themes.Monochrome(primary_hue="indigo", secondary_hue="b
                 with gr.Column(scale=2):
                     code_editor = gr.Code(label="C Code Editor", language="c", lines=25, interactive=True)
                 with gr.Column(scale=1):
-                    instruction_box = gr.Textbox(
-                        label="Instruction", 
-                        placeholder="e.g., 'Compile and run', 'Find bugs', 'Generate a factorial function'...",
-                        lines=3
-                    )
+                    instruction_box = gr.Textbox(label="Instruction", placeholder="e.g., 'Compile and run', 'Find bugs', 'Generate a factorial function'...", lines=3)
                     execute_btn = gr.Button("Execute Instruction", variant="primary")
                     output_box = gr.Textbox(label="Output / Console", lines=15, interactive=False, show_copy_button=True)
 
-            execute_btn.click(
-                fn=handle_instruction, 
-                inputs=[code_editor, instruction_box], 
-                outputs=[code_editor, output_box]
-            )
+            execute_btn.click(fn=handle_instruction, inputs=[code_editor, instruction_box], outputs=[code_editor, output_box])
 
         with gr.TabItem("🛠️ MCP Tools API"):
             gr.Markdown("## Available MCP Tools\nThese APIs can be used by any MCP-compliant client.")