Upload 8 files

Dockerfile

@@ -3,6 +3,9 @@ FROM python:3.10-slim
 # Set working directory in the container
 WORKDIR /app
 
+# Unbuffer STDIO for subprocess pipes (fixes hanging in MCP stdio)
+ENV PYTHONUNBUFFERED=1
+
 # Copy requirements and install dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
@@ -24,5 +27,5 @@ EXPOSE 8501
 # Health check for Streamlit
 HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health || exit 1
 
-# Run MCP servers in background, then Streamlit (using shell for &)
-CMD sh -c "python /app/arithmetic_server.py & python /app/stock_server.py & sleep 5 && streamlit run /app/frontend.py --server.port=8501 --server.address=0.0.0.0"
+# Run Streamlit
+CMD ["streamlit", "run", "frontend.py", "--server.port=8501", "--server.address=0.0.0.0"]

arithmetic_server.py

@@ -41,4 +41,4 @@ app = Starlette(
 )
 
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=8001)
+    mcp.run(transport="stdio")

backend.py

@@ -174,12 +174,14 @@ class MCPAgent:
         # Start the Stock server separately first: `python stockserver.py`
         self.client = MultiServerMCPClient({
             "arithmetic": {
-                "url": "http://localhost:8001",
-                "transport": "streamable_http",
+                "command": sys.executable,
+                "args": [str(here / "arithmetic_server.py")],
+                "transport": "stdio",
             },
             "stocks": {
-                "url": "http://localhost:8002",
-                "transport": "streamable_http",
+                "command": sys.executable,
+                "args": [str(here / "stock_server.py")],
+                "transport": "stdio",
             },
         }
     )

stock_server.py

@@ -75,4 +75,4 @@ app = Starlette(
 )
 
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=8002)
+    mcp.run(transport="stdio")