Update src/text_to_video.py

src/text_to_video.py

@@ -2,11 +2,10 @@ from moviepy.video.io.VideoFileClip import VideoFileClip, AudioFileClip
 from moviepy.video.VideoClip import TextClip, ImageClip
 from moviepy.video.compositing.CompositeVideoClip import concatenate_videoclips, CompositeVideoClip
 from moviepy.audio.AudioClip import concatenate_audioclips
-from moviepy.video.tools.subtitles import SubtitlesClip
-from moviepy.video.VideoClip import ColorClip
 import os
 from itertools import accumulate
 import pysrt
+import subprocess
 
 def format_time(seconds):
     """Chuyển đổi thời gian (giây) thành định dạng SRT hh:mm:ss,ms"""
@@ -144,9 +143,27 @@ def add_subtitles_to_video(video_path, subtitle_path, output_video_path):
     final_video.write_videofile(output_video_path, fps=video.fps, codec='libx264', threads=4)
 
     print(f"Video với phụ đề đã được lưu tại: {output_video_path}")
+def convert_audio_format(video_input, video_output):
+    """Chuyển đổi định dạng âm thanh của video sang AAC."""
+    if not os.path.exists(video_input):
+        raise FileNotFoundError(f"File '{video_input}' không tồn tại!")
+
+    command = [
+        "ffmpeg", "-i", video_input,  
+        "-c:v", "copy", "-c:a", "aac", "-b:a", "192k",
+        "-y",  # Ghi đè nếu file output đã tồn tại
+        video_output
+    ]
+    
+    try:
+        subprocess.run(command, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        print(f"✅ Chuyển đổi thành công: {video_output}")
+    except subprocess.CalledProcessError as e:
+        print(f"❌ Lỗi khi chuyển đổi video: {e.stderr.decode()}")
 def text_to_video():
     duration_time = get_audio_duration("./")
     create_srt_from_time_and_text(duration_time, './', 'subtitle.srt')
-    concatenate_audio_files("./","final_audio.mp3")
-    create_video_from_images("./","final_audio.mp3","output.mp4")
-    add_subtitles_to_video("output.mp4", "subtitle.srt", "final_output.mp4")
+    concatenate_audio_files("./","./final_audio.mp3")
+    create_video_from_images("./","./final_audio.mp3","output.mp4")
+    add_subtitles_to_video("./output.mp4", "./subtitle.srt", "./final_output.mp4")
+    convert_audio_format("./final_output.mp4","./final_fixed_ouput.mp4")