Update app.py

app.py

@@ -284,8 +284,16 @@ with gr.Blocks() as demo:
         print(f"Returning history with {len(history)} items")
         return "", history, is_generating, stop_clicked
 
+    def on_submit(message, history, use_web_search, model, temperature, num_calls, is_generating, stop_clicked):
+        print(f"Submit button clicked with message: {message}")
+        _, new_history, new_is_generating, new_stop_clicked = protected_generate_response(
+            message, history, use_web_search, model, temperature, num_calls, is_generating, lambda: stop_clicked
+        )
+        print(f"New history has {len(new_history)} items")
+        return "", new_history, new_is_generating, new_stop_clicked
+
     submit_btn.click(
-        protected_generate_response,
+        on_submit,
         inputs=[msg, chatbot, use_web_search, model_dropdown, temperature_slider, num_calls_slider, is_generating, stop_clicked],
         outputs=[msg, chatbot, is_generating, stop_clicked],
         show_progress=True
@@ -293,10 +301,6 @@ with gr.Blocks() as demo:
         lambda: gr.update(visible=True),
         None,
         stop_btn
-    ).then(
-        lambda: gr.update(visible=False),
-        None,
-        stop_btn
     )
 
     stop_btn.click(
@@ -310,7 +314,7 @@ with gr.Blocks() as demo:
         inputs=[chatbot],
         outputs=[msg, chatbot]
     ).then(
-        protected_generate_response,
+        on_submit,
         inputs=[msg, chatbot, use_web_search, model_dropdown, temperature_slider, num_calls_slider, is_generating, stop_clicked],
         outputs=[msg, chatbot, is_generating, stop_clicked]
     )
@@ -341,6 +345,5 @@ with gr.Blocks() as demo:
     8. Click "Stop" during generation to halt the process.
     """
     )
-
 if __name__ == "__main__":
     demo.launch(share=True)