Update app_new.py

app_new.py

@@ -21,7 +21,7 @@ from diffsynth import save_video, ModelManager, SVDVideoPipeline
 from diffsynth import SDVideoPipeline, ControlNetConfigUnit, VideoData, save_frames
 from diffsynth.extensions.RIFE import RIFESmoother
 
-
+import cv2
 
 # Constants
 MAX_SEED = np.iinfo(np.int32).max
@@ -74,7 +74,13 @@ def update_frames(video_in):
     up_video = VideoData(
             video_file=video_in)
     frame_len = len(up_video)
-    return gr.update(maximum=frame_len)
+    video_path = video_in
+    cap = cv2.VideoCapture(video_path)
+    fps_in = cap.get(cv2.CAP_PROP_FPS)
+    width_in = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
+    height_in = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
+    cap.release()
+    return gr.update(maximum=frame_len), gr.update(value=fps_in), gr.update(value=width_in), gr.update(value=height_in)
 
 @spaces.GPU(duration=180)
 def generate(
@@ -227,7 +233,7 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
         cache_examples="lazy",
         examples_per_page=4,
     )
-    video_in.upload(update_frames, inputs=[video_in], outputs=[num_frames])
+    video_in.upload(update_frames, inputs=[video_in], outputs=[num_frames, fps_id, width, height])
     submit_event = submit_btn.click(fn=generate, inputs=[video_in, image_in, prompt, seed, num_inference_steps, num_frames, height, width, animatediff_batch_size, animatediff_stride, fps_id], outputs=[video, seed], api_name="video")
     #stop_btn.click(fn=None, inputs=None, outputs=None, cancels=[submit_event])