Update app.py

app.py

@@ -69,5 +69,11 @@ def web_search(query: str):
         messages = messages[-6:]
     return f"You: {query}\nIT-Assistant: {response.content}"
 
-demo = gr.Interface(fn=handle_query, inputs=["text", "text"], outputs="text", title="IT Assistant", description="Choose a mode and enter your message, then click submit to interact.", choices=["chat", "web-search"])
+demo = gr.Interface(
+    fn=handle_query,
+    inputs=["text", "text", gr.Dropdown(["chat", "web-search"], label="Mode", default="chat")],
+    outputs="text",
+    title="IT Assistant",
+    description="Choose a mode and enter your message, then click submit to interact."
+)
 demo.launch()