Update app.py

app.py

@@ -19,41 +19,40 @@ if not os.path.exists(dir_path):
 
 def convert_pdf_page_to_audio(pdf_file, page_number):
     try:
-        # Crear un archivo temporal para el PDF
-        temp_pdf_file = tempfile.NamedTemporaryFile(delete=False, suffix='.pdf')
-        temp_pdf_file.write(pdf_file.read())
-        temp_pdf_file.close()
-        pdf_file_path = temp_pdf_file.name
+        with tempfile.NamedTemporaryFile(delete=False, suffix='.pdf') as temp_pdf_file:
+            temp_pdf_file.write(pdf_file.read())
+            pdf_file_path = temp_pdf_file.name
 
         reader = PyPDF2.PdfReader(pdf_file_path)
         page_number = int(page_number)
         if page_number < 0 or page_number >= len(reader.pages):
-            os.remove(pdf_file_path)  # Eliminar el archivo temporal
+            os.remove(pdf_file_path)
             return None, "Número de página inválido."
 
         text = reader.pages[page_number].extract_text()
         if not text:
-            os.remove(pdf_file_path)  # Eliminar el archivo temporal
+            os.remove(pdf_file_path)
             return None, "No se pudo extraer texto de la página."
 
-        # Crear un archivo temporal para el audio
-        temp_audio_file = tempfile.NamedTemporaryFile(delete=False, suffix='.mp3')
-        tts = gTTS(text=text, lang='es')
-        tts.save(temp_audio_file.name)
-        temp_audio_file.close()
-        audio_file_path = temp_audio_file.name
+        with tempfile.NamedTemporaryFile(delete=False, suffix='.mp3') as temp_audio_file:
+            tts = gTTS(text=text, lang='es')
+            tts.save(temp_audio_file.name)
+            audio_file_path = temp_audio_file.name
 
-        os.remove(pdf_file_path)  # Eliminar el archivo PDF temporal
+        # Leer los datos binarios del archivo de audio
+        with open(audio_file_path, 'rb') as audio_file:
+            audio_data = audio_file.read()
 
-        return audio_file_path, None
+        os.remove(pdf_file_path)
+        os.remove(audio_file_path)
+
+        return audio_data, None  # Devolver los datos binarios del audio
 
     except Exception as e:
-        # Limpieza en caso de excepción
-        if temp_pdf_file:
-            os.remove(temp_pdf_file.name)
         return None, f"Error: {str(e)}"
 
 
+
 # Crear la interfaz de Gradio
 iface = gr.Interface(
     fn=convert_pdf_page_to_audio,