Update app.py

app.py

@@ -1,4 +1,4 @@
-import os, shutil, tempfile, re, json
+import os, shutil, tempfile, re
 from pathlib import Path
 import gradio as gr
 from git import Repo
@@ -6,7 +6,7 @@ import requests
 
 # ---------------- CONFIG ----------------
 OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY")
-OPENROUTER_MODEL = "nvidia/nemotron-nano-12b-v2-vl:free"  # free OpenRouter model
+OPENROUTER_MODEL = "nvidia/nemotron-nano-12b-v2-vl:free"
 OPENROUTER_URL = "https://openrouter.ai/api/v1/chat/completions"
 HEADERS = {
     "Authorization": f"Bearer {OPENROUTER_API_KEY}",
@@ -19,8 +19,8 @@ ALLOWED_EXT = {
     ".css", ".yml", ".yaml", ".toml", ".ini", ".json"
 }
 SKIP_DIRS = {
-    "node_modules", ".git", "dist", "build", "out", "venv", ".venv", "__pycache__",
-    ".next", ".cache", "target", "bin", "obj", ".idea", ".vscode"
+    "node_modules", ".git", "dist", "build", "out", "venv", ".venv",
+    "__pycache__", ".next", ".cache", "target", "bin", "obj", ".idea", ".vscode"
 }
 MAX_FILE_BYTES = 800_000
 
@@ -91,13 +91,13 @@ SYSTEM_PROMPT = (
 
 def chat_repo(user_msg, chat_history, repo_text):
     if not repo_text:
-        chat_history.append(("❌", "Please analyze a repository first."))
+        chat_history.append({"role": "assistant", "content": "❌ Please analyze a repository first."})
         return chat_history, ""
     
-    # Truncate context for performance (first ~120k chars)
-    context = repo_text[:120000]
+    context = repo_text[:120000]  # truncate for token safety
     response = openrouter_chat(SYSTEM_PROMPT, user_msg, context)
-    chat_history.append((user_msg, response))
+    chat_history.append({"role": "user", "content": user_msg})
+    chat_history.append({"role": "assistant", "content": response})
     return chat_history, ""
 
 # ---------------- GRADIO UI ----------------
@@ -124,8 +124,9 @@ with gr.Blocks(title="Repo Chatbot · OpenRouter") as demo:
 
     status_box = gr.Markdown()
 
-    chatbot = gr.Chatbot(label="Repo Chatbot", height=500)
-    user_box = gr.Textbox(label="Type your question about the repo and press Enter")
+    # ✅ Add type='messages' to match new format
+    chatbot = gr.Chatbot(label="Repo Chatbot", height=500, type="messages")
+    user_box = gr.Textbox(label="Ask something about the repo...")
 
     clear_btn = gr.Button("🧹 Clear Chat")
 
@@ -139,5 +140,5 @@ with gr.Blocks(title="Repo Chatbot · OpenRouter") as demo:
                     outputs=[chatbot, user_box])
     clear_btn.click(lambda: ([], ""), None, [chatbot, user_box])
 
-    demo.queue(concurrency_count=3)
+    demo.queue()  # ✅ removed unsupported arg
     demo.launch(server_name="0.0.0.0", server_port=7860)