Update app.py

app.py

@@ -1,4 +1,4 @@
-# Versión 3.2: App Pública con Logs ACTIVOS (Para verificar que funciona)
+# Versión 3.6: App Pública Completa (Watermark, Reparar, Numeración)
 import gradio as gr
 import pandas as pd
 import config
@@ -15,14 +15,11 @@ engine = PDFEngine()
 
 # --- SEGURIDAD: GARBAGE COLLECTOR ---
 # NOTA PARA DESARROLLADORES / DUPLICADORES DEL ESPACIO:
-# Esta función limpia los archivos cada minuto. Por defecto está en modo SILENCIOSO
-# para no ensuciar los logs en producción.
-# Si quieres ver qué está pasando, DESCOMENTA las líneas que empiezan por "#print"
-# y elimina los "pass" si es necesario.
+# Esta función limpia los archivos cada minuto. Por defecto está en modo SILENCIOSO.
+# Si quieres ver qué está pasando, DESCOMENTA las líneas que empiezan por "#print".
 def cleanup_cron():
     """
     Revisa cada minuto. Borra archivos Y carpetas mayores a 5 minutos.
-    Modo Producción: Logs desactivados por defecto.
     """
     print("[INIT] 🛡️  Sistema de seguridad y auto-borrado ACTIVO.")
     
@@ -33,7 +30,6 @@ def cleanup_cron():
             LIMIT_MINUTES = 5
             cutoff = time.time() - (LIMIT_MINUTES * 60)
             
-            # Cabecera de ronda
             #print(f"--- [SEGURIDAD] Ronda de limpieza: {time.strftime('%H:%M:%S')} ---")
             
             if os.path.exists(config.TEMP_DIR):
@@ -49,48 +45,41 @@ def cleanup_cron():
                 for filename in items:
                     filepath = os.path.join(config.TEMP_DIR, filename)
                     
-                    # Ofuscación visual
                     if len(filename) > 8:
                         masked = f"{filename[:4]}****{os.path.splitext(filename)[1]}"
                     else:
                         masked = filename
 
-                    # Calcular edad
                     try:
                         file_time = os.path.getmtime(filepath)
                         age_sec = time.time() - file_time
-                        age_str = f"{int(age_sec // 60)}m {int(age_sec % 60)}s"
-                        
                         is_expired = file_time < cutoff
                     except FileNotFoundError:
                         continue
 
-                    # --- CASO 1: ES UN ARCHIVO ---
+                    # CASO 1: ARCHIVO
                     if os.path.isfile(filepath):
                         if is_expired:
                             try:
                                 os.remove(filepath)
-                                #print(f"❌ [BORRADO] Archivo {masked} | Edad: {age_str}")
+                                #print(f"❌ [BORRADO] Archivo {masked}")
                             except Exception as e:
-                                #print(f"⚠️ [ERROR] No se pudo borrar archivo {masked}: {e}")
                                 pass
                         else:
-                            #print(f"✅ [VIGENTE] Archivo {masked} | Edad: {age_str}")
+                            #print(f"✅ [VIGENTE] Archivo {masked}")
                             pass
                     
-                    # --- CASO 2: ES UNA CARPETA ---
+                    # CASO 2: CARPETA
                     elif os.path.isdir(filepath):
                         if is_expired:
                             try:
                                 shutil.rmtree(filepath)
-                                #print(f"🗑️ [LIMPIEZA] Carpeta {masked} eliminada | Edad: {age_str}")
+                                #print(f"🗑️ [LIMPIEZA] Carpeta {masked} eliminada")
                             except Exception as e:
-                                #print(f"⚠️ [ERROR] No se pudo borrar carpeta {masked}: {e}")
                                 pass
                         else:
-                            #print(f"📂 [VIGENTE] Carpeta {masked} | Edad: {age_str}")
+                            #print(f"📂 [VIGENTE] Carpeta {masked}")
                             pass
-
             else:
                 #print("[INFO] Carpeta temporal aún no creada.")
                 pass
@@ -98,20 +87,35 @@ def cleanup_cron():
         except Exception as e:
             print(f"[CRITICAL] Error en limpieza: {e}")
 
-# Iniciar hilo en segundo plano
 threading.Thread(target=cleanup_cron, daemon=True).start()
 
