Update app.py

app.py

@@ -1,25 +1,24 @@
 import gradio as gr
 from duckduckgo_search import DDGS
-#from huggingface_hub import InferenceClient
 
 # Simulated knowledge base
 TROUBLESHOOTING_KB = {
     "wifi not working": [
-        "Check if your router is turned on.",
+        "Ensure your router is powered on.",
         "Restart your router.",
-        "Ensure your device is connected to the correct Wi-Fi network.",
-        "Try connecting another device to determine if the issue is with your device or the network."
+        "Verify your device is connected to the correct Wi-Fi network.",
+        "Attempt to connect another device to determine if the issue is device-specific."
     ],
     "computer is slow": [
-        "Close unused applications.",
-        "Check for viruses using antivirus software.",
+        "Close unnecessary applications.",
+        "Run a virus scan using your antivirus software.",
         "Restart your computer.",
-        "Check for available updates and install them."
+        "Check for and install available system updates."
     ]
 }
 
 # Tool: search_knowledge_base
-def search_knowledge_base(problem: str):
+def search_knowledge_base(problem: str) -> list[str]:
     problem = problem.lower()
     for key in TROUBLESHOOTING_KB:
         if key in problem:
@@ -27,7 +26,7 @@ def search_knowledge_base(problem: str):
     return ["No matching solution found in the knowledge base."]
 
 # Tool: search_web
-def search_web(query: str):
+def search_web(query: str) -> list[dict]:
     results = []
     with DDGS() as ddgs:
         for r in ddgs.text(query, max_results=3):
@@ -38,26 +37,37 @@ def search_web(query: str):
             })
     return results
 
-# Optional Tool: visit_webpage
-def visit_webpage(url: str):
-    return f"Stub for visiting {url}"
-
-# Optional Tool: format_response
-def format_response(text: str):
+# Tool: format_response
+def format_response(text: str) -> str:
     return f"🛠️ Troubleshooting Steps:\n{text}"
 
-# Gradio interface
-with gr.Blocks() as demo:
-    gr.Markdown("# 🧠 Troubleshooting MCP Server")
+# Create Gradio Interfaces for each tool
+kb_interface = gr.Interface(
+    fn=search_knowledge_base,
+    inputs=gr.Textbox(label="Describe your problem"),
+    outputs=gr.JSON(label="Knowledge Base Steps"),
+    title="Knowledge Base Search",
+    description="Search for troubleshooting steps in the internal knowledge base."
+)
 
-    with gr.Row():
-        kb_input = gr.Textbox(label="Describe your problem")
-        kb_output = gr.JSON(label="Knowledge Base Steps")
-        gr.Button("Search KB").click(search_knowledge_base, inputs=kb_input, outputs=kb_output)
+web_interface = gr.Interface(
+    fn=search_web,
+    inputs=gr.Textbox(label="Enter search query"),
+    outputs=gr.JSON(label="Web Search Results"),
+    title="Web Search",
+    description="Perform a web search for additional troubleshooting information."
+)
 
-    with gr.Row():
-        web_input = gr.Textbox(label="Enter search query")
-        web_output = gr.JSON(label="Web Search Results")
-        gr.Button("Search Web").click(search_web, inputs=web_input, outputs=web_output)
+format_interface = gr.Interface(
+    fn=format_response,
+    inputs=gr.Textbox(label="Response Text"),
+    outputs=gr.Textbox(label="Formatted Response"),
+    title="Response Formatter",
+    description="Format the troubleshooting steps for better readability."
+)
 
-demo.launch(mcp_server=True)
+# Launch the interfaces with MCP server enabled
+if __name__ == "__main__":
+    kb_interface.launch(mcp_server=True)
+    web_interface.launch(mcp_server=True)
+    format_interface.launch(mcp_server=True)