Update app.py

app.py

@@ -327,43 +327,43 @@ with gr.Blocks() as demo:
 
         with gr.Row(elem_id="run_sim_btn", visible=False) as run_sim_btn:
             submit_btn = gr.Button("Run Simulation")
-            submit_btn.click(run_simulation,
-                             inputs=[topic,
-                                     save,
-                                     num_responses,
-                                     n,
-                                     system_prompt,
-                                     with_images,
-                                     img_mode,
-                                     augment_mode,
-                                     image_every_n_posts,
-                                     image_subtoken,
-                                     image_tokens,
-                                     news_company,
-                                     text_model_name],
-                             outputs=[progress_output,
-                                      log_output,
-                                      simulation_output_box,
-                                      saved_file,
-                                      example_image])
+            submt_process = submit_btn.click(run_simulation,
+                                             inputs=[topic,
+                                                     save,
+                                                     num_responses,
+                                                     n,
+                                                     system_prompt,
+                                                     with_images,
+                                                     img_mode,
+                                                     augment_mode,
+                                                     image_every_n_posts,
+                                                     image_subtoken,
+                                                     image_tokens,
+                                                     news_company,
+                                                     text_model_name],
+                                             outputs=[progress_output,
+                                                      log_output,
+                                                      simulation_output_box,
+                                                      saved_file,
+                                                      example_image])
 
         with gr.Row(elem_id="init_btn", visible=True) as init_btn:
             submit_btn = gr.Button("Initialize Bot")
-            submit_btn.click(init_app,
-                             inputs=[api_key,
-                                     model_name,
-                                     model_class,
-                                     temperature,
-                                     diffusion_model,
-                                     keyword_model,
-                                     dtype,
-                                     topic_prompt],
-                             outputs=[progress_output,
-                                      log_output,
-                                      main_block_step_1,
-                                      main_block_step_2,
-                                      init_btn,
-                                      run_sim_btn])
+            submt_process = submit_btn.click(init_app,
+                                             inputs=[api_key,
+                                                     model_name,
+                                                     model_class,
+                                                     temperature,
+                                                     diffusion_model,
+                                                     keyword_model,
+                                                     dtype,
+                                                     topic_prompt],
+                                             outputs=[progress_output,
+                                                      log_output,
+                                                      main_block_step_1,
+                                                      main_block_step_2,
+                                                      init_btn,
+                                                      run_sim_btn])
 
         with gr.Row(elem_id="reset_block", visible=True) as reset_block:
             reset_btn = gr.Button("Reset")
@@ -371,7 +371,7 @@ with gr.Blocks() as demo:
                                                            topic, topic_prompt, with_images, image_params,
                                                            main_block_step_1, main_block_step_2,
                                                            simulation_output_box,
-                                                           run_sim_btn, init_btn], cancels=[submit_btn])
+                                                           run_sim_btn, init_btn], cancels=[submt_process])
 
     with gr.Column(elem_id = "user_consent_container") as user_consent_block:
         accept_checkbox = gr.Checkbox(visible=False)