Update app.py

app.py

@@ -1,9 +1,69 @@
 import gradio as gr
+from duckduckgo_search import DDGS
 
-def greet(name):
-    return f"Hello, {name}!"
+# Tool 1: Knowledge Base Search
+def search_knowledge_base(problem: str) -> list[str]:
+    problem = problem.lower()
+    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 KB[key]
+    return ["No matching solution found in the knowledge base."]
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
+# Tool 2: Web Search
+def search_web(query: str) -> list[dict]:
+    results = []
+    with DDGS() as ddgs:
+        for r in ddgs.text(query, max_results=3):
+            results.append({
+                "title": r["title"],
+                "body": r["body"],
+                "href": r["href"]
+            })
+    return results
 
+# Tool 3: Format response
+def format_response(text: str) -> str:
+    return f"🛠️ Suggested Steps:\n{text}"
+
+# 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."
+
+# 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 with MCP server enabled
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(mcp_server=True)