added tool mcp upload

app.py

@@ -89,6 +89,32 @@ def model_to_json(model_cls) -> str:
     return json.dumps({k: v.default for k, v in model_cls.__fields__.items()}, indent=2)
 
 
+def upload_docs_tool_mcp(payload: str, files=None):
+    """
+    📦 Upload documents to the server's docs folder via FastAPI /upload_docs.
+    Accepts:
+    - attached file(s) from Cursor MCP agent
+    - 'docs_path' in JSON payload
+    """
+    data = json.loads(payload) if payload else {}
+    docs_path = data.get("docs_path", DEFAULT_UPLOAD_PATH)
+
+    if not files:
+        return {"error": "No files provided to upload_docs_tool_mcp"}
+
+    os.makedirs(docs_path, exist_ok=True)
+
+    files_payload = [("files", (f.name, f, "application/octet-stream")) for f in files]
+
+    resp = requests.post(
+        f"{BASE_INTERNAL}/upload_docs",
+        files=files_payload,
+        data={"docs_path": docs_path}
+    )
+
+    return resp.json()
+
+
 # Default inputs
 DEFAULT_UPLOAD_PATH = "data/docs"
 DEFAULT_OPTIMIZE_JSON = model_to_json(OptimizeRequest)
@@ -141,6 +167,20 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         gr.Markdown("---")
 
 if __name__ == "__main__":
+    # Register MCP tools
+    demo.add_mcp_tool(
+        name="ragmint_upload_docs_mcp",
+        func=upload_docs_tool_mcp,
+        description="📂 Upload documents to data/docs for use in RAG optimization and autotuning",
+        parameters={
+            "type": "object",
+            "properties": {
+                "docs_path": {"type": "string", "description": "Target directory for uploaded docs"}
+            }
+        },
+        file_input=True  # <<< enables file attachments from Cursor
+    )
+
     demo.launch(
         server_name="0.0.0.0",
         server_port=7860,