Update src/text_to_video.py

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"./data/audio/{audio_path}") as audio:
+        with AudioFileClip(f"./audio/{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))]
@@ -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"./data/image/{img}").with_duration(duration_per_image).resized(width=1280) for img in image_files]
+    clips = [ImageClip(f"./image/{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 = "./data/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,10 @@ 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("./data/audio")
-    create_srt_from_time_and_text(duration_time, './data/text', './data/output/subtitle.srt')
-    concatenate_audio_files("./data/audio","./data/output/final_audio.wav")
-    create_video_from_images("./data/image","./data/output/final_audio.wav","./data/output/output.mp4")
-    add_subtitles_to_video("./data/output/output.mp4", "./data/output/subtitle.srt", "./data/output/final_output.mp4")
+    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.wav")
+    create_video_from_images("./image","./output/final_audio.wav","./output/output.mp4")
+    add_subtitles_to_video("./output/output.mp4", "./output/subtitle.srt", "./output/final_output.mp4")
 if __name__ == "__main__":
     text_to_video()