refactor: document upload allows FileData

app.py

@@ -4,6 +4,7 @@ import json
 import os
 import shutil
 import threading
+import mimetypes
 from models import OptimizeRequest, AutotuneRequest, QARequest
 from api import start_api
 
@@ -25,20 +26,35 @@ def upload_docs_tool(files, docs_path="data/docs"):
     """
     Upload documents to the server's docs folder.
 
-    Parameters:
-    - files: List of file paths to upload
-    - docs_path: Destination folder (default: data/docs)
-
-    Returns:
-    - dict with uploaded filenames and docs_path
+    Accepts local file paths or URLs. Returns a list of MCP FileData-like dicts.
     """
     os.makedirs(docs_path, exist_ok=True)
     saved = []
+
     for f in files:
-        fname = os.path.basename(f)
-        dest = os.path.join(docs_path, fname)
-        shutil.copy(f, dest)
-        saved.append(fname)
+        if f.startswith("http://") or f.startswith("https://"):
+            # Download file from URL
+            r = requests.get(f, stream=True)
+            fname = f.split("/")[-1]
+            dest = os.path.join(docs_path, fname)
+            with open(dest, "wb") as out_file:
+                shutil.copyfileobj(r.raw, out_file)
+        else:
+            # Local file path
+            fname = os.path.basename(f)
+            dest = os.path.join(docs_path, fname)
+            shutil.copy(f, dest)
+
+        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,
+            "mime_type": mime_type,
+            "is_stream": False,
+            "meta": {}
+        })
+
     return {"status": "ok", "uploaded_files": saved, "docs_path": docs_path}
 
 
@@ -79,11 +95,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     # Upload Documents
     with gr.Column():
         gr.Markdown("## Upload Documents")
-        gr.Markdown("📂 Upload files to your `data/docs` folder")
+        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.Textbox(label="Response")
+        upload_out = gr.JSON(label="Response")
         upload_btn.click(upload_docs_tool, inputs=[upload_files, upload_path], outputs=upload_out)
         gr.Markdown("---")