Update app.py

app.py

@@ -7,11 +7,15 @@ import time
 import logging
 import re
 from smolagents.tools import tool
+from fastapi import FastAPI
+import uvicorn
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+app = FastAPI()
+
 @tool
 def search_knowledge_base(issue: str) -> str:
     """
@@ -146,7 +150,7 @@ def format_response(raw_steps: str) -> str:
         logger.error(f"Error formatting steps: {str(e)}")
         return f"Error formatting your steps: {str(e)}"
 
-# Knowledge Base Search Interface
+# Create interfaces
 demo1 = gr.Interface(
     fn=search_knowledge_base,
     inputs=[gr.Textbox(label="Technical Issue", placeholder="Enter your technical problem")],
@@ -154,7 +158,6 @@ demo1 = gr.Interface(
     title="Knowledge Base Search"
 )
 
-# Web Search Interface
 demo2 = gr.Interface(
     fn=search_web,
     inputs=[gr.Textbox(label="Search Query", placeholder="Enter your search query")],
@@ -162,7 +165,6 @@ demo2 = gr.Interface(
     title="Web Search"
 )
 
-# Response Formatter Interface
 demo3 = gr.Interface(
     fn=format_response,
     inputs=[gr.Textbox(label="Raw Steps", placeholder="Enter steps separated by periods")],
@@ -172,8 +174,46 @@ demo3 = gr.Interface(
 
 demo = gr.TabbedInterface([demo1, demo2, demo3], ["Knowledge Base", "Web Search", "Formatter"])
 
+# Add MCP schema endpoint
+@app.get("/gradio_api/mcp/schema")
+async def mcp_schema():
+    return {
+        "tools": [
+            {
+                "name": "search_knowledge_base",
+                "description": "Search knowledge base for technical solutions",
+                "parameters": {
+                    "type": "object",
+                    "properties": {
+                        "issue": {"type": "string"}
+                    }
+                }
+            },
+            {
+                "name": "search_web",
+                "description": "Perform web search",
+                "parameters": {
+                    "type": "object",
+                    "properties": {
+                        "query": {"type": "string"}
+                    }
+                }
+            },
+            {
+                "name": "format_response",
+                "description": "Format steps into numbered list",
+                "parameters": {
+                    "type": "object",
+                    "properties": {
+                        "raw_steps": {"type": "string"}
+                    }
+                }
+            }
+        ]
+    }
+
+# Mount Gradio app
+app = gr.mount_gradio_app(app, demo, path="/")
+
 if __name__ == "__main__":
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860
-    )
+    uvicorn.run(app, host="0.0.0.0", port=7860)