Merge pull request #830 from elebumm/fix/moviepy-length-checker

Check clip length using moviepy

TTS/engine_wrapper.py

@@ -4,9 +4,9 @@ from typing import Tuple
 import re
 from os import getenv
 
-import sox
-from mutagen import MutagenError
-from mutagen.mp3 import MP3, HeaderNotFoundError
+# import sox
+# from mutagen import MutagenError
+# from mutagen.mp3 import MP3, HeaderNotFoundError
 import translators as ts
 from rich.progress import track
 from moviepy.editor import AudioFileClip, CompositeAudioClip, concatenate_audioclips
@@ -94,14 +94,12 @@ class TTSEngine:
 
     def call_tts(self, filename: str, text: str):
         self.tts_module.run(text=process_text(text), filepath=f"{self.path}/{filename}.mp3")
-        try:
-            try:
-                self.length += MP3(f"{self.path}/{filename}.mp3").info.length
-            except (MutagenError, HeaderNotFoundError):
-                self.length += sox.file_info.duration(f"{self.path}/{filename}.mp3")
-        except Exception: # last resort backup
-            clip = AudioFileClip(f"{self.path}/{filename}.mp3")
-            self.length += clip.duration
+        # try:
+        #     self.length += MP3(f"{self.path}/{filename}.mp3").info.length
+        # except (MutagenError, HeaderNotFoundError):
+        #     self.length += sox.file_info.duration(f"{self.path}/{filename}.mp3")
+        clip = AudioFileClip(f"{self.path}/{filename}.mp3")
+        self.length += clip.duration
 
 def process_text(text: str):
     lang = getenv("POSTLANG", "")

main.py

@@ -1,5 +1,5 @@
 #!/usr/bin/env python
-
+import math
 from subprocess import Popen
 from os import getenv, name
 from dotenv import load_dotenv
@@ -35,6 +35,7 @@ def main(POST_ID=None):
     cleanup()
     reddit_object = get_subreddit_threads(POST_ID)
     length, number_of_comments = save_text_to_mp3(reddit_object)
+    length = math.ceil(length)
     download_screenshots_of_reddit_posts(reddit_object, number_of_comments)
     download_background()
     chop_background_video(length)

video_creation/final_video.py

@@ -51,13 +51,7 @@ def make_final_video(number_of_clips: int, length: int, reddit_obj: dict):
     audio_concat = concatenate_audioclips(audio_clips)
     audio_composite = CompositeAudioClip([audio_concat])
 
-    # Get sum of all clip lengths
-    total_length = sum([clip.duration for clip in audio_clips])
-    # round total_length to an integer
-    int_total_length = round(total_length)
-    # Output Length
-
-    console.log(f"[bold green] Video Will Be: {int_total_length} Seconds Long")
+    console.log(f"[bold green] Video Will Be: {length} Seconds Long")
     # add title to video
     image_clips = []
     # Gather all images
@@ -114,7 +108,7 @@ def make_final_video(number_of_clips: int, length: int, reddit_obj: dict):
         threads=multiprocessing.cpu_count(),
     )
     ffmpeg_tools.ffmpeg_extract_subclip(
-        "assets/temp/temp.mp4", 0, length, targetname=f"results/{subreddit}/{filename}"
+        "assets/temp/temp.mp4", 0, final.duration, targetname=f"results/{subreddit}/{filename}"
     )
     # os.remove("assets/temp/temp.mp4")