Ask before leaving

app.py

@@ -360,7 +360,25 @@ def generate_video_on_gpu(
 
 # --- 3. Gradio User Interface ---
 
-with gr.Blocks() as app:
+
+
+js = """
+function createGradioAnimation() {
+    window.addEventListener("beforeunload", function(e) {
+        if (document.getElementById('dummy_button_id') && !document.getElementById('dummy_button_id').disabled) {
+            var confirmationMessage = 'A process is still running. '
+                                    + 'If you leave before saving, your changes will be lost.';
+    
+            (e || window.event).returnValue = confirmationMessage;
+        }
+        return confirmationMessage;
+    });
+    return 'Animation created';
+}
+"""
+
+# Gradio interface
+with gr.Blocks(js=js) as app:
     gr.Markdown("# Wan 2.2 First/Last Frame Video Fast")
     gr.Markdown("Based on the [Wan 2.2 First/Last Frame workflow](https://www.reddit.com/r/StableDiffusion/comments/1me4306/psa_wan_22_does_first_frame_last_frame_out_of_the/), applied to 🧨 Diffusers + [lightx2v/Wan2.2-Lightning](https://huggingface.co/lightx2v/Wan2.2-Lightning) 8-step LoRA")