rotation

Author: jordimas

open_dubbing/video_processing.py

@@ -19,6 +19,8 @@
 
 from moviepy import AudioFileClip, VideoFileClip, concatenate_videoclips
 
+from open_dubbing import logger
+
 _DEFAULT_FPS: Final[int] = 30
 _DEFAULT_DUBBED_VIDEO_FILE: Final[str] = "dubbed_video"
 _DEFAULT_OUTPUT_FORMAT: Final[str] = ".mp4"
@@ -40,6 +42,11 @@ def split_audio_video(*, video_file: str, output_directory: str) -> tuple[str, s
             video_clip_without_audio = video_clip.with_audio(None)
             fps = video_clip.fps or _DEFAULT_FPS
 
+            rotation_angle = getattr(video_clip, "rotation", 0)
+            if rotation_angle in [90, 180, 270]:
+                video_clip = video_clip.rotated(rotation_angle)
+                logger().info(f"Keep rotation angle {rotation_angle} for video")
+
             video_output_file = os.path.join(output_directory, filename + "_video.mp4")
             video_clip_without_audio.write_videofile(
                 video_output_file, codec="libx264", fps=fps, logger=None