Update app.py

app.py

@@ -553,13 +553,33 @@ def generate_audio(
     combined_text = original_text or ""
 
     # If there's no original text, extract it from the uploaded files
+    '''
     if not combined_text:
         for file in files:
             with Path(file).open("rb") as f:
                 reader = PdfReader(f)
                 text = "\n\n".join([page.extract_text() for page in reader.pages if page.extract_text()])
                 combined_text += text + "\n\n"
+    '''
 
+    if not combined_text:
+        for file in files:
+            file_path = Path(file)
+            suffix = file_path.suffix.lower()
+    
+            if suffix == ".pdf":
+                with file_path.open("rb") as f:
+                    reader = PdfReader(f)
+                    text = "\n\n".join(
+                        page.extract_text() for page in reader.pages if page.extract_text()
+                    )
+                    combined_text += text + "\n\n"
+            elif suffix in [".txt", ".md", ".mmd"]:
+                with file_path.open("r", encoding="utf-8") as f:
+                    text = f.read()
+                    combined_text += text + "\n\n"
+
+    
     # Configure the LLM based on selected model and api_base
     @retry(retry=retry_if_exception_type(ValidationError))
     @conditional_llm(model=text_model, api_base=api_base, api_key=openai_api_key)
@@ -721,7 +741,8 @@ with gr.Blocks(title="PDF to Audio", css="""
 
     with gr.Row(elem_id="main_container"):
         with gr.Column(scale=2):
-            files = gr.Files(label="PDFs", file_types=["pdf"], )
+            #files = gr.Files(label="PDFs", file_types=["pdf"], )
+            files = gr.Files(label="PDFs (.pdf), markdown (.md, .mmd), or text files (.txt)", file_types=[".pdf", ".PDF", ".md", ".mmd", ".txt"], )
             
             openai_api_key = gr.Textbox(
                 label="OpenAI API Key",