Update app.py

app.py

@@ -87,6 +87,38 @@ def make_nearest_multiple_of_32(number):
         number += 32 - remainder
     return number 
 
+def change_video_fps(input_file):
+    print(f"CHANGING FIANL OUTPUT FPS")
+    cap = cv2.VideoCapture(input_path)
+    # Check if the final file already exists
+    if os.path.exists('output_video.mp4'):
+        # Delete the existing file
+        os.remove('output_video.mp4')
+    output_path = 'output_video.mp4'
+    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+    output_fps = 12
+    output_size = (int(cap.get(cv2.CAP_PROP_FRAME_WIDTH)), int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT)))
+    out = cv2.VideoWriter(output_path, fourcc, output_fps, output_size)
+    
+    frame_count = 0
+    while cap.isOpened():
+        ret, frame = cap.read()
+        if not ret:
+            break
+        
+        # Write the current frame to the output video multiple times to increase the frame rate
+        for _ in range(output_fps // 8):
+            out.write(frame)
+        
+        frame_count += 1
+        print(f'Processed frame {frame_count}')
+
+    cap.release()
+    out.release()
+    cv2.destroyAllWindows()
+
+    return 'output_video.mp4'
+
 def run_inference(prompt, video_path, condition, video_length, seed, steps):
     
     seed = math.floor(seed)
@@ -156,8 +188,9 @@ def run_inference(prompt, video_path, condition, video_length, seed, steps):
     # Check generated video FPS
     gen_fps = get_video_dimension(video_path_output)[2] 
     print(f"GEN VIDEO FPS: {gen_fps}")
+    final = change_video_fps(video_path_output)
     print(f"FINISHED !")
-    return "done", video_path_output
+    return "done", final
  
 
 css="""