Update app.py

app.py

@@ -3,6 +3,8 @@ from gtts import gTTS
 import gradio as gr
 import tempfile
 import os
+import hashlib
+from pathlib import Path  # Importa Path desde pathlib
 
 # Ruta del directorio que quieres asegurarte de que exista
 dir_path = '/tmp/gradio/'
@@ -15,6 +17,20 @@ if not os.path.exists(dir_path):
     except Exception as e:
         print(f"No se pudo crear el directorio {dir_path}: {e}")
 
+# Definir la función save_audio_to_cache
+def save_audio_to_cache(data, cache_dir):
+    # Calcular el hash de los datos de audio para generar un nombre de archivo único
+    hash_value = hashlib.sha256(data).hexdigest()
+    file_name = f"{hash_value}.mp3"
+    file_path = Path(cache_dir) / file_name
+
+    # Guardar los datos de audio en el archivo cache_dir/file_name
+    with open(file_path, "wb") as audio_file:
+        audio_file.write(data)
+
+    return file_path
+
+# Definir la función convert_pdf_page_to_audio con la corrección
 def convert_pdf_page_to_audio(pdf_file, page_number):
     try:
         # Crear archivo temporal para el PDF
@@ -39,10 +55,13 @@ def convert_pdf_page_to_audio(pdf_file, page_number):
             tts.save(temp_audio_file.name)
             audio_file_path = temp_audio_file.name
 
-        # Leer y devolver los datos binarios del archivo de audio
+        # Leer los datos binarios del archivo de audio
         with open(audio_file_path, 'rb') as audio_file:
             audio_data = audio_file.read()
 
+        # Llamar a la función save_audio_to_cache para guardar los datos en la caché
+        audio_file_path = save_audio_to_cache(audio_data, dir_path)
+
         # Eliminar archivos temporales
         os.remove(pdf_file_path)
         os.remove(audio_file_path)