Update app.py

app.py

@@ -10,6 +10,7 @@ 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
+from tts import generate_speech
 
 # 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'))
@@ -47,22 +48,22 @@ def process_docx(file_path):
 
 def text_to_speech(content, voice_file):
     """
-    Chuyển đổi nội dung thành giọng nói bằng API.
+    Chuyển đổi nội dung thành giọng nói bằng hàm generate_speech từ tts.py.
     """
     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]
+        # Gọi hàm generate_speech để tạo file âm thanh
+        output_audio = generate_speech(content, language="vi", speaker_wav=voice_file)
+        return output_audio
     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 convert_content_to_speech(content, voice_file):
+    """
+    Chuyển đổi nội dung thành giọng nói.
+    """
+    return text_to_speech(content, voice_file)
+
+
 def interface():
     with gr.Blocks() as app:
         gr.Markdown("# Ứng dụng Tạo Nội dung và Video")
@@ -120,13 +121,6 @@ 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])