Update app.py

app.py

@@ -278,7 +278,7 @@ with gr.Blocks() as demo:
     
     with gr.Row():
         submit_btn = gr.Button("Send")
-        stop_btn = gr.Button("Stop", visible=False)
+        stop_btn = gr.Button("Stop")
         retry_btn = gr.Button("Retry")
         undo_btn = gr.Button("Undo")
         clear_btn = gr.Button("Clear")
@@ -329,16 +329,11 @@ with gr.Blocks() as demo:
         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
-    ).then(
-        lambda: gr.update(visible=True),
-        None,
-        stop_btn
-    )
-
+        )
     stop_btn.click(
-        lambda: (True, gr.update(visible=False)),
+        lambda: True,
         None,
-        [stop_clicked, stop_btn]
+        stop_clicked
     )
 
     retry_btn.click(