Add more code

app.py

@@ -930,12 +930,13 @@ def process(input_image,
     if auto_allocation:
         allocation_time = min(total_second_length * 60 * (1.5 if use_teacache else 3.0) * (1 + ((steps - 25) / 25))**2, 600)
 
-    if input_image_debug_value[0] is not None or prompt_debug_value[0] is not None or total_second_length_debug_value[0] is not None:
+    if input_image_debug_value[0] is not None or end_image_debug_value[0] is not None or prompt_debug_value[0] is not None or total_second_length_debug_value[0] is not None:
         input_image = input_image_debug_value[0]
+        end_image = end_image_debug_value[0]
         prompt = prompt_debug_value[0]
         total_second_length = total_second_length_debug_value[0]
         allocation_time = min(total_second_length_debug_value[0] * 60 * 100, 600)
-        input_image_debug_value[0] = prompt_debug_value[0] = total_second_length_debug_value[0] = None
+        input_image_debug_value[0] = end_image_debug_value[0] = input_video_debug_value[0] = prompt_debug_value[0] = total_second_length_debug_value[0] = None
 
     if torch.cuda.device_count() == 0:
         gr.Warning('Set this space to GPU config to make it work.')
@@ -1127,7 +1128,7 @@ with block:
     local_storage = gr.BrowserState(default_local_storage)
     with gr.Row():
         with gr.Column():
-            generation_mode = gr.Radio([["Text-to-Video", "text"], ["Image-to-Video", "image"], ["Video Extension", "video"]], elem_id="generation-mode", label="Generation mode", value = "image")
+            generation_mode = gr.Radio([["Text-to-Video", "text"], ["Image-to-Video", "image"], ["Start & end frames", "start_end"], ["Video Extension", "video"]], elem_id="generation-mode", label="Generation 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)
             image_position = gr.Slider(label="Image position", minimum=0, maximum=100, value=0, step=1, info='0=Video start; 100=Video end (lower quality)')
@@ -1666,48 +1667,114 @@ with block:
     def check_parameters(generation_mode, input_image, input_video):
         if generation_mode == "image" and input_image is None:
             raise gr.Error("Please provide an image to extend.")
+        if generation_mode == "start_end" and input_image is None:
+            raise gr.Error("Please provide an image to extend.")
         if generation_mode == "video" and input_video is None:
             raise gr.Error("Please provide a video to extend.")
         return [gr.update(interactive=True), gr.update(visible = True)]
     
     def handle_generation_mode_change(generation_mode_data):
         if generation_mode_data == "text":
-            return [gr.update(visible = True), gr.update(visible = False), gr.update(visible = False), gr.update(visible = False), gr.update(visible = True), gr.update(visible = False), gr.update(visible = False), gr.update(visible = False), gr.update(visible = False), gr.update(visible = False), gr.update(visible = False), gr.update(visible = True)]
+            return [
+            gr.update(visible = True),  # text_to_video_hint
+            gr.update(visible = False), # image_position
+            gr.update(visible = False), # input_image
+            gr.update(visible = False), # end_image
+            gr.update(visible = False), # input_video
+            gr.update(visible = True),  # start_button
+            gr.update(visible = False), # start_button_video
+            gr.update(visible = False), # no_resize
+            gr.update(visible = False), # batch
+            gr.update(visible = False), # num_clean_frames
+            gr.update(visible = False), # vae_batch
+            gr.update(visible = False), # prompt_hint
+            gr.update(visible = True)   # fps_number
+            ]
         elif generation_mode_data == "image":
-            return [gr.update(visible = False), gr.update(visible = True), gr.update(visible = True), gr.update(visible = False), gr.update(visible = True), gr.update(visible = False), gr.update(visible = False), gr.update(visible = False), gr.update(visible = False), gr.update(visible = False), gr.update(visible = False), gr.update(visible = True)]
+            return [
+            gr.update(visible = False), # text_to_video_hint
+            gr.update(visible = True),  # image_position
+            gr.update(visible = True),  # input_image
+            gr.update(visible = False), # end_image
+            gr.update(visible = False), # input_video
+            gr.update(visible = True),  # start_button
+            gr.update(visible = False), # start_button_video
+            gr.update(visible = False), # no_resize
+            gr.update(visible = False), # batch
+            gr.update(visible = False), # num_clean_frames
+            gr.update(visible = False), # vae_batch
+            gr.update(visible = False), # prompt_hint
+            gr.update(visible = True)   # fps_number
+            ]
+        elif generation_mode_data == "start_end":
+            return [
+            gr.update(visible = False), # text_to_video_hint
+            gr.update(visible = False), # image_position
+            gr.update(visible = True),  # input_image
+            gr.update(visible = True),  # end_image
+            gr.update(visible = False), # input_video
+            gr.update(visible = True),  # start_button
+            gr.update(visible = False), # start_button_video
+            gr.update(visible = False), # no_resize
+            gr.update(visible = False), # batch
+            gr.update(visible = False), # num_clean_frames
+            gr.update(visible = False), # vae_batch
+            gr.update(visible = False), # prompt_hint
+            gr.update(visible = True)   # fps_number
+            ]
         elif generation_mode_data == "video":
-            return [gr.update(visible = False), gr.update(visible = False), gr.update(visible = False), gr.update(visible = True), gr.update(visible = False), gr.update(visible = True), gr.update(visible = True), gr.update(visible = True), gr.update(visible = True), gr.update(visible = True), gr.update(visible = True), gr.update(visible = False)]
-
-    
-    def handle_field_debug_change(input_image_debug_data, input_video_debug_data, prompt_debug_data, total_second_length_debug_data):
+            return [
+            gr.update(visible = False), # text_to_video_hint
+            gr.update(visible = False), # image_position
+            gr.update(visible = False), # input_image
+            gr.update(visible = False), # 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")
         input_image_debug_value[0] = input_image_debug_data
         input_video_debug_value[0] = input_video_debug_data
+        end_image_debug_value[0] = end_image_debug_data
         prompt_debug_value[0] = prompt_debug_data
         total_second_length_debug_value[0] = total_second_length_debug_data
         return []
 
     input_image_debug.upload(
         fn=handle_field_debug_change,
-        inputs=[input_image_debug, input_video_debug, prompt_debug, total_second_length_debug],
+        inputs=[input_image_debug, input_video_debug, end_image_debug, prompt_debug, total_second_length_debug],
         outputs=[]
     )
     
     input_video_debug.upload(
         fn=handle_field_debug_change,
-        inputs=[input_image_debug, input_video_debug, prompt_debug, total_second_length_debug],
+        inputs=[input_image_debug, input_video_debug, end_image_debug, prompt_debug, total_second_length_debug],
+        outputs=[]
+    )
+    
+    end_image_debug.upload(
+        fn=handle_field_debug_change,
+        inputs=[input_image_debug, input_video_debug, end_image_debug, prompt_debug, total_second_length_debug],
         outputs=[]
     )
     
     prompt_debug.change(
         fn=handle_field_debug_change,
-        inputs=[input_image_debug, input_video_debug, prompt_debug, total_second_length_debug],
+        inputs=[input_image_debug, input_video_debug, end_image_debug, prompt_debug, total_second_length_debug],
         outputs=[]
     )
     
     total_second_length_debug.change(
         fn=handle_field_debug_change,
-        inputs=[input_image_debug, input_video_debug, prompt_debug, total_second_length_debug],
+        inputs=[input_image_debug, input_video_debug, end_image_debug, prompt_debug, total_second_length_debug],
         outputs=[]
     )
     
@@ -1731,7 +1798,7 @@ with block:
     generation_mode.change(
         fn=handle_generation_mode_change,
         inputs=[generation_mode],
-        outputs=[text_to_video_hint, image_position, input_image, input_video, start_button, start_button_video, no_resize, batch, num_clean_frames, vae_batch, prompt_hint, fps_number]
+        outputs=[text_to_video_hint, image_position, input_image, end_image, input_video, start_button, start_button_video, no_resize, batch, num_clean_frames, vae_batch, prompt_hint, fps_number]
     )
 
     # Update display when the page loads
@@ -1739,7 +1806,7 @@ with block:
         fn=handle_generation_mode_change, inputs = [
         generation_mode
     ], outputs = [
-       text_to_video_hint, image_position, input_image, input_video, start_button, start_button_video, no_resize, batch, num_clean_frames, vae_batch, prompt_hint, fps_number
+       text_to_video_hint, image_position, input_image, end_image, input_video, start_button, start_button_video, no_resize, batch, num_clean_frames, vae_batch, prompt_hint, fps_number
     ]
     )