Generation mode

app.py

@@ -1622,7 +1622,7 @@ def process_video_on_gpu(input_video, end_frame, prompts, generation_mode, n_pro
     stream = AsyncStream()
 
     # 20250506 pftq: Pass num_clean_frames, vae_batch, etc
-    async_run(worker_video if generation_mode == "video" else worker_video_end, input_video, end_frame, prompts, n_prompt, seed, batch, resolution, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, enable_preview, use_teacache, no_resize, mp4_crf, num_clean_frames, vae_batch)
+    async_run(worker_video_end if generation_mode == "video_end" else worker_video, input_video, end_frame, prompts, n_prompt, seed, batch, resolution, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, enable_preview, use_teacache, no_resize, mp4_crf, num_clean_frames, vae_batch)
 
     output_filename = None
 
@@ -1692,7 +1692,7 @@ def process_video(input_video, end_frame, prompt, generation_mode, n_prompt, ran
     if cfg > 1:
         gs = 1
 
-    yield from process_video_on_gpu(input_video, prompts, generation_mode, n_prompt, seed, batch, resolution, total_second_length, allocation_time, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, enable_preview, use_teacache, no_resize, mp4_crf, num_clean_frames, vae_batch)
+    yield from process_video_on_gpu(input_video, end_frame, prompt, generation_mode, n_prompt, seed, batch, resolution, total_second_length, allocation_time, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, enable_preview, use_teacache, no_resize, mp4_crf, num_clean_frames, vae_batch)
 
 def end_process():
     stream.input_queue.push('end')
@@ -1765,9 +1765,9 @@ with block:
             generation_mode = gr.Radio([["Text-to-Video", "text"], ["Image-to-Video", "image"], ["Start & end frames", "start_end"], ["Video Extension", "video"], ["Video & end frame", "video_end"]], elem_id="generation-mode", label="Input mode", value = "image")
             text_to_video_hint = gr.HTML("Text-to-Video badly works with a flash effect at the start. I discourage to use the Text-to-Video feature. You should rather generate an image with Flux and use Image-to-Video. You will save time.")
             input_image = gr.Image(sources='upload', type="numpy", label="Image", height=320)
-            end_image = gr.Image(sources='upload', type="numpy", label="End Frame (optional)", height=320)
             image_position = gr.Slider(label="Image position", minimum=0, maximum=100, value=0, step=1, info='0=Video start; 100=Video end (lower quality)')
             input_video = gr.Video(sources='upload', label="Input Video", height=320)
+            end_image = gr.Image(sources='upload', type="numpy", label="End Frame (optional)", height=320)
             timeless_prompt = gr.Textbox(label="Timeless prompt", info='Used on the whole duration of the generation', value='', placeholder="The creature starts to move, fast motion, fixed camera, focus motion, consistent arm, consistent position, mute colors, insanely detailed")
             prompt_number = gr.Slider(label="Timed prompt number", minimum=0, maximum=1000, value=0, step=1, info='Prompts will automatically appear')
 
@@ -2550,6 +2550,22 @@ with block:
             gr.update(visible = True),  # prompt_hint
             gr.update(visible = False)  # fps_number
             ]
+        elif generation_mode_data == "video_end":
+            return [
+            gr.update(visible = False), # text_to_video_hint
+            gr.update(visible = False), # image_position
+            gr.update(visible = False), # input_image
+            gr.update(visible = True), # end_image
+            gr.update(visible = True),  # input_video
+            gr.update(visible = False), # start_button
+            gr.update(visible = True),  # start_button_video
+            gr.update(visible = True),  # no_resize
+            gr.update(visible = True),  # batch
+            gr.update(visible = True),  # num_clean_frames
+            gr.update(visible = True),  # vae_batch
+            gr.update(visible = True),  # prompt_hint
+            gr.update(visible = False)  # fps_number
+            ]
 
     def handle_field_debug_change(input_image_debug_data, input_video_debug_data, end_image_debug_data, prompt_debug_data, total_second_length_debug_data):
         print("handle_field_debug_change")