block = gr.Blocks(css=css, js=js).queue()

app.py

@@ -946,51 +946,6 @@ def refresh_prompt():
     print(str(array))
     return ";".join(array)
 
-import time
-@spaces.GPU(duration=get_duration_video)
-def process_yield(input_image, final_prompt, generation_mode, n_prompt, randomize_seed, seed, resolution, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache, mp4_crf):
-    global stream, high_vram, input_video_debug_value, prompt_debug_value, total_second_length_debug_value
-
-    if torch.cuda.device_count() == 0:
-        gr.Warning('Set this space to GPU config to make it work.')
-        yield None, None, None, None, None, None
-        return
-
-    if input_video_debug_value is not None or prompt_debug_value is not None or total_second_length_debug_value is not None:
-        input_video = input_video_debug_value
-        prompt = prompt_debug_value
-        total_second_length = total_second_length_debug_value
-        input_video_debug_value = prompt_debug_value = total_second_length_debug_value = None
-
-    if randomize_seed:
-        seed = random.randint(0, np.iinfo(np.int32).max)
-
-    prompts = prompt.split(";")
-
-    # 20250506 pftq: Updated assertion for video input
-    assert input_video is not None, 'No input video!'
-
-    yield None, None, '', '', gr.update(interactive=False), gr.update(interactive=True)
-
-    # 20250507 pftq: Even the H100 needs offloading if the video dimensions are 720p or higher
-    if high_vram and (no_resize or resolution>640):
-        print("Disabling high vram mode due to no resize and/or potentially higher resolution...")
-        high_vram = False
-        vae.enable_slicing()
-        vae.enable_tiling()
-        DynamicSwapInstaller.install_model(transformer, device=gpu)
-        DynamicSwapInstaller.install_model(text_encoder, device=gpu)
-
-    # 20250508 pftq: automatically set distilled cfg to 1 if cfg is used
-    if cfg > 1:
-        gs = 1
-
-    for i in range(3):
-        time.sleep(1)
-        yield (gr.update(), gr.update(), "In progress..." + str(i), gr.update(), gr.update(), gr.update(interactive=True))
-    yield (gr.update(), gr.update(), "Finished", gr.update(), gr.update(), gr.update(interactive=False))
-    return (gr.update(), gr.update(), "Finished", gr.update(), gr.update(), gr.update(interactive=False))
-
 title_html = """
     <h1><center>FramePack</center></h1>
     <big><center>Generate videos from text/image/video freely, without account, without watermark and download it</center></big>
@@ -1000,7 +955,15 @@ title_html = """
     """
 
 css = make_progress_bar_css()
-block = gr.Blocks(css=css).queue()
+js = """
+window.addEventListener("beforeunload", function (e) {
+    var confirmationMessage = 'It looks like you have been editing something. If you leave before saving, your changes will be lost.';
+
+    (e || window.event).returnValue = confirmationMessage;
+    return confirmationMessage;
+});
+"""
+block = gr.Blocks(css=css, js=js).queue()
 with block:
     if torch.cuda.device_count() == 0:
         with gr.Row():
@@ -1093,8 +1056,6 @@ with block:
             result_video = gr.Video(label="Finished Frames", autoplay=True, show_share_button=False, height=512, loop=True)
             progress_desc = gr.Markdown('', elem_classes='no-generating-animation')
             progress_bar = gr.HTML('', elem_classes='no-generating-animation')
-            
-            run_button = gr.Button("Run", variant="primary")
 
     # 20250506 pftq: Updated inputs to include num_clean_frames
     ips = [input_image, final_prompt, generation_mode, n_prompt, randomize_seed, seed, resolution, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache, mp4_crf]
@@ -1120,8 +1081,6 @@ with block:
             raise gr.Error("Please provide a video to extend.")
         return [gr.update(interactive=True)]
 
-    run_button.click(fn=process_yield, inputs=ips, outputs=[result_video, preview_image, progress_desc, progress_bar, start_button, end_button])
-
     prompt_number.change(fn=handle_prompt_number_change, inputs=[], outputs=[])
     timeless_prompt.change(fn=handle_timeless_prompt_change, inputs=[timeless_prompt], outputs=[final_prompt])
     start_button.click(fn = check_parameters, inputs = [
@@ -1233,7 +1192,7 @@ with block:
                 ],
             ],
         run_on_click = True,
-        fn = process_yield,
+        fn = process_video,
 	    inputs = ips_video,
 	    outputs = [result_video, preview_image, progress_desc, progress_bar, start_button_video, end_button],
         cache_examples = torch.cuda.device_count() > 0,