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 # Khởi tạo client OpenAI với API key từ biến môi trường client = OpenAI(api_key=os.environ.get('OPENAI_API_KEY')) def create_docx(content, output_path): """ Tạo file docx từ nội dung. """ doc = Document() doc.add_paragraph(content) doc.save(output_path) def process_pdf(file_path): """ Xử lý file PDF và trích xuất nội dung. """ doc = fitz.open(file_path) text = "" for page in doc: text += page.get_text() return text def process_docx(file_path): """ Xử lý file DOCX và trích xuất nội dung. """ doc = Document(file_path) text = "" for para in doc.paragraphs: text += para.text return text def get_bgm_file_list(): """ Trả về danh sách các tệp nhạc nền. """ # 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 extract_key_contents(script, num_contents=10): """ Trích xuất các ý chính từ script. """ 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() as app: gr.Markdown("# Ứng dụng Tạo Nội dung và Video") with gr.Tab("Tạo Nội dung"): 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ý..." if file and os.path.exists(file): mime_type, _ = mimetypes.guess_type(file) if mime_type == "application/pdf": file_content = process_pdf(file) prompt = f"{prompt}\n\nDưới đây là nội dung của file tài liệu:\n\n{file_content}" elif mime_type in ( "application/vnd.openxmlformats-officedocument.wordprocessingml.document", "application/msword"): file_content = process_docx(file) 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, status except Exception as e: status = f"Đã xảy ra lỗi: {str(e)}" 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, 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)