Update app.py

app.py

@@ -75,6 +75,10 @@ def format_prompt(user_prompt, chat_history):
     return prompt
 
 # Model inference function
+def start_inference(prompt):
+    web_search = prompt.pop("Enable Web Search", False)
+    return next(model_inference(prompt["User Prompt"], web_search))
+
 def model_inference(prompt, web_search):
     for response in fetch_response(prompt, web_search):
         yield response
@@ -108,20 +112,15 @@ def fetch_response(prompt, web_search):
             yield output
 
 # Create a chatbot interface with a Fetch button
-def start_inference(prompt, web_search):
-    for response in model_inference(prompt, web_search):
-        return response
-
 chatbot = gr.Interface(
     fn=start_inference,
     inputs=[
         gr.Textbox(label="User Prompt", placeholder="Enter your prompt here..."),
-        gr.Checkbox(label="Enable Web Search", value=False),
-        "button"
+        gr.Checkbox(label="Enable Web Search", default=False, label_for_default="Search Web", label_for_changed="Searching Web")
     ],
     outputs=gr.Textbox(label="Response", placeholder="Responses will appear here..."),
     live=True
 )
 
 # Launch the Gradio interface
-chatbot.launch(share=True)
+chatbot.launch()