Update app.py

app.py

@@ -4,6 +4,10 @@ import asyncio
 import tempfile
 import os
 import time
+import fitz 
+import ebooklib
+from ebooklib import epub
+from bs4 import BeautifulSoup
 
 async def get_voices():
     # Mock voice list to include specified voices
@@ -15,6 +19,40 @@ async def get_voices():
     ]
     return {f"{v['ShortName']} - {v['Locale']} ({v['Gender']})": v['ShortName'] for v in voices}
 
+def extract_text_from_file(file):
+    if file is None:
+        return None, "কোনো ফাইল আপলোড করা হয়নি"
+    
+    file_path = file.name
+    file_ext = os.path.splitext(file_path)[1].lower()
+    
+    try:
+        if file_ext == ".pdf":
+            doc = fitz.open(file_path)
+            text = ""
+            for page in doc:
+                text += page.get_text("text")
+            doc.close()
+            return text.strip(), None
+        elif file_ext == ".epub":
+            book = epub.read_epub(file_path)
+            text = ""
+            for item in book.get_items_of_type(ebooklib.ITEM_DOCUMENT):
+                content = item.get_content().decode("utf-8")
+                soup = BeautifulSoup(content, "lxml")
+                for element in soup.find_all(text=True):
+                    if element.strip():
+                        text += element.strip() + "\n"
+            return text.strip(), None
+        elif file_ext == ".txt":
+            with open(file_path, "r", encoding="utf-8") as f:
+                text = f.read()
+            return text.strip(), None
+        else:
+            return None, "অসমর্থিত ফাইল ফরম্যাট। শুধুমাত্র PDF, EPUB, এবং TXT ফাইল সমর্থিত।"
+    except Exception as e:
+        return None, f"টেক্সট নিষ্কাশনে ত্রুটি: {str(e)}"
+
 async def text_to_speech(text, voice, rate, pitch):
     if not text.strip():
         return None, "বাংলা লেখা সংযুক্ত করুন"
@@ -40,8 +78,14 @@ async def tts_interface(text, voice, rate, pitch):
         return audio, None, gr.Warning(warning)
     return audio, audio, None
 
+async def handle_file_upload(file):
+    text, warning = extract_text_from_file(file)
+    if warning:
+        return None, gr.Warning(warning)
+    return text, None
+
 def reset_fields():
-    return "", "", 0, 0, None, None, ""
+    return None, "", "", 0, 0, None, None, ""
 
 async def create_demo():
     voices = await get_voices()
@@ -78,7 +122,7 @@ async def create_demo():
         background: linear-gradient(45deg, #d32f2f, #e91e63) !important;
         box-shadow: 0 4px 12px rgba(0, 0, 0, 0.2) !important;
     }
-    .gr-textbox, .gr-dropdown, .gr-slider {
+    .gr-textbox, .gr-dropdown, .gr-slider, .gr-file {
         border-radius: 8px !important;
         border: 1px solid #b0bec5 !important;
         box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1) !important;
@@ -107,18 +151,23 @@ async def create_demo():
     with gr.Blocks(css=css, analytics_enabled=False) as demo:
         gr.Markdown(
             """
-            # 🎙️ Edge TTS For Bangla Language
-            লেখা থেকে উচ্চ-মানের কণ্ঠস্বরে রূপান্তর করুন। বাংলা ভাষায় স্বাভাবিক এবং সুন্দর কণ্ঠস্বর উপভোগ করুন।
+            # 🎙️ Edge TTS Text-to-Speech
+            লেখা বা ফাইল (PDF, EPUB, TXT) থেকে উচ্চ-মানের কণ্ঠস্বরে রূপান্তর করুন। বাংলা ভাষায় স্বাভাবিক এবং সুন্দর কণ্ঠস্বর উপভোগ করুন।
             """,
             elem_classes=["container"]
         )
         
         with gr.Group():
-            gr.Markdown("### লেখা ইনপুট এবং সেটিংস")
+            gr.Markdown("### ফাইল আপলোড এবং লেখা ইনপুট")
+            file_input = gr.File(
+                label="ফাইল আপলোড করুন (PDF, EPUB, TXT)",
+                file_types=[".pdf", ".epub", ".txt"],
+                file_count="single"
+            )
             text_input = gr.Textbox(
                 label="প্রদত্ত লেখা",
                 lines=5,
-                placeholder="এখানে আপনার বাংলা লেখা লিখুন, যেমন: 'আমি বাংলায় কথা বলি।'",
+                placeholder="এখানে আপনার বাংলা লেখা লিখুন বা ফাইল আপলোড করুন, যেমন: 'আমি বাংলায় কথা বলি।'",
                 show_copy_button=True
             )
             voice_dropdown = gr.Dropdown(
@@ -155,6 +204,12 @@ async def create_demo():
             warning_md = gr.Markdown(label="Warning", visible=False)
         
         # Event handlers
+        file_input.change(
+            fn=handle_file_upload,
+            inputs=[file_input],
+            outputs=[text_input, warning_md],
+            show_progress=True
+        )
         generate_btn.click(
             fn=tts_interface,
             inputs=[text_input, voice_dropdown, rate_slider, pitch_slider],
@@ -164,7 +219,7 @@ async def create_demo():
         reset_btn.click(
             fn=reset_fields,
             inputs=[],
-            outputs=[text_input, voice_dropdown, rate_slider, pitch_slider, audio_output, download_output, warning_md]
+            outputs=[file_input, text_input, voice_dropdown, rate_slider, pitch_slider, audio_output, download_output, warning_md]
         )
     
     return demo