Parameters

app.py

@@ -38,6 +38,12 @@ MAX_FRAMES_MODEL = 81
 MIN_DURATION = round(MIN_FRAMES_MODEL/FIXED_FPS, 1)
 MAX_DURATION = round(MAX_FRAMES_MODEL/FIXED_FPS, 1)
 
+input_image_debug_value = [None]
+input_video_debug_value = [None]
+end_image_debug_value = [None]
+prompt_debug_value = [None]
+total_second_length_debug_value = [None]
+
 default_negative_prompt = "色调艳丽，过曝，静态，细节模糊不清，字幕，风格，作品，画作，画面，静止，整体发灰，最差质量，低质量，JPEG压缩残留，丑陋的，残缺的，多余的手指，画得不好的手部，画得不好的脸部，畸形的，毁容的，形态畸形的肢体，手指融合，静止不动的画面，杂乱的背景，三条腿，背景人很多，倒着走,过曝，"
 
 print("Loading models into memory. This may take a few minutes...")
@@ -192,6 +198,13 @@ def generate_video(
     Generates a video by interpolating between a start and end image, guided by a text prompt,
     using the diffusers Wan2.2 pipeline.
     """
+    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:
+        start_image_pil = input_image_debug_value[0]
+        end_image_pil = end_image_debug_value[0]
+        prompt = prompt_debug_value[0]
+        duration_seconds = total_second_length_debug_value[0]
+        #allocation_time = min(total_second_length_debug_value[0] * 60 * 100, 600)
+
     if start_image_pil is None or end_image_pil is None:
         raise gr.Error("Please upload both a start and an end image.")
 
@@ -279,6 +292,13 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as app:
                         seed_input = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=42)
                         randomize_seed_checkbox = gr.Checkbox(label="Randomize seed", value=True)
 
+                with gr.Accordion("Debug", open=False):
+                    input_image_debug = gr.Image(type="numpy", label="Image Debug", height=320)
+                    input_video_debug = gr.Video(sources='upload', label="Input Video Debug", height=320)
+                    end_image_debug = gr.Image(type="numpy", label="End Image Debug", height=320)
+                    prompt_debug = gr.Textbox(label="Prompt Debug", value='')
+                    total_second_length_debug = gr.Slider(label="Additional Video Length to Generate (seconds) Debug", minimum=1, maximum=120, value=1, step=0.1)
+
                 generate_button = gr.Button("Generate Video", variant="primary")
 
         with gr.Column():
@@ -319,6 +339,45 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as app:
         outputs=ui_outputs
     )
 
+    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, 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, 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, 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, end_image_debug, prompt_debug, total_second_length_debug],
+        outputs=[]
+    )
+    
     gr.Examples(
         examples = [
             ["poli_tower.png", "tower_takes_off.png", "the man turns around"],