From a9775f348af808f5f2c27cd41003f78692b23557 Mon Sep 17 00:00:00 2001
From: Drugsosos <44712637+Drugsosos@users.noreply.github.com>
Date: Wed, 13 Jul 2022 00:36:02 +0300
Subject: [PATCH] cherry-picked commit with TTS/pictures offsets from
 async-tts-api

---
 video_creation/final_video.py | 72 ++++++++++++++++++++++++-----------
 1 file changed, 49 insertions(+), 23 deletions(-)

diff --git a/video_creation/final_video.py b/video_creation/final_video.py
index 2dba251..1145e9f 100755
--- a/video_creation/final_video.py
+++ b/video_creation/final_video.py
@@ -28,6 +28,10 @@ console = Console()
 W, H = 1080, 1920  # TODO move to config
 
 max_length: int = 50  # TODO move to config
+time_before_first_picture: float = 1  # TODO move to config
+time_before_tts: float = 0.5  # TODO move to config
+time_between_pictures: float = 1  # TODO move to config
+delay_before_end: int = 1  # TODO move to config
 
 
 def name_normalize(
@@ -61,7 +65,7 @@ def make_final_video(
     Gathers audio clips, gathers all screenshots, stitches them together and saves the final video to assets/temp
 
     Args:
-        indexes_of_clips (list): Indexes with created comments'
+        indexes_of_clips (list): Indexes of voiced comments
         reddit_obj (dict): The reddit object that contains the posts to read.
         background_config (Tuple[str, str, str, Any]): The background config to use.
     """
@@ -83,28 +87,31 @@ def make_final_video(
 
     # Gather all audio clips
     audio_clips = list()
+    correct_audio_offset = time_before_tts * 2 + time_between_pictures
 
     audio_title = create_audio_clip(
         'title',
-        0,
+        time_before_first_picture + time_before_tts,
     )
-    video_duration += audio_title.duration
+    video_duration += audio_title.duration + time_before_first_picture + time_before_tts
     audio_clips.append(audio_title)
     indexes_for_videos = list()
 
-    for idx in track(
-            indexes_of_clips,
+    for idx, audio in track(
+            enumerate(indexes_of_clips, start=1),
             description='Gathering audio clips...',
     ):
         temp_audio_clip = create_audio_clip(
-            idx,
-            video_duration,
+            audio,
+            correct_audio_offset + video_duration,
         )
-        if video_duration + temp_audio_clip.duration <= max_length:
-            video_duration += temp_audio_clip.duration
+        if video_duration + temp_audio_clip.duration + correct_audio_offset + delay_before_end <= max_length:
+            video_duration += temp_audio_clip.duration + correct_audio_offset
             audio_clips.append(temp_audio_clip)
             indexes_for_videos.append(idx)
 
+    video_duration += delay_before_end
+
     audio_composite = concatenate_audioclips(audio_clips)
 
     console.log('[bold green] Video Will Be: %.2f Seconds Long' % video_duration)
@@ -120,9 +127,9 @@ def make_final_video(
     ) -> 'ImageClip':
         return (
             ImageClip(f'assets/temp/png/{image_title}.png')
-            .set_start(audio_start)
-            .set_end(audio_end)
-            .set_duration(audio_duration, change_end=False)
+            .set_start(audio_start - time_before_tts)
+            .set_end(audio_end + time_before_tts)
+            .set_duration(time_before_tts * 2 + audio_duration, change_end=False)
             .set_opacity(new_opacity)
             .resize(width=W - 100)
         )
@@ -139,14 +146,13 @@ def make_final_video(
         )
     )
 
-    for photo_idx in range(indexes_for_videos.__len__()):
+    for photo_idx in indexes_for_videos:
         image_clips.append(
             create_image_clip(
-                f'comment_{indexes_for_videos[photo_idx]}',
-                # + title clip
-                audio_clips[photo_idx + 1].start,
-                audio_clips[photo_idx + 1].end,
-                audio_clips[photo_idx + 1].duration
+                f'comment_{indexes_of_clips[photo_idx]}',
+                audio_clips[photo_idx].start,
+                audio_clips[photo_idx].end,
+                audio_clips[photo_idx].duration
             )
         )
 
@@ -169,20 +175,40 @@ def make_final_video(
     background_clip = (
         VideoFileClip('assets/temp/background.mp4')
         .set_start(0)
-        .set_end(video_duration)
+        .set_end(video_duration + delay_before_end)
         .without_audio()
         .resize(height=H)
-        .crop(x1=1166.6, y1=0, x2=2246.6, y2=1920)
     )
 
+    back_video_width, back_video_height = background_clip.size
+
+    # Fix for crop with vertical videos
+    if back_video_width < H:
+        background_clip = (
+            background_clip
+            .resize(width=W)
+        )
+        back_video_width, back_video_height = background_clip.size
+        background_clip = background_clip.crop(
+            x1=0,
+            x2=back_video_width,
+            y1=back_video_height / 2 - H / 2,
+            y2=back_video_height / 2 + H / 2
+        )
+    else:
+        background_clip = background_clip.crop(
+            x1=back_video_width / 2 - W / 2,
+            x2=back_video_width / 2 + W / 2,
+            y1=0,
+            y2=back_video_height
+        )
+
     final = CompositeVideoClip([background_clip, image_concat])
     title = re.sub(r'[^\w\s-]', '', reddit_obj['thread_title'])
     idx = re.sub(r'[^\w\s-]', '', reddit_obj['thread_id'])
 
     filename = f'{name_normalize(title)}.mp4'
-    subreddit = settings.config['reddit']['thread']['subreddit']
-
-    save_data(subreddit, filename, title, idx, background_config[2])
+    subreddit = str(settings.config['reddit']['thread']['subreddit'])
 
     if not exists(f'./results/{subreddit}'):
         print_substep('The results folder didn\'t exist so I made it')