+# --- GENERADOR DE ENLACE ---
+def get_duplicate_html():
+    space_id = os.environ.get('SPACE_ID', None)
+    url = f"https://huggingface.co/spaces/{space_id}?duplicate=true" if space_id else "https://huggingface.co/spaces?duplicate=true"
+    return f"""
+    <div style="display: flex; align-items: center; gap: 10px; margin-bottom: 20px; background-color: #f9fafb; padding: 10px; border-radius: 8px; border: 1px solid #e5e7eb;">
+        <a href="{url}" target="_blank">
+            <img src="https://huggingface.co/datasets/huggingface/badges/raw/main/duplicate-this-space-sm.svg" alt="Duplicate Space" style="height: 30px;">
+        </a>
+        <span style="font-size: 0.9em; color: #4b5563;">
+            ⚡ <b>¿Va lento?</b> Haz clic para duplicar este espacio y tener tu propia instancia privada (y activar logs si quieres).
+        </span>
+    </div>
+    """
+
 # --- WRAPPERS ---
 def update_file_list(files):
     if not files: return pd.DataFrame(), ""
     data = [[i, f.split("/")[-1]] for i, f in enumerate(files)]
     return pd.DataFrame(data, columns=["ID", "Archivo"]), ",".join([str(i) for i in range(len(files))])
 
-def process_merge(files, order_str):
+# MODIFICADO: Acepta use_numbering
+def process_merge(files, order_str, use_numbering):
     if not files: return None
     try:
         indices = [int(x.strip()) for x in order_str.split(",") if x.strip().isdigit()]
-        return engine.merge_pdfs(files, indices)
+        return engine.merge_pdfs(files, indices, use_numbering)
     except Exception as e: raise gr.Error(str(e))
 
 def load_info(f):
@@ -179,6 +183,17 @@ def process_text(f):
     try: return engine.extract_text(f)
     except Exception as e: raise gr.Error(str(e))
 
+# NUEVOS WRAPPERS
+def process_watermark(f, t):
+    if not f or not t: return None
+    try: return engine.add_watermark(f, t)
+    except Exception as e: raise gr.Error(str(e))
+
+def process_repair(f):
+    if not f: return None
+    try: return engine.repair_pdf(f)
+    except Exception as e: raise gr.Error(str(e))
+
 # WRAPPERS OFFICE
 def process_word(f):
     if not f: return None
@@ -207,44 +222,40 @@ with gr.Blocks(title=config.APP_TITLE, theme=gr.themes.Soft()) as demo:
     
     gr.Markdown(f"# {config.APP_TITLE}")
     gr.Markdown("""
+    ### 🛡️ Suite PDF: Libre, Gratuita y Privada
     Los archivos se procesan en memoria y se **autodestruyen tras 5 minutos**.
     """)
     
-    gr.HTML("""
-    <div style="display: flex; gap: 10px; margin-bottom: 20px;">
-        <span>⚡ ¿Funciona lento o quieres seguridad extra? Duplica este espacio para tener tu propia instancia privada.</span>
-    </div>
-    """)
+    gr.HTML(get_duplicate_html())
 
     with gr.Tabs():
-        
-        # 1. UNIR
+        # 1. UNIR (Actualizado con Numeración)
         with gr.TabItem("Unir"):
             with gr.Row():
                 with gr.Column(scale=1):
                     m_files = gr.File(file_count="multiple", label="Archivos", file_types=[".pdf"])
                 with gr.Column(scale=2):
                     m_tbl = gr.Dataframe(headers=["ID", "Archivo"], interactive=False)
-                    m_ord = gr.Textbox(label="Orden de los documentos según ID", placeholder="Ej: 0, 2, 1")
-                    m_btn = gr.Button("Unir", variant="primary")
+                    m_ord = gr.Textbox(label="Orden", placeholder="Ej: 0, 2, 1")
+                    # NUEVO CHECKBOX
+                    m_nums = gr.Checkbox(label="Numerar páginas (1 de X)", value=False)
+                    m_btn = gr.Button("Unir PDF", variant="primary")
             m_out = gr.File(label="Resultado")
             m_files.change(update_file_list, m_files, [m_tbl, m_ord])
-            m_btn.click(process_merge, [m_files, m_ord], m_out)
+            # Se pasa el valor del checkbox a la funcion
+            m_btn.click(process_merge, [m_files, m_ord, m_nums], m_out)
 
         # 2. DIVIDIR / REORDENAR
         with gr.TabItem("Dividir / Reordenar"):
-            
-            # BLOQUE SUPERIOR
             dr_f = gr.File(label="PDF Origen", file_types=[".pdf"])
             dr_inf = gr.Markdown("")
             dr_pg = gr.State(0)
-
-            # BLOQUE INFERIOR
             with gr.Tabs():
                 with gr.Tab("Extraer"):
+                    gr.Markdown("Separa páginas en un ZIP.")
                     with gr.Row():
                         with gr.Column():
