Update app.py

app.py

@@ -7,12 +7,6 @@ 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
 
@@ -47,52 +41,25 @@ def process_docx(file_path):
         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")
+            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)
+                    confirm_button = gr.Button("Xác nhận nội dung")
+                    download_docx = gr.File(label="Tải xuống file DOCX", interactive=False)
+                    status_message = gr.Label(label="Trạng thái")
 
             def generate_content(prompt, file, content_type):
                 try:
@@ -127,98 +94,15 @@ def interface():
                 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
+    return app
 
-            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)