Update src/text_to_video.py

src/text_to_video.py

@@ -7,6 +7,7 @@ from moviepy.video.VideoClip import ColorClip
 import os
 from itertools import accumulate
 import pysrt
+import subprocess
 
 
 def format_time(seconds):
@@ -74,80 +75,70 @@ def concatenate_audio_files(audio_folder, output_audio_path):
     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):
-    # Đọc file âm thanh để lấy thời lượng
-    audio = AudioFileClip(audio_path)
-    total_duration = audio.duration  # Tổng thời lượng video bằng thời lượng audio
-
-    # Đọc tất cả các file ảnh trong thư mục và sắp xếp theo tên
-    image_files = [file for file in sorted(os.listdir(image_folder)) if file.endswith(".png")]
-    
-    if not image_files:
-        raise ValueError("Không tìm thấy ảnh nào trong thư mục!")
-
-    # Tính thời lượng hiển thị cho mỗi ảnh
-    duration_per_image = total_duration / len(image_files)
-
-    # Tạo danh sách các clip ảnh
-    clips = [ImageClip(f"{img}").with_duration(duration_per_image).resized(width=1280) for img in image_files]
-
-    # Ghép các clip ảnh lại với nhau
-    final_video = concatenate_videoclips(clips, method="chain")
-    
-    # Gán âm thanh vào video
-    final_video .audio = audio
-
-    # Xuất video
-    final_video.write_videofile(output_video_path, codec="libx264", audio_codec="aac", fps=30)
-
-    print(f"Video đã được lưu tại: {output_video_path}")
 def wrap_text(text, max_width):
     """
     Tự động xuống dòng để vừa với chiều rộng max_width.
     """
     import textwrap
     return "\n".join(textwrap.wrap(text, width=max_width))
-def add_subtitles_to_video(video_path, subtitle_path, output_video_path):
+def create_video(image_folder=None, audio_path=None, subtitle_path=None, output_video_path=None, convert_audio=False):
     """
-    Thêm phụ đề từ file .srt trực tiếp vào video.
+    Tạo video từ ảnh và âm thanh, có thể thêm phụ đề và chuyển đổi định dạng âm thanh.
     
-    :param video_path: Đường dẫn video gốc
-    :param subtitle_path: Đường dẫn file .srt
+    :param image_folder: Thư mục chứa ảnh (nếu tạo video từ ảnh)
+    :param audio_path: Đường dẫn file âm thanh (nếu tạo video từ ảnh và âm thanh)
+    :param subtitle_path: Đường dẫn file phụ đề .srt (nếu muốn thêm phụ đề)
     :param output_video_path: Đường dẫn lưu video đầu ra
+    :param convert_audio: Nếu True, chuyển đổi âm thanh sang AAC
     """
-    
-    # Đọc file video
-    video = VideoFileClip(video_path)
+    if image_folder and audio_path:
+        # Tạo video từ ảnh và âm thanh
+        audio = AudioFileClip(audio_path)
+        total_duration = audio.duration
 
-    # Đọc file .srt
-    subs = pysrt.open(subtitle_path)
-
-    subtitle_clips = []  # Danh sách các đoạn phụ đề
-
-    # Xử lý từng dòng phụ đề
-    for sub in subs:
-        # Chuyển thời gian thành giây
-        start_time = sub.start.ordinal / 1000  # Chuyển từ milliseconds sang giây
-        end_time = sub.end.ordinal / 1000
-        font = "BeVietnamPro-Light.ttf"
-        # Tạo clip phụ đề
-        txt_clip = TextClip(font=font, text=wrap_text(sub.text, max_width=85), font_size=30, stroke_color="black", stroke_width=3, color="#fff")
-         
-        # Đặt vị trí hiển thị (giữa phía dưới video)
-        txt_clip = txt_clip.with_position(('center', 'bottom')).with_duration(end_time - start_time).with_start(start_time)
+        image_files = [file for file in sorted(os.listdir(image_folder)) if file.endswith(".png")]
+        if not image_files:
+            raise ValueError("Không tìm thấy ảnh nào trong thư mục!")
         
-        subtitle_clips.append(txt_clip)
-
-    # Ghép phụ đề vào video
-    final_video = CompositeVideoClip([video] + subtitle_clips)
-
-    # Xuất video với phụ đề
-    final_video.write_videofile(output_video_path, fps=video.fps, codec='libx264', threads=4)
+        duration_per_image = total_duration / len(image_files)
+        clips = [ImageClip(f"{img}").with_duration(duration_per_image).resized(width=1280) for img in image_files]
+        final_video = concatenate_videoclips(clips, method="chain")
+        final_video.audio = audio
+        final_video.write_videofile(output_video_path, codec="libx264", audio_codec="aac", fps=30)
+        print(f"✅ Video đã được tạo tại: {output_video_path}")
+    
+    if convert_audio:
+        # Chuyển đổi định dạng âm thanh của video
+        command = [
+            "ffmpeg", "-i", output_video_path,
+            "-c:v", "copy", "-c:a", "aac", "-b:a", "192k",
+            "-y", output_video_path.replace(".mp4", "_aac.mp4")
+        ]
+        try:
+            subprocess.run(command, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+            print(f"✅ Đã chuyển đổi âm thanh: {output_video_path.replace('.mp4', '_aac.mp4')}")
+        except subprocess.CalledProcessError as e:
+            print(f"❌ Lỗi khi chuyển đổi âm thanh: {e.stderr.decode()}")
+    
+    if subtitle_path:
+        # Thêm phụ đề vào video
+        video = VideoFileClip(output_video_path)
+        subs = pysrt.open(subtitle_path)
+        subtitle_clips = []
+        
+        for sub in subs:
+            start_time = sub.start.ordinal / 1000
+            end_time = sub.end.ordinal / 1000
+            font = "BeVietnamPro-Light.ttf"
+            txt_clip = TextClip(font=font, text=wrap_text(sub.text, max_width=85), font_size=30, stroke_color="black", stroke_width=3, color="#fff")
+            txt_clip = txt_clip.with_position(('center', 'bottom')).with_duration(end_time - start_time).with_start(start_time)
+            subtitle_clips.append(txt_clip)
+        
+        final_video = CompositeVideoClip([video] + subtitle_clips)
+        final_video.write_videofile(output_video_path.replace(".mp4", "_subtitled.mp4"), fps=video.fps, codec='libx264', threads=4)
+        print(f"✅ Video với phụ đề đã được lưu tại: {output_video_path.replace('.mp4', '_subtitled.mp4')}")
 
-    print(f"Video với phụ đề đã được lưu tại: {output_video_path}")
 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(image_folder="./", audio_path="final_audio.mp3", subtitle_path="subtitle.srt", output_video_path="final_output.mp4")