fix: uploads on client

app.py

@@ -5,12 +5,14 @@ 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 calls
+# Base URL for internal API calls
 BASE_INTERNAL = "http://127.0.0.1:8000"
 
 
@@ -25,31 +27,46 @@ 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 MCP FileData-like dicts.
+    Accepts:
+      - MCP FileData objects
+      - Local file paths
+      - URLs (http/https)
+    Returns MCP-compatible FileData dicts.
     """
     os.makedirs(docs_path, exist_ok=True)
     saved = []
 
     for f in files:
-        if f.startswith("http://") or f.startswith("https://"):
-            # Download file from URL
+        # ----- 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://")):
             r = requests.get(f, stream=True)
-            fname = f.split("/")[-1]
+            fname = f.split("/")[-1] or "downloaded_file"
             dest = os.path.join(docs_path, fname)
             with open(dest, "wb") as out_file:
                 shutil.copyfileobj(r.raw, out_file)
-        else:
-            # Local file path
+
+        # ----- Local file path -----
+        elif isinstance(f, str) and os.path.exists(f):
             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": fname,
+            "orig_name": os.path.basename(dest),
             "mime_type": mime_type,
             "is_stream": False,
             "meta": {}
@@ -59,21 +76,18 @@ def upload_docs_tool(files, docs_path="data/docs"):
 
 
 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 model docstrings to MCP tool functions
+# Dynamically assign Pydantic docstrings
 optimize_rag_tool.__doc__ = OptimizeRequest.__doc__
 autotune_tool.__doc__ = AutotuneRequest.__doc__
 generate_qa_tool.__doc__ = QARequest.__doc__
@@ -95,8 +109,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     # Upload Documents
     with gr.Column():
         gr.Markdown("## Upload Documents")
-        gr.Markdown("📂 Upload files (local paths or URLs) to your `data/docs` folder")
-        upload_files = gr.File(file_count="multiple", type="filepath")
+        gr.Markdown("📂 Accepts MCP FileData, local paths, or URLs")
+        upload_files = gr.File(file_count="multiple", type="file")  # MCP FileData compatible
         upload_path = gr.Textbox(value=DEFAULT_UPLOAD_PATH, label="Docs Path")
         upload_btn = gr.Button("Upload", variant="primary")
         upload_out = gr.JSON(label="Response")