Update app.py

app.py

@@ -1,69 +1,47 @@
 import gradio as gr
-from duckduckgo_search import DDGS
 
-# 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."]
+# --- Tool: Simulated troubleshooting knowledge base ---
+def search_knowledge_base(issue: str) -> str:
+    if "wifi" in issue.lower():
+        return "Check if your router is powered on. Restart your router. Try connecting again."
+    elif "screen" in issue.lower():
+        return "Adjust display cable. Update graphics drivers. Restart the system."
+    return "No predefined solution found in the knowledge base."
 
-# 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: Simulated web search (basic dummy) ---
+def search_web(query: str) -> str:
+    return f"(Simulated Web Result) Top search result for '{query}'."
 
-# Tool 3: Format response
-def format_response(text: str) -> str:
-    return f"🛠️ Suggested Steps:\n{text}"
+# --- Tool: Format output into step-by-step troubleshooting response ---
+def format_response(raw_steps: str) -> str:
+    steps = raw_steps.split(". ")
+    steps = [f"{i+1}. {step.strip('.')}" for i, step in enumerate(steps) if step.strip()]
+    return "\n".join(steps)
 
-# 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."
+# Register each tool with a separate interface
+tools = [
+    gr.Interface(fn=search_knowledge_base,
+                 inputs="text",
+                 outputs="text",
+                 title="Knowledge Base Search",
+                 description="Search for predefined troubleshooting instructions.",
+                 name="search_knowledge_base"),
 
-# 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."
-)
+    gr.Interface(fn=search_web,
+                 inputs="text",
+                 outputs="text",
+                 title="Web Search Tool",
+                 description="Fetch dynamic info from the web (simulated).",
+                 name="search_web"),
 
-# Launch with MCP server enabled
+    gr.Interface(fn=format_response,
+                 inputs="text",
+                 outputs="text",
+                 title="Response Formatter",
+                 description="Format the raw instructions into steps.",
+                 name="format_response")
+]
+
+# Launch all tools under one MCP server
 if __name__ == "__main__":
-    demo.launch()
+    gr.mount_gradio_apps(tools).launch()