Update app.py

app.py

@@ -2,27 +2,22 @@ import asyncio
 import mimetypes
 import os
 import tempfile
+import glob
 import fitz  # PyMuPDF
 import random
 import gradio as gr
 from docx import Document
+from audio_processing import async_text_to_speech, text_to_speech
+from content_generation import create_content, CONTENT_TYPES
+from video_processing import create_video_func
+from moviepy.editor import AudioFileClip, VideoFileClip, CompositeAudioClip
+from utils import (combine_videos, get_pexels_video, get_bgm_file, download_video)
+from video_processing import create_video
 from content_generation import create_content, CONTENT_TYPES
 from openai import OpenAI
-import edge_tts
-from moviepy.editor import VideoFileClip, AudioFileClip, CompositeVideoClip, TextClip
-
-# Kiểm tra file video nền
-background_video = "default_background.mp4"
-if not os.path.exists(background_video):
-    raise FileNotFoundError(f"File video nền không tồn tại: {background_video}")
-
-# Kiểm tra file font
-font_file = "/LHanoienne.otf"
-if not os.path.exists(font_file):
-    raise FileNotFoundError(f"File font không tồn tại: {font_file}")
 
 # Khởi tạo client OpenAI với API key từ biến môi trường
-client_openai = OpenAI(api_key=os.environ.get('OPENAI_API_KEY'))
+client = OpenAI(api_key=os.environ.get('OPENAI_API_KEY'))
 
 def create_docx(content, output_path):
     """
@@ -52,80 +47,53 @@ def process_docx(file_path):
         text += para.text
     return text
 
-async def convert_text_to_speech(text, language='vi', output_file="output.wav"):
+def get_bgm_file_list():
     """
-    Chuyển đổi văn bản thành giọng nói sử dụng edge_tts.
+    Trả về danh sách các tệp nhạc nền.
     """
-    try:
-        communicate = edge_tts.Communicate(text, "vi-VN-JennyNeural")
-        with open(output_file, "wb") as file:
-            await communicate.save(file)
-        return output_file
-    except Exception as e:
-        raise Exception(f"Lỗi khi chuyển đổi văn bản thành giọng nói: {str(e)}")
+    # Giả sử bạn có một thư mục chứa các tệp nhạc nền
+    song_dir = "/data/bg-music"
+    return [os.path.basename(file) for file in glob.glob(os.path.join(song_dir, "*.mp3"))]
 
-def create_video(audio_file, custom_text, text_color, output_video="output.mp4"):
+def extract_key_contents(script, num_contents=10):
     """
