Debug

app_endframe.py

@@ -107,6 +107,9 @@ stream = AsyncStream()
 outputs_folder = './outputs/'
 os.makedirs(outputs_folder, exist_ok=True)
 
+input_video_debug_value = None
+prompt_debug_value = None
+
 # 20250506 pftq: Added function to encode input video frames into latents
 @torch.no_grad()
 def video_encode(video_path, resolution, no_resize, vae, vae_batch_size=16, device="cuda", width=None, height=None):
@@ -667,7 +670,16 @@ def get_duration(input_video, end_frame, end_frame_weight, prompt, n_prompt, see
 
 @spaces.GPU(duration=get_duration)
 def process(input_video, end_frame, end_frame_weight, prompt, n_prompt, seed, batch, resolution, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache, no_resize, mp4_crf, num_clean_frames, vae_batch):
-    global stream, high_vram
+    global stream, high_vram, input_video_debug_value, prompt_debug_value
+
+    if input_video_debug_value is not None:
+        input_video = input_video_debug_value
+        input_video_debug_value = None
+
+    if prompt_debug_value is not None:
+        prompt = prompt_debug_value
+        prompt_debug_value = None
+
     # 20250506 pftq: Updated assertion for video input
     assert input_video is not None, 'No input video!'
 
@@ -741,8 +753,8 @@ with block:
             #example_quick_prompts.click(lambda x: x[0], inputs=[example_quick_prompts], outputs=prompt, show_progress=False, queue=False)
 
             with gr.Row():
-                start_button = gr.Button(value="Start Generation")
-                end_button = gr.Button(value="End Generation", interactive=False)
+                start_button = gr.Button(value="Start Generation", variant="primary")
+                end_button = gr.Button(value="End Generation", variant="stop", interactive=False)
 
             with gr.Group():
                 with gr.Row():
@@ -783,13 +795,18 @@ with block:
 
                 mp4_crf = gr.Slider(label="MP4 Compression", minimum=0, maximum=100, value=16, step=1, info="Lower means better quality. 0 is uncompressed. Change to 16 if you get black outputs. ")
 
+            with gr.Row():
+                input_video_debug = gr.Video(sources='upload', label="Input Video Debug", height=320)
+                prompt_debug = gr.Textbox(label="Prompt Debug", value='')
+
         with gr.Column():
             preview_image = gr.Image(label="Next Latents", height=200, visible=False)
             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')
 
-    gr.Examples(
+    with gr.Row(visible=False):
+        gr.Examples(
         examples = [
                 [
                     "./img_examples/Example1.mp4", # input_video
@@ -830,4 +847,26 @@ with block:
     end_button.click(fn=end_process)
 
 
+    def handle_input_video_debug_upload(input):
+        global input_video_debug_value
+        input_video_debug_value = input
+        return []
+    
+    def handle_prompt_debug_change(input):
+        global prompt_debug_value
+        prompt_debug_value = input
+        return []
+    
+    input_video_debug.upload(
+        fn=handle_input_video_debug_upload,
+        inputs=[input_video_debug],
+        outputs=[]
+    )
+    
+    prompt_debug.change(
+        fn=handle_prompt_debug_change,
+        inputs=[prompt_debug],
+        outputs=[]
+    )
+
 block.launch(share=True)