Update app.py

app.py

@@ -1,39 +1,46 @@
 import gradio as gr
 from TTS.api import TTS
 import fitz  # PyMuPDF for PDF text extraction
-from docx import Document
 from pydub import AudioSegment
 import tempfile
 import os
 
-# Initialize TTS models with accents and genders
+# Initialize TTS models
 TTS_MODELS = {
     "American - Male": "tts_models/en/ljspeech/tacotron2-DDC",
     "American - Female": "tts_models/en/ljspeech/tacotron2-DDC",
-    "British - Male": "tts_models/en-gb/apopeye/tacotron2-DDC",
-    "British - Female": "tts_models/en-gb/marlowe/tacotron2-DDC",
-    "Australian - Male": "tts_models/en-au/cameron/tacotron2-DDC",
-    "Australian - Female": "tts_models/en-au/aussie/tacotron2-DDC",
-    "Canadian - Male": "tts_models/en-ca/maple/tacotron2-DDC",
-    "Canadian - Female": "tts_models/en-ca/snow/tacotron2-DDC"
 }
 
-def extract_text_from_file(file_path, file_extension):
-    """Extract text from a .txt, .pdf, or .docx file."""
-    if file_extension == "txt":
-        with open(file_path, 'r', encoding='utf-8') as f:
-            return f.read()
-    elif file_extension == "pdf":
-        pdf_document = fitz.open(file_path)
-        text = ""
-        for page in pdf_document:
-            text += page.get_text()
-        return text
-    elif file_extension == "docx":
-        doc = Document(file_path)
-        return "\n".join(paragraph.text for paragraph in doc.paragraphs)
+def extract_text_from_pdf(pdf_path, page_selection=None):
+    """Extract specific pages from a PDF."""
+    pdf_document = fitz.open(pdf_path)
+    text = ""
+    total_pages = len(pdf_document)
+
+    # Parse the page_selection input (e.g., "1,3-5")
+    pages_to_read = []
+    if page_selection:
+        for part in page_selection.split(","):
+            if "-" in part:
+                start, end = map(int, part.split("-"))
+                pages_to_read.extend(range(start - 1, end))  # Convert to 0-indexed
+            else:
+                pages_to_read.append(int(part) - 1)  # Convert to 0-indexed
     else:
-        raise ValueError("Unsupported file type. Please upload a .txt, .pdf, or .docx file.")
+        pages_to_read = list(range(total_pages))  # Default: all pages
+
+    # Ensure valid page selection
+    pages_to_read = [p for p in pages_to_read if 0 <= p < total_pages]
+
+    # Extract text from selected pages
+    for page_num in pages_to_read:
+        page = pdf_document[page_num]
+        page_text = page.get_text()
+        text += f"\n--- Page {page_num + 1} ---\n{page_text}"  # Add page info
+
+    if not text.strip():
+        return "Error: No text found on the selected pages. They might contain images only."
+    return text
 
 def generate_audio(text, accent_gender, speed):
     """Generate audio from text using selected accent, gender, and speed."""
@@ -51,35 +58,45 @@ def generate_audio(text, accent_gender, speed):
     audio.export(output_mp3_path, format="mp3")
     return output_mp3_path
 
-def process_input(input_text, uploaded_file, accent_gender, speed):
+def process_input(input_text, uploaded_file, page_selection, accent_gender, speed):
     """Process input (text or file) and generate audio."""
-    if not input_text and not uploaded_file:
-        return "Please provide input text or upload a file.", None
-
-    # Extract text from file if uploaded
-    if uploaded_file:
-        file_extension = uploaded_file.name.split('.')[-1].lower()
-        with tempfile.NamedTemporaryFile(delete=False) as temp_file:
-            temp_file.write(uploaded_file.read())
-            temp_file_path = temp_file.name
-        text = extract_text_from_file(temp_file_path, file_extension)
-        os.remove(temp_file_path)
-    else:
-        text = input_text
-
-    # Generate audio
     try:
+        if not input_text and not uploaded_file:
+            return "Please provide input text or upload a file.", None
+
+        # Extract text from uploaded file
+        if uploaded_file:
+            file_extension = uploaded_file.name.split('.')[-1].lower()
+            with tempfile.NamedTemporaryFile(delete=False) as temp_file:
+                temp_file.write(uploaded_file.read())
+                temp_file_path = temp_file.name
+
+            if file_extension == "pdf":
+                text = extract_text_from_pdf(temp_file_path, page_selection)
+            else:
+                return "Error: Only PDF file support for page selection.", None
+
+            os.remove(temp_file_path)
+        else:
+            text = input_text
+
+        # Generate audio
         mp3_path = generate_audio(text, accent_gender, float(speed))
         return "Audio generated successfully!", mp3_path
+
     except Exception as e:
-        return f"Error: {str(e)}", None
+        import traceback
+        error_message = f"Error: {str(e)}\n{traceback.format_exc()}"
+        print(error_message)
+        return error_message, None
 
 # Gradio interface
 interface = gr.Interface(
     fn=process_input,
     inputs=[
         gr.Textbox(label="Enter Text", placeholder="Type or paste text here...", lines=5),
-        gr.File(label="Upload File (.txt, .pdf, .docx)", file_types=[".txt", ".pdf", ".docx"]),
+        gr.File(label="Upload File (.pdf only for page selection)", file_types=[".pdf"]),
+        gr.Textbox(label="Page Selection (e.g., '1,3-5' or leave blank for all pages)", placeholder="Pages to read"),
         gr.Dropdown(label="Accent & Gender", choices=list(TTS_MODELS.keys()), value="American - Male"),
         gr.Slider(label="Speed (e.g., 1.0 = Normal, 0.75 = Slower, 1.25 = Faster)", minimum=0.5, maximum=2.0, value=1.0, step=0.1),
     ],
@@ -88,7 +105,7 @@ interface = gr.Interface(
         gr.Audio(label="Generated Audio"),
     ],
     title="Text-to-Speech (TTS) Application",
-    description="Upload a text, PDF, or Word file or enter text directly. Customize accent, gender, and speed. Download the generated audio as MP3."
+    description="Upload a PDF file or enter text directly. Specify pages to extract text from, customize accent, gender, and speed. Download the generated audio as MP3."
 )
 
 # Launch the app