Update components/utils.py

components/utils.py

@@ -1,17 +1,18 @@
 from moviepy.editor import VideoFileClip, concatenate_videoclips,vfx
-from moviepy.editor import AudioFileClip, CompositeAudioClip
+from moviepy.editor import AudioFileClip, CompositeAudioClip, TextClip, 
+from moviepy.video.tools.subtitles import SubtitlesClip
 import os 
 
-def combine_videos(folder_name):
+def combine_videos(folder_name, length_speech):
 
     length = len(os.listdir(folder_name))
 
     vids = [VideoFileClip(os.path.join(folder_name,x)) for x in sorted(os.listdir(folder_name))]
 
-    print(vids, length)
+    print(vids, length, len(length_speech))
     
     combined = concatenate_videoclips(
-        [vid.subclip(0,7).fx(vfx.fadein,0.5).fx(vfx.fadeout,0.5) if vid.duration > 7 else vid.fx(vfx.fadein,0.5).fx(vfx.fadeout,0.5) for vid in vids],
+        [vid.subclip(0,ls).fx(vfx.fadein,0.5).fx(vfx.fadeout,0.5) if vid.duration > ls else vid.fx(vfx.fadein,0.5).fx(vfx.fadeout,0.5) for vid, ls in zip(vids, length_speech)],
         # [VideoFileClip(os.path.join(folder_name,x)).subclip(0,7).fx(vfx.fadein,0.5).fx(vfx.fadeout,0.5) for x in sorted(os.listdir(folder_name))], 
         "compose",bg_color=None, padding=0
     )
@@ -25,10 +26,26 @@ def combine_videos(folder_name):
 #         combined.write_videofile(os.path.join(folder_name, "Ad_Video.mp4"))
     print("Done! Your ads video has been created")
 
-def combine_audio_video(folder_name, vid, speech):
+
+def combine_audio_video(folder_name, vid, speech, sentences, length_speech):
     videoclip = VideoFileClip(vid)
     audioclip = AudioFileClip(speech)
 
-    new_audioclip = CompositeAudioClip([audioclip])
-    videoclip.audio = new_audioclip
+    generator = lambda txt: TextClip(txt, font='Arial', fontsize=24, color='white')
+    # subs = [((0, 4), 'subs1'),
+    #         ((4, 9), 'subs2'),
+    #         ((9, 12), 'subs3'),
+    #         ((12, 16), 'subs4')]
+
+    ls = [0] + length_speech
+    subs = [ ( (ls[i],ls[i+1]), sub) for i,sub in enumerate(sentences)]
+    
+    subtitles = SubtitlesClip(subs, generator)
+
+    
+    videoclip = CompositeVideoClip([videoclip, subtitles.set_pos(('center','bottom'))])
+
+    # new_audioclip = CompositeAudioClip([audioclip])
+    # videoclip.audio = new_audioclip
+    videoclip_audio = CompositeAudioClip([audioclip])
     videoclip.write_videofile(os.path.join(folder_name,"new_filename.mp4"))