Update app.py

app.py

@@ -1,6 +1,9 @@
 import gradio as gr
 
 def search_knowledge_base(issue: str) -> str:
+    """
+    Search the knowledge base for solutions related to the provided issue.
+    """
     if "wifi" in issue.lower():
         return "Check if your router is powered on. Restart your router. Try connecting again."
     elif "screen" in issue.lower():
@@ -8,26 +11,23 @@ def search_knowledge_base(issue: str) -> str:
     return "No predefined solution found in the knowledge base."
 
 def search_web(query: str) -> str:
+    """
+    Perform a simulated web search for the given query.
+    """
     return f"(Simulated Web Result) Top search result for '{query}'."
 
 def format_response(raw_steps: str) -> str:
+    """
+    Format the raw steps into a numbered list.
+    """
     steps = raw_steps.split(". ")
     steps = [f"{i+1}. {step.strip('.')}" for i, step in enumerate(steps) if step.strip()]
     return "\n".join(steps)
 
-# Create Gradio interfaces
-kb_tool = gr.Interface(fn=search_knowledge_base, inputs="text", outputs="text")
-web_tool = gr.Interface(fn=search_web, inputs="text", outputs="text")
-format_tool = gr.Interface(fn=format_response, inputs="text", outputs="text")
-
-# Mount apps using gr.mount_gradio_apps()
-demo = gr.mount_gradio_apps(
-    apps=[
-        (kb_tool, "/search_knowledge_base"),
-        (web_tool, "/search_web"),
-        (format_tool, "/format_response"),
-    ]
-)
+with gr.Blocks() as demo:
+    gr.Interface(fn=search_knowledge_base, inputs="text", outputs="text", title="Knowledge Base Search")
+    gr.Interface(fn=search_web, inputs="text", outputs="text", title="Web Search")
+    gr.Interface(fn=format_response, inputs="text", outputs="text", title="Response Formatter")
 
 if __name__ == "__main__":
-    demo.launch(mcp_server=True)
+    demo.launch(mcp_server=True, share=True)