Update app.py

app.py

@@ -26,25 +26,27 @@ def dividir_audio(audio_path, segment_duration_ms=5*60*1000):
     
     return segments
 
-# Función para crear el PDF
-def crear_pdf(transcripcion):
+# Función para crear el PDF con barra de progreso
+def crear_pdf(transcripcion, progress=gr.Progress()):
     pdf = FPDF()
     pdf.add_page()
 
     pdf.set_font("Arial", size=12)
-    
-    # Agregar la transcripción al PDF línea por línea
-    for line in transcripcion.split("\n"):
+    lines = transcripcion.split("\n")
+
+    # Agregar la transcripción al PDF línea por línea con progreso
+    for i, line in enumerate(lines):
         pdf.multi_cell(0, 10, line)
-    
-    # Guardar el archivo PDF en la carpeta actual
+        progress(i / len(lines))  # Actualiza la barra de progreso
+
+    # Guardar el archivo PDF
     pdf_path = "/mnt/data/transcripcion_audio.pdf"
     pdf.output(pdf_path)
     
     return pdf_path
 
 # Función para transcribir los segmentos de audio y generar el PDF
-def transcribe_audio_y_pdf(audio_file):
+def transcribe_audio_y_pdf(audio_file, progress=gr.Progress()):
     # Cargar el archivo de audio completo
     audio = AudioSegment.from_file(audio_file)
     audio_duration_minutes = len(audio) / (1000 * 60)  # Duración en minutos
@@ -85,24 +87,33 @@ def transcribe_audio_y_pdf(audio_file):
     # Combina todas las transcripciones en una sola cadena
     full_transcription = "\n".join(all_transcriptions)
 
-    # Crear PDF y devolver la ruta del archivo
-    pdf_path = crear_pdf(full_transcription)
+    # Crear PDF con progreso y devolver la ruta del archivo
+    pdf_path = crear_pdf(full_transcription, progress)
     yield full_transcription, pdf_path
 
 # Crea la interfaz de Gradio con scroll automático en el campo de texto
 with gr.Blocks() as demo:
-    gr.Markdown("# Transcriptor de Audio a Texto (con actualizaciones progresivas y PDF)")
+    gr.Markdown("# Transcriptor de Audio a Texto (con actualizaciones progresivas, scroll y PDF)")
 
     with gr.Row():
         audio_input = gr.Audio(type="filepath", label="Sube tu archivo de audio")
         transcribe_button = gr.Button("Transcribir y Crear PDF")
 
     # Caja de texto con scroll para ver la transcripción progresivamente
-    output_text = gr.Textbox(label="Transcripción (se muestra de a poco)", lines=20, interactive=False)
+    output_text = gr.Textbox(label="Transcripción (scroll automático y manual)", lines=20, interactive=True, elem_id="transcripcion")
     output_pdf = gr.File(label="Descarga el PDF")
 
     # Botón de transcripción y conexión con las funciones
     transcribe_button.click(fn=transcribe_audio_y_pdf, inputs=audio_input, outputs=[output_text, output_pdf])
 
+    # JavaScript para hacer el scroll automático de la caja de texto
+    demo.load(js="""
+        function autoScroll() {
+            const textbox = document.getElementById('transcripcion');
+            textbox.scrollTop = textbox.scrollHeight;
+        }
+        document.addEventListener('gradio:output', autoScroll);
+    """)
+
 # Ejecuta la aplicación
 demo.launch()