Update app.py

app.py

@@ -9,7 +9,7 @@ from docx import Document
 from content_generation import create_content, CONTENT_TYPES
 from openai import OpenAI
 import edge_tts
-import subprocess
+from moviepy.editor import VideoFileClip, AudioFileClip, CompositeVideoClip, TextClip
 import time
 
 # Khởi tạo client OpenAI với API key từ biến môi trường
@@ -62,9 +62,6 @@ def create_video(audio_file, custom_text, text_color, output_video="output.mp4")
     # Chọn file video nền mặc định
     background_video = "default_background.mp4"  # Đặt tên file video nền mặc định
     
-    # Định nghĩa văn bản trên video
-    text_on_video = custom_text if custom_text else ""
-    
     # Định nghĩa màu sắc chữ
     color_map = {
         "Trắng": "white",
@@ -75,22 +72,23 @@ def create_video(audio_file, custom_text, text_color, output_video="output.mp4")
     }
     font_color = color_map.get(text_color, "white")  # Mặc định là trắng nếu không tìm thấy
     
-    # Lệnh ffmpeg để tạo video
-    command = [
-        "ffmpeg",
-        "-stream_loop", "-1",  # Lặp lại video nền vô hạn
-        "-i", background_video,  # Video nền mặc định
-        "-i", audio_file,      # File audio đầu vào
-        "-vf", f"drawtext=text='{text_on_video}':fontcolor={font_color}:fontsize=50:fontfile=/LHanoienne.otf:x=10:y=10:box=1:[email protected]:boxborderw=5",  # Văn bản trên video
-        "-t", "45.7",          # Thời lượng video
-        "-c:v", "libx264",     # Codec video
-        "-c:a", "aac",         # Codec audio
-        "-shortest",           # Dừng khi audio kết thúc
-        "-y",                  # Ghi đè file nếu tồn tại
-        output_video           # File video đầu ra
-    ]
-    # Chạy lệnh ffmpeg
-    subprocess.run(command, check=True)
+    # Tạo video nền
+    video_clip = VideoFileClip(background_video)
+    
+    # Tạo audio clip
+    audio_clip = AudioFileClip(audio_file)
+    
+    # Tạo video clip với audio
+    final_video = video_clip.set_audio(audio_clip)
+    
+    # Tạo text clip nếu có văn bản tùy chỉnh
+    if custom_text:
+        txt_clip = TextClip(custom_text, fontsize=50, color=font_color, font='/LHanoienne.otf')
+        txt_clip = txt_clip.set_position(('left', 'top')).set_duration(final_video.duration)
+        final_video = CompositeVideoClip([final_video, txt_clip])
+    
+    # Lưu video cuối cùng
+    final_video.write_videofile(output_video, codec='libx264', audio_codec='aac')
     return output_video
 
 def interface():