Update webui.py

webui.py

@@ -642,16 +642,33 @@ with shared.gradio_root:
         metadata_import_button.click(trigger_metadata_import, inputs=[metadata_input_image, state_is_generating], outputs=load_data_outputs, queue=False, show_progress=True) \
             .then(style_sorter.sort_styles, inputs=style_selections, outputs=style_selections, queue=False, show_progress=False)
 
-        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]) \
+        # Inisialisasi currentTask menjadi None saat diluncurkan
+        currentTask = None
+
+        # Saat tombol Generate diklik
+        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=set_current_task, 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(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) \
             .then(fn=lambda: None, _js='playNotification').then(fn=lambda: None, _js='refresh_grid_delayed')
 
+        def set_current_task():
+            global currentTask
+            currentTask = gr.State(worker.AsyncTask(args=[]))
+
         for notification_file in ['notification.ogg', 'notification.mp3']:
             if os.path.exists(notification_file):
                 gr.Audio(interactive=False, value=notification_file, elem_id='audio_notification', visible=False)
@@ -677,9 +694,6 @@ def dump_default_english_config():
 
 # dump_default_english_config()
 
-# Ketika launch(), currentTask dinonaktifkan
-currentTask.set(None)
-
 shared.gradio_root.launch(
     inbrowser=args_manager.args.in_browser,
     server_name=args_manager.args.listen,
@@ -689,7 +703,3 @@ shared.gradio_root.launch(
     allowed_paths=[modules.config.path_outputs],
     blocked_paths=[constants.AUTH_FILENAME],
 )
-
-# Ketika tombol diklik, currentTask diaktifkan
-generate_button.click(lambda: (currentTask.set(worker.AsyncTask(args=[])), 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])