Update webui.py

webui.py

@@ -29,6 +29,8 @@ def get_task(*args):
     return worker.AsyncTask(args=args)
 
 def generate_clicked(task):
+    worker.AsyncTask(args=[])
+
     import ldm_patched.modules.model_management as model_management
 
     with model_management.interrupt_processing_mutex:
@@ -53,7 +55,7 @@ def generate_clicked(task):
                 # help bad internet connection by skipping duplicated preview
                 if len(task.yields) > 0:  # if we have the next item
                     if task.yields[0][0] == 'preview':   # if the next item is also a preview
-                        print('Skipped one preview for better internet connection.')
+                        # print('Skipped one preview for better internet connection.')
                         continue
 
                 percentage, title, image = product
@@ -124,6 +126,8 @@ with shared.gradio_root:
                     skip_button = gr.Button(label="Skip", value="Skip", elem_classes='type_row_half', visible=False)
                     stop_button = gr.Button(label="Stop", value="Stop", elem_classes='type_row_half', elem_id='stop_button', visible=False)
 
+                    currentTask = gr.State(worker.AsyncTask(args=[]))
+
                     def stop_clicked(currentTask):
                         import ldm_patched.modules.model_management as model_management
                         currentTask.last_stop = 'stop'
@@ -576,6 +580,8 @@ with shared.gradio_root:
             inpaint_strength, inpaint_respective_field
         ], show_progress=False, queue=False)
 
+        currentTask = gr.State(worker.AsyncTask(args=[]))
+
         ctrls = [currentTask, generate_image_grid]
         ctrls += [
             prompt, negative_prompt, style_selections,
@@ -645,6 +651,8 @@ with shared.gradio_root:
         generate_button.click(lambda: (gr.update(visible=True, interactive=True), gr.update(visible=True, interactive=True), gr.update(visible=False, interactive=False), [], True),
                               outputs=[stop_button, skip_button, generate_button, gallery, state_is_generating]) \
             .then(fn=refresh_seed, inputs=[seed_random, image_seed], outputs=image_seed) \
+            .then(fn=get_task, inputs=ctrls, outputs=currentTask) \
+            .then(fn=generate_clicked, inputs=currentTask, outputs=[progress_html, progress_window, progress_gallery, gallery]) \
             .then(lambda: (gr.update(visible=True, interactive=True), gr.update(visible=False, interactive=False), gr.update(visible=False, interactive=False), False),
                   outputs=[generate_button, stop_button, skip_button, state_is_generating]) \
             .then(fn=update_history_link, outputs=history_link) \
@@ -675,12 +683,12 @@ def dump_default_english_config():
 
 # dump_default_english_config()
 
-shared.gradio_root.launch()
-     inbrowser=args_manager.args.in_browser,
-     server_name=args_manager.args.listen,
-     server_port=args_manager.args.port,
-     share=args_manager.args.share,
-     auth=check_auth if (args_manager.args.share or args_manager.args.listen) and auth_enabled else None,
-     allowed_paths=[modules.config.path_outputs],
-     blocked_paths=[constants.AUTH_FILENAME]
-)
+shared.gradio_root.launch(
+    inbrowser=args_manager.args.in_browser,
+    server_name=args_manager.args.listen,
+    server_port=args_manager.args.port,
+    share=args_manager.args.share,
+    auth=check_auth if (args_manager.args.share or args_manager.args.listen) and auth_enabled else None,
+    allowed_paths=[modules.config.path_outputs],
+    blocked_paths=[constants.AUTH_FILENAME]
+)