Update app.py

app.py

@@ -129,7 +129,7 @@ def cargar_modelo_traduccion(origen, destino):
 
 def traducir_texto(tokenizer, model, textos, dispositivo, batch_size=8):
     """
-    Extrae el contenido del PDF, traduce el texto y crea un nuevo PDF traducido.
+    Traduce una lista de textos utilizando el modelo y tokenizer proporcionados.
     """
     if not textos:  # Add a check for empty text list
         logger.warning("Lista de textos vacía. Saltando traducción.")
@@ -279,9 +279,10 @@ def ajustar_tamano_fuente(texto, bbox, c, max_width, tamaño_fuente_original):
         logger.warning(f"Error al ajustar tamaño de fuente: {e}")
         return tamaño_fuente_original
 
-def extraer_y_traducir_pdf(archivo_pdf, tokenizer, model, dispositivo, idioma_destino):
+def extraer_y_traducir_pdf_generador(archivo_pdf, tokenizer, model, dispositivo, idioma_destino):
     """
     Extrae el contenido del PDF, traduce el texto y crea un nuevo PDF traducido.
+    Esta versión utiliza 'yield' para emitir actualizaciones.
     """
     try:
         documento = fitz.open(archivo_pdf.name)
@@ -317,7 +318,8 @@ def extraer_y_traducir_pdf(archivo_pdf, tokenizer, model, dispositivo, idioma_de
         if not textos:
             logger.warning("No se encontraron textos válidos para traducir en el PDF.")
             documento.close()
-            return archivo_pdf.name
+            yield ("No se encontraron textos para traducir.", None)
+            return pdf_traducido_path
         
         # Traducir texto
         try:
@@ -339,14 +341,14 @@ def extraer_y_traducir_pdf(archivo_pdf, tokenizer, model, dispositivo, idioma_de
         for numero_pagina in range(total_paginas):
             pagina = documento.load_page(numero_pagina)
             rect = pagina.rect
-            ancho, alto = rect.width, rect.height
+            ancho_pagina, alto_pagina = rect.width, rect.height
             
             # Ajustar el tamaño de página al tamaño original del PDF
-            c.setPageSize((ancho, alto))
+            c.setPageSize((ancho_pagina, alto_pagina))
             
             # Definir márgenes dinámicos en base al tamaño de la página, ej: 5% de ancho y alto
-            margen_x = ancho * 0.05
-            margen_y = alto * 0.05
+            margen_x = ancho_pagina * 0.05
+            margen_y = alto_pagina * 0.05
             
             # Procesar texto de esta página
             pagina_bloques = pagina.get_text("dict")["blocks"]
@@ -375,7 +377,7 @@ def extraer_y_traducir_pdf(archivo_pdf, tokenizer, model, dispositivo, idioma_de
                             
                             # Ajustar coordenadas al sistema de ReportLab (y invertida)
                             x = x0
-                            y = alto - y1
+                            y = alto_pagina - y1
                             
                             # Buscar fuente similar
                             fuente_encontrada = buscar_fuente_similar(font, fuentes_sistema)
@@ -389,7 +391,7 @@ def extraer_y_traducir_pdf(archivo_pdf, tokenizer, model, dispositivo, idioma_de
                                 c.setFont(fuente_encontrada, size)
                             
                             # Ajustar el tamaño del texto si excede el ancho disponible
-                            max_width = (x1 - x0) - margen_x if (x1 - x0) > 0 else (ancho - 2 * margen_x)
+                            max_width = (x1 - x0) - margen_x if (x1 - x0) > 0 else (ancho_pagina - 2 * margen_x)
                             nuevo_tamaño = ajustar_tamano_fuente(texto_traducido, bbox, c, max_width, size)
                             
                             # Establecer el nuevo tamaño de fuente
@@ -425,29 +427,33 @@ def extraer_y_traducir_pdf(archivo_pdf, tokenizer, model, dispositivo, idioma_de
                     imagen_path = os.path.join(tempfile.gettempdir(), f"imagen_{numero_pagina}.png")
                     img.save(imagen_path)
                     img.close()
-                    c.drawImage(imagen_path, x0, alto - y1 - alto_img, width=ancho_img, height=alto_img)
+                    c.drawImage(imagen_path, x0, alto_pagina - y1 - alto_img, width=ancho_img, height=alto_img)
                 
                 except Exception as e:
                     logger.error(f"Error al procesar imagen: {e}")
                     continue
-
+        
+            # Emitir estado y vista previa de la página actual
+            yield (f"Traduciendo página {numero_pagina + 1} de {total_paginas} al idioma {idioma_destino}...", pdf_preview(archivo_pdf.name, numero_pagina))
+            
             c.showPage()
         
         c.save()
         documento.close()
+        yield ("Traducción completada exitosamente.", pdf_preview(pdf_traducido_path, pagina=0))
         return pdf_traducido_path
-    
     except Exception as e:
-        logger.error(f"Error general en extraer_y_traducir_pdf: {e}")
-        return archivo_pdf.name
+        logger.error(f"Error al extraer y traducir pdf: {e}")
 
-def pdf_preview(file):
+def pdf_preview(file_path, pagina=0):
     """
-    Previsualiza la primera página del PDF como una imagen.
+    Previsualiza una página específica del PDF como una imagen.
     """
     try:
-        doc = fitz.open(file.name)
-        page = doc[0]
+        doc = fitz.open(file_path)
+        if pagina >= len(doc):
+            pagina = 0  # Fallback a la primera página si el número es inválido
+        page = doc[pagina]
         pix = page.get_pixmap()
         image = np.frombuffer(pix.samples, np.uint8).reshape(pix.height, pix.width, pix.n)
         if pix.n == 4:
@@ -510,11 +516,16 @@ def boton_actualizar_fuentes(files):
 
 def procesar_pdf(archivo_pdf, fuentes_subidas):
     """
-    Función principal para procesar y traducir el PDF.
+    Función generadora para procesar y traducir el PDF.
+    Emite actualizaciones de estado y vista previa durante el proceso.
     """
     try:
         if not archivo_pdf:
-            return None, "No se ha subido ningún archivo PDF."
+            yield ("No se ha subido ningún archivo PDF.", None, None)
+            return
+        
+        # Emitir estado inicial
+        yield ("Iniciando traducción...", pdf_preview(archivo_pdf.name, pagina=0), None)
         
         # Extraer texto para detectar el idioma
         documento = fitz.open(archivo_pdf.name)
@@ -528,29 +539,27 @@ def procesar_pdf(archivo_pdf, fuentes_subidas):
         # Si el idioma de origen y destino son iguales, no realizar traducción
         if idioma_origen == idioma_sistema:
             logger.info("El idioma de origen y destino son iguales. No se realizará la traducción.")
-            return archivo_pdf.name, "El idioma de origen y destino son iguales. No se realizó la traducción."
+            yield ("El idioma de origen y destino son iguales. No se realizó la traducción.", pdf_preview(archivo_pdf.name, pagina=0), None)
+            return
         
         # Cargar el modelo de traducción automáticamente
         tokenizer, model, dispositivo = cargar_modelo_traduccion(idioma_origen, idioma_sistema)
         
-        # Traducir el PDF
-        pdf_traducido_path = extraer_y_traducir_pdf(archivo_pdf, tokenizer, model, dispositivo, idioma_sistema)
+        # Realizar la traducción y emitir actualizaciones
+        generator = extraer_y_traducir_pdf_generador(
+            archivo_pdf, tokenizer, model, dispositivo, idioma_sistema
+        )
+        for update in generator:
+            status, imagen = update
+            yield (status, imagen, None)
         
-        return pdf_traducido_path, "Traducción completada exitosamente."
+        # Emitir estado final
+        pdf_traducido_path = os.path.splitext(archivo_pdf.name)[0] + f"_traducido_{idioma_sistema}.pdf"
+        yield ("Traducción completada exitosamente.", pdf_preview(pdf_traducido_path, pagina=0), pdf_traducido_path)
+    
     except Exception as e:
         logger.error(f"Error en procesar_pdf: {e}")
-        return None, f"Error en la traducción: {e}"
-
-def actualizar_fuentes_cache():
-    """
-    Función para actualizar el caché de fuentes.
-    """
-    try:
-        cachear_fuentes()
-        return "Fuentes cacheadas exitosamente."
-    except Exception as e:
-        logger.error(f"Error al cachear fuentes: {e}")
-        return f"Error al cachear fuentes: {e}"
+        yield (f"Error en la traducción: {e}", None, None)
 
 # Interfaz de usuario con Gradio
 with gr.Blocks(
@@ -577,17 +586,24 @@ with gr.Blocks(
         with gr.Column(scale=1):
             gr.Markdown("## Vista Previa")
             preview = gr.Image(label="Vista Previa", visible=True)
+            status_md = gr.Markdown("**Estado:** Esperando traducción...")
             traducir_btn = gr.Button("Traducir PDF")
-            estado_traduccion = gr.Textbox(label="Estado", interactive=False)
+            traducir_output = gr.File(label="Descargar PDF traducido", visible=True)
+            
+            # Configurar el botón para usar la función generadora
             traducir_btn.click(
                 fn=procesar_pdf,
                 inputs=[pdf_input, fuentes_subidas],
-                outputs=[gr.File(label="Descargar PDF traducido"), estado_traduccion]
+                outputs=[status_md, preview, traducir_output],
+                show_progress=True  # Opcional: muestra una barra de progreso
             )
     
             # Vista previa del PDF
+            def actualizar_preview(file):
+                return pdf_preview(file.name, pagina=0) if file else None
+    
             pdf_input.change(
-                fn=pdf_preview,
+                fn=actualizar_preview,
                 inputs=pdf_input,
                 outputs=preview
             )