fix: uploads on client

app.py

@@ -2,12 +2,11 @@ import gradio as gr
 import requests
 import json
 import os
-import shutil
 import threading
-import mimetypes
 from models import OptimizeRequest, AutotuneRequest, QARequest
 from api import start_api
 
+# Start FastAPI server in background
 threading.Thread(target=start_api, daemon=True).start()
 
 # Base URL for internal calls
@@ -24,32 +23,45 @@ def call_api(endpoint: str, payload: dict) -> str:
 
 def upload_docs_tool(files, docs_path="data/docs"):
     """
-    Upload documents to the server's docs folder.
-
-    Accepts local file paths or URLs. Returns a list of uploaded file paths (strings)
-    to ensure MCP compatibility.
+    Upload documents to the server's docs folder via FastAPI /upload_docs.
+    Accepts:
+    - local file paths (str)
+    - URLs (str)
+    - file-like objects (from Cursor attachment)
     """
+    import shutil
+
     os.makedirs(docs_path, exist_ok=True)
-    saved = []
+    files_payload = []
 
     for f in files:
-        if f.startswith("http://") or f.startswith("https://"):
-            # Download file from URL
-            r = requests.get(f, stream=True)
+        if isinstance(f, str) and (f.startswith("http://") or f.startswith("https://")):
+            # Download URL temporarily
             fname = f.split("/")[-1]
-            dest = os.path.join(docs_path, fname)
-            with open(dest, "wb") as out_file:
+            tmp_path = os.path.join("/tmp", fname)
+            r = requests.get(f, stream=True)
+            with open(tmp_path, "wb") as out_file:
                 shutil.copyfileobj(r.raw, out_file)
-        else:
+            files_payload.append(("files", open(tmp_path, "rb")))
+        elif isinstance(f, str):
             # Local file path
-            fname = os.path.basename(f)
-            dest = os.path.join(docs_path, fname)
-            shutil.copy(f, dest)
-
-        # Only store the file path string for MCP compatibility
-        saved.append(dest)
+            files_payload.append(("files", open(f, "rb")))
+        else:
+            # Assume file-like object (Cursor attachment)
+            files_payload.append(("files", f))
 
-    return {"status": "ok", "uploaded_files": saved, "docs_path": docs_path}
+    try:
+        resp = requests.post(
+            f"{BASE_INTERNAL}/upload_docs",
+            files=files_payload,
+            data={"docs_path": docs_path}
+        )
+        return resp.json()
+    finally:
+        # Close all file handles
+        for _, file_obj in files_payload:
+            if not file_obj.closed:
+                file_obj.close()
 
 
 def optimize_rag_tool(payload: str) -> str:
@@ -67,7 +79,7 @@ def generate_qa_tool(payload: str) -> str:
     return call_api("/generate_validation_qa", json.loads(payload))
 
 
-# Dynamically assign Pydantic model docstrings to MCP tool functions
+# Assign Pydantic docstrings
 optimize_rag_tool.__doc__ = OptimizeRequest.__doc__
 autotune_tool.__doc__ = AutotuneRequest.__doc__
 generate_qa_tool.__doc__ = QARequest.__doc__
@@ -83,6 +95,7 @@ DEFAULT_OPTIMIZE_JSON = model_to_json(OptimizeRequest)
 DEFAULT_AUTOTUNE_JSON = model_to_json(AutotuneRequest)
 DEFAULT_QA_JSON = model_to_json(QARequest)
 
+
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# Ragmint MCP Client")
 
@@ -113,7 +126,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         gr.Markdown(AutotuneRequest.__doc__ or "No description available.")
         autotune_input = gr.Textbox(lines=12, value=DEFAULT_AUTOTUNE_JSON, label="AutotuneRequest JSON")
         autotune_btn = gr.Button("Submit", variant="primary")
-        autotune_out = gr.Textbox(lines=15)
+        autotune_out = gr.Textbox(lines=15, label="Response")
         autotune_btn.click(autotune_tool, inputs=autotune_input, outputs=autotune_out)
         gr.Markdown("---")