Update webui.py

webui.py

@@ -645,6 +645,10 @@ with shared.gradio_root:
         # Inisialisasi currentTask menjadi None saat diluncurkan
         currentTask = None
 
+        def set_current_task():
+            global currentTask
+            currentTask = gr.State(worker.AsyncTask(args=[]))
+            
         # Saat tombol Generate diklik
         generate_button.click(lambda: (
             gr.update(visible=True, interactive=True),
@@ -665,10 +669,6 @@ with shared.gradio_root:
             .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)