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""" mins, sec = divmod(seconds, 60) 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 .mp3 audio_paths = os.listdir(audio_path) audio_list = [file for file in audio_paths if file.endswith(".mp3")] # Khởi tạo danh sách audio duration duration_list = [] for audio_path in audio_list: # Mở file âm thanh và lấy thời gian with AudioFileClip(f"{audio_path}") as audio: duration_list.append(audio.duration) # Tính tổng tích lũy thời gian duration_list = [format_time(time) for time in list(accumulate(duration_list))] return [format_time(0.0)] + duration_list def create_srt_from_time_and_text(duration_time, text_folder, output_srt): subtitle = "" subtitle_index = 1 text_list = sorted([file for file in os.listdir(text_folder) if file.endswith('.txt') and file != "text.txt" and file != "requirements.txt"]) print(f"Accessing duration_time list: {len(text_list)} elements") # Duyệt qua các mốc thời gian và file text for i in range(len(duration_time) - 1): print(f"Accessing text_list at index {i}, length of text_list: {len(text_list)}") start_time = duration_time[i] end_time = duration_time[i + 1] # Lấy tên file text tương ứng text_file = text_list[i] text_path = os.path.join(text_folder, text_file) if os.path.exists(text_path): with open(text_path, 'r', encoding='utf-8') as f: text = f.read().strip() # Thêm phần subtitle vào chuỗi kết quả subtitle += f"{subtitle_index}\n{start_time} --> {end_time}\n{text}\n\n" subtitle_index += 1 else: print(f"File {text_file} không tồn tại!") # Lưu vào file 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 .mp3 trong thư mục audio_clips = [] for file in sorted(os.listdir(audio_folder)): if file.endswith('.mp3'): audio_path = os.path.join(audio_folder, file) audio_clip = AudioFileClip(audio_path) audio_clips.append(audio_clip) # Ghép tất cả các audio clip lại với nhau final_audio = concatenate_audioclips(audio_clips) # Lưu kết quả vào file output final_audio.write_audiofile(output_audio_path, codec = 'libmp3lame') print(f"File audio đã được lưu tại: {output_audio_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): """ Tạo video từ ảnh và âm thanh, có thể thêm phụ đề và chuyển đổi định dạng âm thanh. :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 """ 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) 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')}") 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")