Update app.py

app.py

@@ -9,10 +9,14 @@ import gradio as gr
 from docx import Document
 from content_generation import create_content, CONTENT_TYPES
 from openai import OpenAI
+from gradio_client import Client, handle_file  # Thêm thư viện để gọi API
 
 # 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'))
 
+# Đường dẫn đến thư mục chứa các file âm thanh
+VOICES_DIR = "voices"
+
 def create_docx(content, output_path):
     """
     Tạo file docx từ nội dung.
@@ -41,6 +45,24 @@ def process_docx(file_path):
         text += para.text
     return text
 
+def text_to_speech(content, voice_file):
+    """
+    Chuyển đổi nội dung thành giọng nói bằng API.
+    """
+    try:
+        client = Client("thinhlpg/vixtts-demo")
+        result = client.predict(
+            prompt=content,
+            language="vi",
+            audio_file_pth=handle_file(voice_file),  # Sử dụng file âm thanh được chọn
+            normalize_text=True,
+            api_name="/predict"
+        )
+        # Chỉ lấy phần tử đầu tiên (filepath) từ kết quả trả về
+        return result[0]
+    except Exception as e:
+        return f"Lỗi khi chuyển đổi văn bản thành giọng nói: {str(e)}"
+
 def interface():
     with gr.Blocks() as app:
         gr.Markdown("# Ứng dụng Tạo Nội dung và Video")
@@ -53,6 +75,8 @@ def interface():
                     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
+                    voice_files = [os.path.join(VOICES_DIR, f) for f in os.listdir(VOICES_DIR) if f.endswith(".wav")]
+                    voice_selector = gr.Dropdown(label="Chọn giọng đọc", choices=voice_files)  # Dropdown để chọn file âm thanh
                     content_button = gr.Button("Tạo Nội dung")
                 
                 with gr.Column():
@@ -60,6 +84,8 @@ def interface():
                     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")
+                    convert_to_speech_button = gr.Button("Chuyển đổi thành giọng nói")
+                    audio_output = gr.Audio(label="Synthesised Audio", autoplay=True)  # Phát tự động
 
             def generate_content(prompt, file, content_type):
                 try:
@@ -94,13 +120,22 @@ def interface():
                 docx_path = "script.docx"
                 create_docx(content, docx_path)
 
+            def convert_content_to_speech(content, voice_file):
+                audio_path = text_to_speech(content, voice_file)
+                if os.path.exists(audio_path):
+                    return audio_path  # Chỉ trả về filepath để phát audio
+                else:
+                    return None
+
             content_button.click(generate_content,
                                  inputs=[prompt, file_upload, content_type],
                                  outputs=[content_output, download_docx, status_message])
 
-    return app
-
+            convert_to_speech_button.click(convert_content_to_speech,
+                                           inputs=[content_output, voice_selector],
+                                           outputs=[audio_output])
 
+    return app
 
 # Khởi chạy ứng dụng
 if __name__ == "__main__":