rotate video

tasks.py

@@ -19,7 +19,11 @@ def process_video(video_path: str,vitpose: VitPose,user_id: str):
     
     vitpose.output_video_path = new_file_name
     annotated_frames = vitpose.run(video_path)
-    annotated_video_path = vitpose.frames_to_video(annotated_frames,rotate=True)
+    
+    
+    
+    
+    annotated_video_path = vitpose.frames_to_video(annotated_frames)
     
     logger.info(f"Video processed {annotated_video_path}")
     

vitpose.py

@@ -81,16 +81,17 @@ class VitPose:
 
         return annotated_frame
     
-    def frames_to_video(self, frames, rotate=False):
-        
+    def frames_to_video(self, frames):
         fourcc = cv2.VideoWriter_fourcc(*'mp4v')
         height = self.video_metadata["height"]
         width = self.video_metadata["width"]
         
-        # If rotation is requested, swap dimensions for the output video
+        # Always ensure vertical orientation
+        rotate = width > height  # Rotate only if the video is in landscape mode
+        
+        # For the VideoWriter, we need to specify the dimensions of the output frames
         if rotate:
             print(f"Original dimensions: {width}x{height}, Rotated dimensions: {height}x{width}")
-            # For the VideoWriter, we need to specify the dimensions of the output frames
             out = cv2.VideoWriter(self.output_video_path, fourcc, self.video_metadata["fps"], (height, width))
         else:
             print(f"Dimensions: {width}x{height}")
@@ -98,10 +99,11 @@ class VitPose:
         
         for frame in frames:
             if rotate:
-                # Rotate 90 degrees clockwise
+                # Rotate landscape videos 90 degrees to make them vertical
                 rotated_frame = cv2.rotate(frame, cv2.ROTATE_90_COUNTERCLOCKWISE)
                 out.write(rotated_frame)
             else:
+                # Already vertical, no rotation needed
                 out.write(frame)
             
         out.release()