"LOL"

src/text_to_video.py

@@ -24,7 +24,7 @@ def get_audio_duration(audio_path):
     
     for audio_path in audio_list:
         # Mở file âm thanh và lấy thời gian
-        with AudioFileClip(f"./audio/{audio_path}") as audio:
+        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))]
@@ -32,7 +32,7 @@ def get_audio_duration(audio_path):
 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')])
+    text_list = sorted([file for file in os.listdir(text_folder) if file.endswith('txt') and file != "text.txt"])
     # Duyệt qua các mốc thời gian và file text
     for i in range(len(duration_time) - 1):
         start_time = duration_time[i]
@@ -78,7 +78,7 @@ def create_video_from_images(image_folder, audio_path, output_video_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")]
+    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!")
@@ -87,7 +87,7 @@ def create_video_from_images(image_folder, audio_path, output_video_path):
     duration_per_image = total_duration / len(image_files)
 
     # Tạo danh sách các clip ảnh
-    clips = [ImageClip(f"./image/{img}").with_duration(duration_per_image).resized(width=1280) for img in image_files]
+    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")
@@ -127,7 +127,7 @@ def add_subtitles_to_video(video_path, subtitle_path, output_video_path):
         # 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"
+        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")
          
@@ -144,10 +144,8 @@ def add_subtitles_to_video(video_path, subtitle_path, output_video_path):
 
     print(f"Video với phụ đề đã được lưu tại: {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.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()
+    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")