Update app.py

fixed show time error

app.py

@@ -25,11 +25,11 @@ def procesar_pdf(pdf_file=None, url_pdf=None, formatos=[], verbose="auto"):
     os.makedirs(output_base, exist_ok=True)
     start_time = time.time()
 
-    # Verificar que marker_single está disponible
+    # Verificar marker_single
     try:
         subprocess.run(["marker_single", "--help"], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
     except FileNotFoundError:
-        yield "Error: el comando 'marker_single' no está disponible. Asegúrate de que 'marker-pdf' esté correctamente instalado en requirements.txt.", "", None, None, None
+        yield "Error: el comando 'marker_single' no está disponible.", "", None, None, None
         return
 
     if not formatos:
@@ -44,7 +44,7 @@ def procesar_pdf(pdf_file=None, url_pdf=None, formatos=[], verbose="auto"):
             ruta_pdf = obtener_ruta_segura(output_base, nombre_pdf)
             response = requests.get(url_pdf)
             if response.status_code != 200:
-                yield "No se pudo descargar correctamente el PDF desde la URL.", "", None, None, None
+                yield "No se pudo descargar correctamente el PDF desde la URL.", "", None, None, None, None
                 return
             with open(ruta_pdf, "wb") as f:
                 f.write(response.content)
@@ -53,17 +53,17 @@ def procesar_pdf(pdf_file=None, url_pdf=None, formatos=[], verbose="auto"):
             ruta_pdf = obtener_ruta_segura(output_base, nombre_pdf)
             shutil.copyfile(pdf_file.name, ruta_pdf)
         else:
-            yield "No se proporcionó archivo ni URL.", "", None, None, None
+            yield "No se proporcionó archivo ni URL.", "", None, None, None, None
             return
 
-        # Verificar cabecera %PDF
+        # Verificar cabecera PDF
         with open(ruta_pdf, "rb") as f:
             if f.read(4) != b"%PDF":
-                yield "El archivo no es un PDF válido (no comienza por %PDF).", "", None, None, None
+                yield "El archivo no es un PDF válido (no comienza por %PDF).", "", None, None, None, None
                 return
 
         if os.path.getsize(ruta_pdf) > 3 * 1024 * 1024:
-            yield "Aviso: el PDF es grande y puede tardar más de lo normal.", "", None, None, None
+            yield "Aviso: el PDF es grande y puede tardar más de lo normal.", "", None, None, None, None
 
         nombre_sin_ext = os.path.splitext(os.path.basename(ruta_pdf))[0]
         carpeta_salida = os.path.join(output_base, nombre_sin_ext)
@@ -71,7 +71,7 @@ def procesar_pdf(pdf_file=None, url_pdf=None, formatos=[], verbose="auto"):
             shutil.rmtree(carpeta_salida)
         os.makedirs(carpeta_salida, exist_ok=True)
 
-        # Procesar con marker_single
+        # Procesar cada formato
         for fmt in formatos:
             tiempo_parcial = f"{round(time.time() - start_time, 1)} s"
             if fmt == "md":
@@ -81,7 +81,7 @@ def procesar_pdf(pdf_file=None, url_pdf=None, formatos=[], verbose="auto"):
             else:
                 cmd = ["marker_single", ruta_pdf, "--output_format", fmt, "--output_dir", carpeta_salida]
 
-            yield f"Procesando formato '{fmt}'... (tiempo: {tiempo_parcial})", "", None, None, None
+            yield f"Procesando formato '{fmt}'...", tiempo_parcial, None, None, None
 
             for intento in range(1, 4):
                 try:
@@ -89,14 +89,14 @@ def procesar_pdf(pdf_file=None, url_pdf=None, formatos=[], verbose="auto"):
                     if result.returncode == 0:
                         break
                     elif intento == 3:
-                        yield f"Error en marker_single (intento {intento}):\n{result.stderr}", "", None, None, None
+                        yield f"Error en marker_single:\n{result.stderr}", tiempo_parcial, None, None, None
                         return
                     time.sleep(2)
                 except Exception as e:
-                    yield f"Error inesperado durante el intento {intento}:\n{str(e)}", "", None, None, None
+                    yield f"Error inesperado:\n{str(e)}", tiempo_parcial, None, None, None
                     return
 
-        # Leer contenido .md
+        # Leer contenido Markdown
         md_path = os.path.join(carpeta_salida, f"{nombre_sin_ext}.md")
         md_content = ""
         if os.path.exists(md_path):
@@ -123,12 +123,12 @@ def procesar_pdf(pdf_file=None, url_pdf=None, formatos=[], verbose="auto"):
 
         estado_final = f"Procesamiento completado correctamente (tiempo total: {tiempo_total} segundos)"
         salida_md = md_content + resumen if md_content else resumen
-        yield estado_final, "", gr.update(value=md_path, visible=True), gr.update(value=zip_path, visible=True), salida_md
+        yield estado_final, f"{tiempo_total} s", gr.update(value=md_path, visible=True), gr.update(value=zip_path, visible=True), salida_md
 
     except Exception as e:
         yield f"Error general inesperado: {str(e)}", "", None, None, None
 
-# Interfaz Gradio
+# Interfaz con 5 salidas
 demo = gr.Interface(
     fn=procesar_pdf,
     inputs=[
@@ -141,13 +141,14 @@ demo = gr.Interface(
         )
     ],
     outputs=[
-        gr.Textbox(label="Estado del procesamiento y tiempo total", lines=3),
+        gr.Textbox(label="Estado del procesamiento", lines=2),
+        gr.Textbox(label="Tiempo parcial", interactive=False),
         gr.File(label="Descargar Markdown", visible=False),
         gr.File(label="Descargar ZIP completo", visible=False),
-        gr.Textbox(label="Contenido Markdown extraído", lines=25, visible=False)
+        gr.Textbox(label="Contenido extraído en Markdown", lines=25, visible=False)
     ],
     title="Marker PDF",
-    description="Convierte PDFs científicos en Markdown, HTML o JSON. Incluye OCR opcional y resumen detallado del contenido.",
+    description="Convierte PDFs científicos en Markdown, HTML o JSON. Incluye OCR opcional y resumen completo.",
     flagging_mode="never"
 )