Update app.py

app.py

@@ -1,33 +1,71 @@
 import gradio as gr
 
-def search_knowledge_base(issue: str) -> str:
+def search_knowledge_base(issue):
     """
     Search the knowledge base for solutions related to the provided issue.
+    Args:
+        issue (str): The technical issue to search for
+    Returns:
+        str: Solution steps or message if no solution found
     """
-    if "wifi" in issue.lower():
+    issue = issue.lower()
+    if "wifi" in issue:
         return "Check if your router is powered on. Restart your router. Try connecting again."
-    elif "screen" in issue.lower():
+    elif "screen" in issue:
         return "Adjust display cable. Update graphics drivers. Restart the system."
     return "No predefined solution found in the knowledge base."
 
-def search_web(query: str) -> str:
+def search_web(query):
     """
     Perform a simulated web search for the given query.
+    Args:
+        query (str): The search query
+    Returns:
+        str: Simulated search results
     """
     return f"(Simulated Web Result) Top search result for '{query}'."
 
-def format_response(raw_steps: str) -> str:
+def format_response(raw_steps):
     """
     Format the raw steps into a numbered list.
+    Args:
+        raw_steps (str): Raw text steps separated by periods
+    Returns:
+        str: Formatted numbered list of steps
     """
     steps = raw_steps.split(". ")
     steps = [f"{i+1}. {step.strip('.')}" for i, step in enumerate(steps) if step.strip()]
     return "\n".join(steps)
 
-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")
+# Knowledge Base Search Interface
+demo1 = gr.Interface(
+    fn=search_knowledge_base,
+    inputs=[gr.Textbox("wifi connection problem")],
+    outputs=[gr.Textbox()],
+    title="Knowledge Base Search",
+    description="Enter a technical issue to search for solutions in the knowledge base."
+)
+
+# Web Search Interface  
+demo2 = gr.Interface(
+    fn=search_web,
+    inputs=[gr.Textbox("latest tech news")],
+    outputs=[gr.Textbox()],
+    title="Web Search",
+    description="Enter a search query to get simulated web search results."
+)
+
+# Response Formatter Interface
+demo3 = gr.Interface(
+    fn=format_response,
+    inputs=[gr.Textbox("Step one. Step two. Step three")],
+    outputs=[gr.Textbox()],
+    title="Response Formatter",
+    description="Enter raw steps separated by periods to format them as a numbered list."
+)
+
+# Combine all interfaces using Tabs (following demo pattern)
+demo = gr.TabbedInterface([demo1, demo2, demo3], ["Knowledge Base", "Web Search", "Formatter"])
 
 if __name__ == "__main__":
     demo.launch(share=True)