Fix time

app.py

@@ -108,6 +108,8 @@ stream = AsyncStream()
 outputs_folder = './outputs/'
 os.makedirs(outputs_folder, exist_ok=True)
 
+input_image_debug_value = input_video_debug_value = prompt_debug_value = total_second_length_debug_value = None
+
 default_local_storage = {
         "generation-mode": "image",
     }
@@ -516,7 +518,11 @@ def worker(input_image, prompts, n_prompt, seed, resolution, total_second_length
     return
 
 def get_duration(input_image, prompt, generation_mode, n_prompt, randomize_seed, seed, resolution, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, enable_preview, use_teacache, mp4_crf):
-    return total_second_length * 60 * (0.7 if use_teacache else 1.3) * (2**((resolution - 640) / 640)) * (1 + ((steps - 25) / 100))
+    global total_second_length_debug_value
+
+    if total_second_length_debug_value is not None:
+        return min(total_second_length_debug_value * 60 * 10, 600)
+    return total_second_length * 60 * (0.9 if use_teacache else 1.5) * (2**((resolution - 640) / 640)) * (1 + ((steps - 25) / 100))
 
 
 @spaces.GPU(duration=get_duration)
@@ -537,13 +543,19 @@ def process(input_image, prompt,
             use_teacache=False,
             mp4_crf=16
            ):
-    global stream
+    global stream, input_image_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 gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update()
         return
 
+    if input_image_debug_value is not None or prompt_debug_value is not None or total_second_length_debug_value is not None:
+        input_image = input_image_debug_value
+        prompt = prompt_debug_value
+        total_second_length = total_second_length_debug_value
+        input_image_debug_value = prompt_debug_value = total_second_length_debug_value = None
+
     if randomize_seed:
         seed = random.randint(0, np.iinfo(np.int32).max)
 
@@ -682,6 +694,10 @@ def worker_video(input_video, prompts, n_prompt, seed, batch, resolution, total_
             history_pixels = None
             previous_video = None
 
+            # 20250507 pftq: hot fix for initial video being corrupted by vae encoding, issue with ghosting because of slight differences
+            #history_pixels = input_video_pixels
+            #save_bcthw_as_mp4(vae_decode(video_latents, vae).cpu(), os.path.join(outputs_folder, f'{job_id}_input_video.mp4'), fps=fps, crf=mp4_crf) # 20250507 pftq: test fast movement corrupted by vae encoding if vae batch size too low
+
             for section_index in range(total_latent_sections):
                 if stream.input_queue.top() == 'end':
                     stream.output_queue.push(('end', None))
@@ -844,18 +860,27 @@ def worker_video(input_video, prompts, n_prompt, seed, batch, resolution, total_
     return
 
 def get_duration_video(input_video, prompt, n_prompt, randomize_seed, 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):
-    return total_second_length * 60 * (0.7 if use_teacache else 2) * (2**((resolution - 640) / 640)) * (1 + ((steps - 25) / 100))
+    global total_second_length_debug_value
+    if total_second_length_debug_value is not None:
+        return min(total_second_length_debug_value * 60 * 10, 600)
+    return total_second_length * 60 * (0.9 if use_teacache else 2.3) * (2**((resolution - 640) / 640)) * (1 + ((steps - 25) / 100))
 
 # 20250506 pftq: Modified process to pass clean frame count, etc from video_encode
 @spaces.GPU(duration=get_duration_video)
 def process_video(input_video, prompt, n_prompt, randomize_seed, 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):
-    global stream, high_vram
+    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 gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update()
         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)
 
@@ -1039,6 +1064,12 @@ with block:
                     randomize_seed = gr.Checkbox(label='Randomize seed', value=True, info='If checked, the seed is always different')
                     seed = gr.Slider(label="Seed", minimum=0, maximum=np.iinfo(np.int32).max, step=1, randomize=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)
+                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)
+
         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)
@@ -1167,7 +1198,7 @@ with block:
                     True, # randomize_seed
                     42, # seed
                     1, # batch
-                    640, # resolution
+                    672, # resolution
                     1, # total_second_length
                     9, # latent_window_size
                     35, # steps
@@ -1199,7 +1230,7 @@ with block:
                     "Missing arm, unrealistic position, impossible contortion, blurred, blurry", # n_prompt
                     True, # randomize_seed
                     42, # seed
-                    640, # resolution
+                    672, # resolution
                     1, # total_second_length
                     9, # latent_window_size
                     25, # steps
@@ -1228,7 +1259,7 @@ with block:
                     True, # randomize_seed
                     42, # seed
                     1, # batch
-                    640, # resolution
+                    672, # resolution
                     1, # total_second_length
                     9, # latent_window_size
                     25, # steps
@@ -1260,12 +1291,44 @@ with block:
             return [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)]
 
     
+    def handle_field_debug_change(input_image_debug_data, input_video_debug_data, prompt_debug_data, total_second_length_debug_data):
+        global input_image_debug_value, input_video_debug_value, prompt_debug_value, total_second_length_debug_value
+        input_image_debug_value = input_image_debug_data
+        input_video_debug_value = input_video_debug_data
+        prompt_debug_value = prompt_debug_data
+        total_second_length_debug_value = total_second_length_debug_data
+        return []
+    
     generation_mode.change(
         fn=handle_generation_mode_change,
         inputs=[generation_mode],
         outputs=[text_to_video_hint, input_image, input_video, start_button, start_button_video, no_resize, batch, num_clean_frames, vae_batch]
     )
 
+    input_image_debug.upload(
+        fn=handle_field_debug_change,
+        inputs=[input_image_debug, input_video_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],
+        outputs=[]
+    )
+    
+    prompt_debug.change(
+        fn=handle_field_debug_change,
+        inputs=[input_image_debug, input_video_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],
+        outputs=[]
+    )
+
     # Update display when the page loads
     block.load(
         fn=handle_generation_mode_change, inputs = [