Update src/text_to_video.py

src/text_to_video.py

@@ -7,7 +7,6 @@ from moviepy.video.VideoClip import ColorClip
 import os
 from itertools import accumulate
 import pysrt
-import subprocess
 
 
 def format_time(seconds):
@@ -75,71 +74,80 @@ 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 create_video(image_folder=None, audio_path=None, subtitle_path=None, output_video_path=None, convert_audio=False):
+def add_subtitles_to_video(video_path, subtitle_path, output_video_path):
     """
-    Tạo video từ ảnh và âm thanh, có thể thêm phụ đề và chuyển đổi định dạng âm thanh.
+    Thêm phụ đề từ file .srt trực tiếp vào video.
     
-    :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 video_path: Đường dẫn video gốc
+    :param subtitle_path: Đường dẫn file .srt
     :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
     """
-    if image_folder and audio_path:
-        # Tạo video từ ảnh và âm thanh
-        audio = AudioFileClip(audio_path)
-        total_duration = audio.duration
-
-        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!")
-        
-        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)
+    # Đọc file video
+    video = VideoFileClip(video_path)
+
+    # Đọ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)
         
-        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')}")
+        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)
 
+    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(image_folder="./", audio_path="final_audio.mp3", subtitle_path="subtitle.srt", output_video_path="final_output.mp4")
+    create_video_from_images("./","final_audio.mp3","output.mp4")
+    add_subtitles_to_video("output.mp4", "subtitle.srt", "final_output.mp4")