Update app.py

app.py

@@ -4,7 +4,8 @@ import io
 import os
 import time
 from gtts.lang import _main_langs
-from docx import Document  # Thêm thư viện này để làm việc với tệp docx
+from docx import Document  # Thư viện để làm việc với tệp .docx
+from ebooklib import epub  # Thư viện để làm việc với tệp .epub
 
 AUDIO_DIR = 'audio_files'
 MAX_FILE_AGE = 24 * 60 * 60  # maximum age of audio files in seconds (24 hours)
@@ -48,6 +49,15 @@ def docx_to_speech(file, lang, tld):
     text = "\n".join([para.text for para in doc.paragraphs])  # Lấy tất cả văn bản từ các đoạn
     return text_to_speech(text, lang, tld)
 
+# Hàm chuyển đổi file .epub thành giọng nói
+def epub_to_speech(file, lang, tld):
+    book = epub.read_epub(file.name)
+    text = ""
+    
+    for item in book.get_items_of_type(epub.EpubHtml):
+        text += item.get_body_content_str().decode('utf-8')  # Lấy nội dung của mỗi trang
+    return text_to_speech(text, lang, tld)
+
 # Tạo giao diện Gradio với tab
 with gr.Blocks() as iface:
     with gr.Tab("Text to Speech"):
@@ -83,4 +93,15 @@ with gr.Blocks() as iface:
                                     inputs=[docx_file_input, lang_input_docx, tld_input_docx], 
                                     outputs=[audio_output_docx, file_output_docx])
 
+    with gr.Tab("EPUB to Speech"):
+        gr.Markdown("### Convert .epub file to speech")
+        epub_file_input = gr.File(label="Upload your .epub file")
+        lang_input_epub = gr.Dropdown(choices=list(_main_langs().values()), label="Select language:")
+        tld_input_epub = gr.Dropdown(choices=["com", "co.uk", "ca"], label="Select TLD:", value="com")  # Bạn có thể điều chỉnh TLD
+        
+        audio_output_epub, file_output_epub = gr.Audio(label="Audio"), gr.File(label="Audio File")
+        gr.Button("Convert").click(fn=lambda file, lang, tld: epub_to_speech(file, lang, tld),
+                                    inputs=[epub_file_input, lang_input_epub, tld_input_epub], 
+                                    outputs=[audio_output_epub, file_output_epub])
+
 iface.launch(enable_queue=True)