fix: uploads on client

app.py

@@ -5,14 +5,12 @@ import os
 import shutil
 import threading
 import mimetypes
-import base64
 from models import OptimizeRequest, AutotuneRequest, QARequest
 from api import start_api
-from gradio_client import FileData
 
 threading.Thread(target=start_api, daemon=True).start()
 
-# Base URL for internal API calls
+# Base URL for internal calls
 BASE_INTERNAL = "http://127.0.0.1:8000"
 
 
@@ -27,67 +25,49 @@ 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:
-      - MCP FileData objects
-      - Local file paths
-      - URLs (http/https)
-    Returns MCP-compatible FileData dicts.
+
+    Accepts local file paths or URLs. Returns a list of uploaded file paths (strings)
+    to ensure MCP compatibility.
     """
     os.makedirs(docs_path, exist_ok=True)
     saved = []
 
     for f in files:
-        # ----- MCP FileData -----
-        if isinstance(f, dict) and "data" in f:
-            content = base64.b64decode(f["data"])
-            fname = f.get("orig_name", f.get("name", "unknown_file"))
-            dest = os.path.join(docs_path, fname)
-            with open(dest, "wb") as out_file:
-                out_file.write(content)
-
-        # ----- URL -----
-        elif isinstance(f, str) and (f.startswith("http://") or f.startswith("https://")):
+        if f.startswith("http://") or f.startswith("https://"):
+            # Download file from URL
             r = requests.get(f, stream=True)
-            fname = f.split("/")[-1] or "downloaded_file"
+            fname = f.split("/")[-1]
             dest = os.path.join(docs_path, fname)
             with open(dest, "wb") as out_file:
                 shutil.copyfileobj(r.raw, out_file)
-
-        # ----- Local file path -----
-        elif isinstance(f, str) and os.path.exists(f):
+        else:
+            # Local file path
             fname = os.path.basename(f)
             dest = os.path.join(docs_path, fname)
             shutil.copy(f, dest)
 
-        else:
-            continue
-
-        mime_type = mimetypes.guess_type(dest)[0] or "application/octet-stream"
-        saved.append({
-            "path": dest,
-            "url": f"file://{os.path.abspath(dest)}",
-            "orig_name": os.path.basename(dest),
-            "mime_type": mime_type,
-            "is_stream": False,
-            "meta": {}
-        })
+        # Only store the file path string for MCP compatibility
+        saved.append(dest)
 
     return {"status": "ok", "uploaded_files": saved, "docs_path": docs_path}
 
 
 def optimize_rag_tool(payload: str) -> str:
+    """🔧 Explicit optimization request: user provides all pipeline configs manually."""
     return call_api("/optimize_rag", json.loads(payload))
 
 
 def autotune_tool(payload: str) -> str:
+    """🔧 Autotune RAG: recommends chunk sizes and embedding models automatically."""
     return call_api("/autotune_rag", json.loads(payload))
 
 
 def generate_qa_tool(payload: str) -> str:
+    """🧩 Generates a validation QA dataset for RAG evaluation."""
     return call_api("/generate_validation_qa", json.loads(payload))
 
 
-# Dynamically assign Pydantic docstrings
+# Dynamically assign Pydantic model docstrings to MCP tool functions
 optimize_rag_tool.__doc__ = OptimizeRequest.__doc__
 autotune_tool.__doc__ = AutotuneRequest.__doc__
 generate_qa_tool.__doc__ = QARequest.__doc__
@@ -109,8 +89,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     # Upload Documents
     with gr.Column():
         gr.Markdown("## Upload Documents")
-        gr.Markdown("📂 Accepts MCP FileData, local paths, or URLs")
-        upload_files = gr.File(file_count="multiple", type="file")  # MCP FileData compatible
+        gr.Markdown("📂 Upload files (local paths or URLs) to your `data/docs` folder")
+        upload_files = gr.File(file_count="multiple", type="filepath")
         upload_path = gr.Textbox(value=DEFAULT_UPLOAD_PATH, label="Docs Path")
         upload_btn = gr.Button("Upload", variant="primary")
         upload_out = gr.JSON(label="Response")