Update app.py

app.py

@@ -1,3 +1,5 @@
+# app.py
+
 import os
 import cv2
 import numpy as np
@@ -84,7 +86,7 @@ def extract_text_from_pdf(pdf_path, lang='eng'):
             for page_num in range(len(pdf_reader.pages)):
                 page = pdf_reader.pages[page_num]
                 page_text = page.extract_text()
-                if page_text.strip():
+                if page_text and page_text.strip():
                     text += page_text
                 else:
                     images = convert_from_path(pdf_path, first_page=page_num + 1, last_page=page_num + 1)
@@ -248,16 +250,16 @@ def create_gradio_interface():
             outputs=[text_input]
         )
 
-        def process_generation(*args):
-            args = list(args)
+        def process_generation(api_key, input_text, model_name, voice_type, voice_speed, language, output_option):
+            """Wrapper function to process generation with updated parameters"""
             # Convert language name to code
-            args[5] = LANGUAGE_CODES[args[5]]
-            return generate_audio_and_text(*args)
+            language_code = LANGUAGE_CODES.get(language, "en")  # Default to English if not found
+            return generate_audio_and_text(api_key, input_text, model_name, voice_type, voice_speed, language_code, output_option)
         
         generate_button.click(
             fn=process_generation,
             inputs=[
-                api_key_input, text_input, model_dropdown, voice_type,
+                api_key_input, text_input, "gpt-4o-mini", voice_type,
                 voice_speed, language, output_option
             ],
             outputs=[audio_output, script_output, status_output]
@@ -267,4 +269,4 @@ def create_gradio_interface():
 
 if __name__ == "__main__":
     demo = create_gradio_interface()
-    demo.launch()
+    demo.launch()