Update app.py

app.py

@@ -1,31 +1,28 @@
 import gradio as gr
 from duckduckgo_search import DDGS
 
-# Simulated knowledge base
-TROUBLESHOOTING_KB = {
-    "wifi not working": [
-        "Ensure your router is powered on.",
-        "Restart your router.",
-        "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 unnecessary applications.",
-        "Run a virus scan using your antivirus software.",
-        "Restart your computer.",
-        "Check for and install available system updates."
-    ]
-}
-
-# Tool: search_knowledge_base
+# Tool 1: Knowledge Base Search
 def search_knowledge_base(problem: str) -> list[str]:
     problem = problem.lower()
-    for key in TROUBLESHOOTING_KB:
+    KB = {
+        "wifi not working": [
+            "Check if the router is on.",
+            "Restart the router.",
+            "Check if your device is connected to the right network."
+        ],
+        "computer is slow": [
+            "Close unused programs.",
+            "Run antivirus scan.",
+            "Restart the computer.",
+            "Update your OS."
+        ]
+    }
+    for key in KB:
         if key in problem:
-            return TROUBLESHOOTING_KB[key]
+            return KB[key]
     return ["No matching solution found in the knowledge base."]
 
-# Tool: search_web
+# Tool 2: Web Search
 def search_web(query: str) -> list[dict]:
     results = []
     with DDGS() as ddgs:
@@ -37,37 +34,36 @@ def search_web(query: str) -> list[dict]:
             })
     return results
 
-# Tool: format_response
+# Tool 3: Format response
 def format_response(text: str) -> str:
-    return f"🛠️ Troubleshooting Steps:\n{text}"
-
-# 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."
-)
+    return f"🛠️ Suggested Steps:\n{text}"
 
-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."
-)
+# Combine tools into a single unified function
+def toolbox(tool_selector, input_text):
+    if tool_selector == "search_knowledge_base":
+        return search_knowledge_base(input_text)
+    elif tool_selector == "search_web":
+        return search_web(input_text)
+    elif tool_selector == "format_response":
+        return format_response(input_text)
+    else:
+        return "Invalid tool selected."
 
-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."
+# Define Interface
+demo = gr.Interface(
+    fn=toolbox,
+    inputs=[
+        gr.Dropdown(
+            ["search_knowledge_base", "search_web", "format_response"],
+            label="Tool Selector"
+        ),
+        gr.Textbox(label="Input Text")
+    ],
+    outputs=gr.JSON(label="Output"),
+    title="Troubleshooting MCP Server",
+    description="Select a tool and provide input."
 )
 
-# Launch the interfaces with MCP server enabled
+# Launch 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)
+    demo.launch(mcp_server=True)