Update src/text_to_video.py

src/text_to_video.py

@@ -15,9 +15,9 @@ def format_time(seconds):
     hours, mins = divmod(mins, 60)
     return f"{int(hours):02}:{int(mins):02}:{int(sec):02},{int((sec % 1) * 1000):03}"
 def get_audio_duration(audio_path):
-    # Lọc các file có đuôi .wav
+    # Lọc các file có đuôi .mp3
     audio_paths = os.listdir(audio_path)
-    audio_list = [file for file in audio_paths if file.endswith(".wav")]
+    audio_list = [file for file in audio_paths if file.endswith(".mp3")]
     
     # Khởi tạo danh sách audio duration
     duration_list = []
@@ -56,11 +56,11 @@ def create_srt_from_time_and_text(duration_time, text_folder, output_srt):
     with open(output_srt, 'w', encoding='utf-8') as f:
         f.write(subtitle)
 def concatenate_audio_files(audio_folder, output_audio_path):
-    # Lọc tất cả các file âm thanh .wav trong thư mục
+    # Lọc tất cả các file âm thanh .mp3 trong thư mục
     audio_clips = []
     
     for file in sorted(os.listdir(audio_folder)):
-        if file.endswith('.wav'):
+        if file.endswith('.mp3'):
             audio_path = os.path.join(audio_folder, file)
             audio_clip = AudioFileClip(audio_path)
             audio_clips.append(audio_clip)
@@ -69,7 +69,7 @@ def concatenate_audio_files(audio_folder, output_audio_path):
     final_audio = concatenate_audioclips(audio_clips)
     
     # Lưu kết quả vào file output
-    final_audio.write_audiofile(output_audio_path, codec = 'pcm_s16le')
+    final_audio.write_audiofile(output_audio_path, codec = 'libmp3lame')
 
     print(f"File audio đã được lưu tại: {output_audio_path}")
 def create_video_from_images(image_folder, audio_path, output_video_path):
@@ -96,7 +96,7 @@ def create_video_from_images(image_folder, audio_path, output_video_path):
     final_video .audio = audio
 
     # Xuất video
-    final_video.write_videofile(output_video_path, codec="libx264", audio_codec="libx264", fps=24)
+    final_video.write_videofile(codec="libx264", audio_codec="aac", fps=30)
 
     print(f"Video đã được lưu tại: {output_video_path}")
 def wrap_text(text, max_width):
@@ -146,8 +146,8 @@ def add_subtitles_to_video(video_path, subtitle_path, output_video_path):
 def text_to_video():
     duration_time = get_audio_duration("./audio")
     create_srt_from_time_and_text(duration_time, './text', './output/subtitle.srt')
-    concatenate_audio_files("./audio","./output/final_audio.wav")
-    create_video_from_images("./image","./output/final_audio.wav","./output/output.mp4")
+    concatenate_audio_files("./audio","./output/final_audio.mp3")
+    create_video_from_images("./image","./output/final_audio.mp3","./output/output.mp4")
     add_subtitles_to_video("./output/output.mp4", "./output/subtitle.srt", "./output/final_output.mp4")
 if __name__ == "__main__":
     text_to_video()