Update app.py

app.py

@@ -1,23 +1,28 @@
-# app.py
-import gradio as gr
-from datetime import datetime, timezone
+from datetime import datetime
+from fastapi import FastAPI
+from mcp.server.fastmcp import FastMCP
+import zoneinfo
 
-def now_utc():
-    """
-    Return the current time in ISO 8601 (UTC).
-    No inputs. Useful as a 'time' tool for MCP clients.
-    """
-    # generator -> streamable over HTTP/SSE (client will see a streamed event)
-    yield datetime.now(timezone.utc).isoformat()
-
-demo = gr.Interface(
-    fn=now_utc,
-    inputs=None,
-    outputs=gr.Textbox(label="UTC datetime (ISO 8601)"),
-    title="Datetime MCP Server",
-    description="Returns the current UTC datetime."
+# --- Minimal MCP server that returns current datetime ---
+mcp = FastMCP(
+    name="datetime",
+    stateless_http=True,         # no auth/state, HTTP transport
+    # json_response=True,        # optional: force JSON-only if your client prefers it
 )
 
-if __name__ == "__main__":
-    # Starts the web UI and an MCP SSE endpoint; no authentication configured.
-    demo.launch(mcp_server=True)
+@mcp.tool(description="Return current datetime. tz is an IANA TZ name (e.g. 'UTC', 'Asia/Kolkata').")
+def get_datetime(tz: str = "UTC", iso: bool = True) -> str:
+    z = zoneinfo.ZoneInfo(tz)
+    now = datetime.now(tz=z)
+    return now.isoformat() if iso else now.strftime("%Y-%m-%d %H:%M:%S %Z")
+
+# --- Host app ---
+app = FastAPI()
+
+# Mount the Streamable HTTP MCP server at /mcp with NO extra middleware around it
+app.mount("/mcp", mcp.streamable_http_app())
+
+# (Optional) tiny root page so Spaces shows something at "/"
+@app.get("/")
+def root():
+    return {"ok": True, "mcp": "/mcp"}