-                            s_rng = gr.Textbox(label="Rangos de división", placeholder="Ej: 1-3, 5")
+                            s_rng = gr.Textbox(label="Rango", placeholder="Ej: 1-3, 5")
                             with gr.Row():
                                 s_prv = gr.Button("Preview")
                                 s_btn = gr.Button("Dividir (ZIP)", variant="primary")
@@ -253,7 +264,6 @@ with gr.Blocks(title=config.APP_TITLE, theme=gr.themes.Soft()) as demo:
                             s_out = gr.File(label="ZIP")
                     s_prv.click(update_split_preview, [dr_f, s_rng, dr_pg], s_gal)
                     s_btn.click(process_split, [dr_f, s_rng], s_out)
-
                 with gr.Tab("Reordenar"):
                     gr.Markdown("Crea un PDF con nuevo orden.")
                     with gr.Row():
@@ -263,7 +273,6 @@ with gr.Blocks(title=config.APP_TITLE, theme=gr.themes.Soft()) as demo:
                         with gr.Column():
                             r_out = gr.File(label="PDF Reordenado")
                     r_btn.click(process_reorder, [dr_f, r_ord], r_out)
-
             dr_f.change(load_info, dr_f, [dr_inf, dr_pg, s_out])
 
         # 3. COMPRIMIR
@@ -272,7 +281,7 @@ with gr.Blocks(title=config.APP_TITLE, theme=gr.themes.Soft()) as demo:
                 with gr.Column():
                     c_f = gr.File(label="PDF Original", file_types=[".pdf"])
                     c_l = gr.Radio(["Baja (Máxima calidad)", "Media (Recomendado - eBook)", "Alta (Pantalla - 72dpi)"], label="Nivel", value="Media (Recomendado - eBook)")
-                    c_b = gr.Button("Comprimir Documento", variant="primary")
+                    c_b = gr.Button("Comprimir", variant="primary")
                 with gr.Column():
                     c_out = gr.File(label="PDF Comprimido")
             c_b.click(process_compress, [c_f, c_l], c_out)
@@ -324,8 +333,8 @@ with gr.Blocks(title=config.APP_TITLE, theme=gr.themes.Soft()) as demo:
             co = gr.File(label="Informe PDF")
             cb_btn.click(process_compare, [ca, cb], co)
 
-        # 6. EXTRAS
-        with gr.TabItem("Extras (Rotar,Proteger,Metadatos)"):
+        # 6. EXTRAS (Actualizado con Marca de Agua y Reparar)
+        with gr.TabItem("Extras"):
             with gr.Tab("Rotar"):
                 with gr.Row():
                     with gr.Column():
@@ -338,6 +347,29 @@ with gr.Blocks(title=config.APP_TITLE, theme=gr.themes.Soft()) as demo:
                 rf.change(update_rot_preview, [rf, ra], rp)
                 ra.change(update_rot_preview, [rf, ra], rp)
                 rb.click(process_rotate, [rf, ra], ro)
+            
+            # NUEVO: Marca de Agua
+            with gr.Tab("Marca de Agua"):
+                gr.Markdown("Añade una marca de agua diagonal en todas las páginas.")
+                with gr.Row():
+                    with gr.Column():
+                        wf = gr.File(label="PDF")
+                        wt = gr.Textbox(label="Texto Marca de Agua", placeholder="Ej: CONFIDENCIAL")
+                        wb = gr.Button("Estampar", variant="primary")
+                    with gr.Column():
+                        wo = gr.File(label="PDF Marcado")
+                wb.click(process_watermark, [wf, wt], wo)
+
+            # NUEVO: Reparar
+            with gr.Tab("Reparar"):
+                gr.Markdown("Intenta arreglar PDFs corruptos o dañados reescribiéndolos con Ghostscript.")
+                with gr.Row():
+                    with gr.Column():
+                        repf = gr.File(label="PDF Dañado")
+                        repb = gr.Button("Reparar", variant="primary")
+                    with gr.Column():
+                        repo = gr.File(label="PDF Reparado")
+                repb.click(process_repair, repf, repo)
 
             with gr.Tab("Proteger"):
                 with gr.Row():
@@ -348,7 +380,7 @@ with gr.Blocks(title=config.APP_TITLE, theme=gr.themes.Soft()) as demo:
                     with gr.Column():
                         po = gr.File(label="Protegido")
                 pb.click(process_protect, [pf, pp], po)
-
+            
             with gr.Tab("Info/Texto"):
                 with gr.Row():
                     with gr.Column():