Update app.py

app.py

@@ -79,45 +79,6 @@ def resize_video(input_vid, output_vid, width, height, fps):
     print(f"RESIZE VIDEO DONE!")
     return output_vid
 
-def normalize_and_save_video(input_video_path, output_video_path):
-    print(f"NORMALIZING ...")
-    cap = cv2.VideoCapture(input_video_path)
-
-    # Get video properties
-    frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
-    width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
-    height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
-    fps = cap.get(cv2.CAP_PROP_FPS)
-
-    # Create VideoWriter object to save the normalized video
-    fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # Specify the codec (e.g., 'mp4v', 'XVID', 'MPEG')
-    out = cv2.VideoWriter(output_video_path, fourcc, fps, (width, height))
-
-    # Iterate through each frame in the video
-    for _ in range(frame_count):
-        ret, frame = cap.read()
-        if not ret:
-            break
-
-        # Convert frame to floating point
-        frame = frame.astype(np.float32)
-
-        # Normalize pixel values to the range [0, 1]
-        frame /= 255.0
-
-        # Convert normalized frame back to 8-bit unsigned integer
-        frame = (frame * 255.0).astype(np.uint8)
-
-        # Write the normalized frame to the output video file
-        out.write(frame)
-
-    # Release the VideoCapture and VideoWriter objects
-    cap.release()
-    out.release()
-
-    print(f"NORMALIZE DONE!")
-    return output_video_path
-
 def make_nearest_multiple_of_32(number):
     remainder = number % 32
     if remainder <= 16:
@@ -163,10 +124,6 @@ def run_inference(prompt, video_path, condition, video_length, seed, steps):
     # if video_length > resized_video_fcount :
     # video_length = int((target_fps * video_length) / original_fps)
     
-
-    # normalize pixels
-    #normalized = normalize_and_save_video(resized, 'normalized.mp4')
-
     output_path = 'output/'
     os.makedirs(output_path, exist_ok=True)
             
@@ -196,6 +153,9 @@ def run_inference(prompt, video_path, condition, video_length, seed, steps):
     #o_height = get_video_dimension(video_path)[1]
     #resize_video(video_path_output, 'resized_final.mp4', o_width, o_height, target_fps)
 
+    # Check generated video FPS
+    gen_fps = get_video_dimension(video_path_output)[2] 
+    print(f"GEN VIDEO FPS: {gen_fps}")
     print(f"FINISHED !")
     return "done", video_path_output