-    Tạo video với văn bản tùy chỉnh.
+    Trích xuất các ý chính từ script.
     """
-    # Định nghĩa màu sắc chữ
-    color_map = {
-        "Trắng": "white",
-        "Đỏ": "red",
-        "Xanh dương": "blue",
-        "Xanh lá": "green",
-        "Vàng": "yellow"
-    }
-    font_color = color_map.get(text_color, "white")  # Mặc định là trắng nếu không tìm thấy
-    
-    # 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=font_file)
-        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
+    try:
+        response = client.chat.completions.create(
+            model="gpt-3.5-turbo",
+            messages=[
+                {"role": "system", "content": f"Bạn là một chuyên gia phân tích nội dung. Hãy trích xuất chính xác {num_contents} ý chính quan trọng nhất từ đoạn văn sau, mỗi ý không quá 20 từ."},
+                {"role": "user", "content": script}
+            ]
+        )
+        # Sửa lỗi truy cập đối tượng không thể subscript
+        key_contents = response.choices[0].message.content.split('\n')
+        return key_contents[:num_contents]
+    except Exception as e:
+        print(f"Lỗi khi trích xuất nội dung: {str(e)}")
+        return []
 
 def interface():
-    with gr.Blocks(theme=gr.themes.Soft(font=[gr.themes.GoogleFont("Roboto Mono")])) as app:
+    with gr.Blocks() as app:
         gr.Markdown("# Ứng dụng Tạo Nội dung và Video")
+
         with gr.Tab("Tạo Nội dung"):
-            with gr.Row():
-                with gr.Column():
-                    prompt = gr.Textbox(label="Nhập yêu cầu nội dung")
-                    file_upload = gr.File(label="Tải lên file kèm theo", type="filepath")
-                    content_type = gr.Radio(label="Chọn loại nội dung",
-                                            choices=CONTENT_TYPES,
-                                            value=None)  # Giá trị mặc định là không có gì được chọn
-                    content_button = gr.Button("Tạo Nội dung")
-                
-                with gr.Column():
-                    content_output = gr.Textbox(label="Nội dung tạo ra", interactive=True)
-                    use_generated_text = gr.Checkbox(label="Sử dụng văn bản được tạo bởi LLM", value=True)
-                    manual_text_input = gr.Textbox(label="Nhập văn bản thủ công", visible=False)
-                    language = gr.Dropdown(label="Chọn ngôn ngữ", choices=["vi", "en"], value="vi")
-                    custom_text_input = gr.Textbox(label="Nhập văn bản tùy chỉnh (trống nếu không muốn hiển thị)", value="")
-                    text_color = gr.Dropdown(label="Chọn màu chữ", choices=["Trắng", "Đỏ", "Xanh dương", "Xanh lá", "Vàng"], value="Trắng")
-                    confirm_button = gr.Button("Xác nhận nội dung")
-                    download_docx = gr.File(label="Tải xuống file DOCX", interactive=False)
-                    download_audio = gr.File(label="Tải xuống file âm thanh", interactive=False)
-                    download_video = gr.File(label="Tải xuống file video", interactive=False)
-                    status_message = gr.Label(label="Trạng thái")
-            
-            def toggle_manual_text(use_generated_text):
-                return {"visible": not use_generated_text}
-            
+            prompt = gr.Textbox(label="Nhập yêu cầu nội dung")
+            file_upload = gr.File(label="Tải lên file kèm theo", type="filepath")
+
+            # Sử dụng gr.Radio thay vì gr.CheckboxGroup
+            content_type = gr.Radio(label="Chọn loại nội dung",
+                                    choices=CONTENT_TYPES,
+                                    value=None)  # Giá trị mặc định là không có gì được chọn
+
+            content_button = gr.Button("Tạo Nội dung")
+            content_output = gr.Textbox(label="Nội dung tạo ra", interactive=True)
+            confirm_button = gr.Button("Xác nhận nội dung")
+            download_docx = gr.File(label="Tải xuống file DOCX", interactive=False)
+            download_audio = gr.File(label="Tải xuống file âm thanh", interactive=False)
+            status_message = gr.Label(label="Trạng thái")
+
             def generate_content(prompt, file, content_type):
                 try:
                     status = "Đang xử lý..."
@@ -141,56 +109,116 @@ def interface():
                             prompt = f"{prompt}\n\nDưới đây là nội dung của file tài liệu:\n\n{file_content}"
                         else:
                             raise ValueError("Định dạng file không được hỗ trợ.")
+
                     if not content_type:
                         raise ValueError("Vui lòng chọn một loại nội dung")
+
                     script_content = create_content(prompt, content_type, "Tiếng Việt")
                     docx_path = "script.docx"
                     create_docx(script_content, docx_path)
+
                     status = "Đã tạo nội dung thành công!"
-                    return script_content, docx_path, True, None, status
+                    return script_content, docx_path, status
                 except Exception as e:
                     status = f"Đã xảy ra lỗi: {str(e)}"
-                    return "", None, True, None, status
-            
-            async def confirm_content(content_output, use_generated_text, manual_text_input, language, custom_text_input, text_color):
-                try:
-                    if use_generated_text:
-                        text_to_convert = content_output
-                    else:
-                        text_to_convert = manual_text_input
-                    
-                    if not text_to_convert:
-                        return None, None, None, "Vui lòng cung cấp văn bản để chuyển đổi."
-                    
-                    docx_path = "script.docx"
-                    create_docx(text_to_convert, docx_path)
-                    status_message.value = "Đang chuyển đổi văn bản thành giọng nói..."
-                    audio_path = await convert_text_to_speech(text_to_convert, language)
-                    status_message.value = "Đã chuyển đổi văn bản thành giọng nói thành công!"
-                    status_message.value = "Đang tạo video..."
-                    video_path = create_video(audio_path, custom_text_input, text_color)
-                    status_message.value = "Đã tạo video thành công!"
-                    return docx_path, audio_path, video_path, status_message.value
-                except Exception as e:
-                    status_message.value = f"Đã xảy ra lỗi: {str(e)}"
-                    return None, None, None, status_message.value
-            
-            # Liên kết sự kiện change của checkbox với hàm toggle_manual_text
-            use_generated_text.change(fn=toggle_manual_text, inputs=[use_generated_text], outputs=[manual_text_input])
-            
-            # Liên kết sự kiện click của nút Tạo Nội dung
+                    return "", None, status
+
+            async def confirm_content(content):
+                docx_path = "script.docx"
+                create_docx(content, docx_path)
+
+                audio_path = await async_text_to_speech(content, "alloy", "Tiếng Việt")
+                return docx_path, audio_path, "Nội dung đã được xác nhận và âm thanh đã được tạo!"
+
             content_button.click(generate_content,
                                  inputs=[prompt, file_upload, content_type],
-                                 outputs=[content_output, download_docx, use_generated_text, manual_text_input, status_message])
-            
-            # Liên kết sự kiện click của nút Xác nhận nội dung
-            confirm_button.click(confirm_content,
-                                 inputs=[content_output, use_generated_text, manual_text_input, language, custom_text_input, text_color],
-                                 outputs=[download_docx, download_audio, download_video, status_message])
-    
+                                 outputs=[content_output, download_docx, status_message])
+
+            confirm_button.click(lambda x: asyncio.run(confirm_content(x)),
+                                 inputs=[content_output],
+                                 outputs=[download_docx, download_audio, status_message])
+        
+        # Định nghĩa danh sách giọng đọc
+        VOICES = ["vi-VN-HoaiMyNeural"]
+
+        with gr.Tab("Tạo Âm thanh"):
+            text_input = gr.Textbox(label="Nhập văn bản để chuyển đổi")
+            voice_select = gr.Dropdown(label="Chọn giọng đọc", choices=VOICES)  # Dropdown cho voice_select
+            audio_button = gr.Button("Tạo Âm thanh")
+            audio_output = gr.Audio(label="Âm thanh tạo ra")
+            download_audio = gr.File(label="Tải xuống file âm thanh", interactive=False)
+
+            def text_to_speech_func(text, voice):
+                try:
+                    audio_path = await async_text_to_speech(text, voice, "Tiếng Việt")
+                    return audio_path, audio_path
+                except Exception as e:
+                    print(f"Lỗi khi chuyển đổi văn bản thành giọng nói: {e}")
+                    return None, None
+
+            audio_button.click(text_to_speech_func,
+                               inputs=[text_input, voice_select],
+                               outputs=[audio_output, download_audio])
+
+        with gr.Tab("Tạo Video"):
+            script_input = gr.Textbox(label="Nhập kịch bản")
+            audio_file = gr.File(label="Chọn file âm thanh", type="filepath")
+            keywords_output = gr.Textbox(label="Từ khóa", interactive=True)
+            max_clip_duration = gr.Slider(minimum=2, maximum=5, step=1, label="Thời lượng tối đa mỗi video (giây)")
+            join_order = gr.Checkbox(label="Ghép ngẫu nhiên", value=True)
+            bgm_files = gr.Dropdown(choices=get_bgm_file_list(), label="Chọn nhạc nền")
+            video_output = gr.Video(label="Video tạo ra")
+            video_button = gr.Button("Tạo Video")
+            status_message = gr.Label(label="Trạng thái")  # Thêm thông báo trạng thái
+
+            def create_video_func(script, audio_file, max_clip_duration, join_order, bgm_file):
+                """ Tạo video từ các thông tin đầu vào. """
+                try:
+                    # Cập nhật trạng thái ban đầu
+                    status = "Đang xử lý..."
+
+                    # 1. Tính toán thời lượng video
+                    audio_clip = AudioFileClip(audio_file)
+                    video_duration = audio_clip.duration
+
+                    # 2. Trích xuất từ khóa từ kịch bản
+                    keywords = extract_key_contents(script)
+                    video_paths = []
+                    for keyword in keywords:
+                        video_url = get_pexels_video(keyword.strip())
+                        if video_url:
+                            video_path = download_video(video_url)
+                            video_paths.append(video_path)
+
+                    # 3. Ghép video
+                    temp_dir = tempfile.mkdtemp()
+                    if join_order:
+                        random.shuffle(video_paths)
+                    combined_video_path = os.path.join(temp_dir, "combined_video.mp4")
+                    combine_videos(combined_video_path, video_paths, audio_file, max_clip_duration)
+
+                    # 4. Gộp audio và nhạc nền
+                    final_video_path = "final_video.mp4"
+                    bgm_clip = AudioFileClip(bgm_file)
+                    final_audio = CompositeAudioClip([audio_clip, bgm_clip])
+                    final_video = VideoFileClip(combined_video_path).set_audio(final_audio)
+                    final_video.write_videofile(final_video_path)
+
+                    # Cập nhật trạng thái thành công
+                    status = "Video đã được tạo thành công!"
+                    return final_video_path, status
+                except Exception as e:
+                    # Cập nhật trạng thái lỗi
+                    status = f"Lỗi khi tạo video: {e}"
+                    return None, status
+
+            video_button.click(create_video_func,
+                               inputs=[script_input, audio_file, max_clip_duration, join_order, bgm_files],
+                               outputs=[video_output, status_message])
+
     return app
 
 # Khởi chạy ứng dụng
 if __name__ == "__main__":
     app = interface()
-    app.launch(share=True)
+    app.launch(share=True)