Update app.py

app.py

@@ -1,38 +1,49 @@
-# Versión 1.8: UI Plana y Reordenada con Compresión
+# Versión 1.9: Interfaz Gráfica Completa (Todas las herramientas)
+# Autor: Gemini (AI Assistant)
+# Descripción: UI construida con Gradio. Se encarga solo de la presentación y llama a core.py.
+
 import gradio as gr
 import pandas as pd
 import config
 from core import PDFEngine
 
+# Inyección de dependencias
 engine = PDFEngine()
 
-# --- FUNCIONES DE INTERFAZ (Wrappers) ---
+# --- FUNCIONES DE ENLACE (WRAPPERS) ---
+# Estas funciones conectan la UI con el Core y manejan errores visuales.
 
-# 1. Unir
 def update_file_list(files):
+    """Actualiza la tabla de IDs cuando se suben archivos para Unir."""
     if not files: return pd.DataFrame(), ""
+    # Gradio devuelve lista de rutas
     data = [[i, f.split("/")[-1]] for i, f in enumerate(files)]
+    # Generar orden por defecto "0, 1, 2..."
     default_order = ",".join([str(i) for i in range(len(files))])
     return pd.DataFrame(data, columns=["ID", "Archivo"]), default_order
 
 def process_merge(files, order_str):
     if not files: return None
     try:
+        # Parsear string "0, 2, 1" a lista [0, 2, 1]
         indices = [int(x.strip()) for x in order_str.split(",") if x.strip().isdigit()]
         return engine.merge_pdfs(files, indices)
-    except Exception as e: raise gr.Error(str(e))
+    except Exception as e: raise gr.Error(f"Error al unir: {str(e)}")
 
-# 2. Dividir
 def load_pdf_info(file):
+    """Carga información básica al subir un archivo."""
     if not file: return None, 0, gr.update(visible=False)
     info = engine.get_pdf_info(file)
-    return f"Documento: {info['name']} ({info['pages']} págs).", info['pages'], gr.update(visible=True)
+    return f"📄 {info['name']} ({info['pages']} páginas)", info['pages'], gr.update(visible=True)
 
 def update_split_preview(file, range_str, total_pages):
+    """Muestra previsualización de las páginas seleccionadas en Dividir."""
     if not file or not range_str: return None
     key_pages = engine.get_preview_indices_from_string(range_str, total_pages)
     if not key_pages: return None
+    
     gallery = []
+    # Limitamos a 8 imágenes para no saturar la UI
     for p in key_pages[:8]:
         path = engine.generate_preview(file, p)
         if path: gallery.append((path, f"Pág {p}"))
@@ -43,10 +54,19 @@ def process_split(file, range_str):
     try: return engine.split_pdf_custom(file, range_str)
     except Exception as e: raise gr.Error(str(e))
 
-# 3. Comprimir (NUEVO)
-def interface_compress(file, level_label):
+def process_reorder(file, order_str):
+    if not file: return None
+    try: return engine.reorder_pages(file, order_str)
+    except Exception as e: raise gr.Error(str(e))
+
+def process_compare(file_a, file_b):
+    if not file_a or not file_b: return None
+    try: return engine.compare_pdfs_visual(file_a, file_b)
+    except Exception as e: raise gr.Error(f"Error comparando: {str(e)}")
+
+def process_compress(file, level_label):
     if not file: return None
-    # Mapear label a int
+    # Mapeo de etiqueta UI a nivel Ghostscript
     levels = {
         "Baja (Máxima calidad)": 1, 
         "Media (Recomendado - eBook)": 3, 
@@ -56,19 +76,19 @@ def interface_compress(file, level_label):
     try: return engine.compress_pdf(file, power)
     except Exception as e: raise gr.Error(str(e))
 
-# 4. Word
-def interface_word(file):
+def process_word(file):
     if not file: return None
     try: return engine.pdf_to_word(file)
     except Exception as e: raise gr.Error(str(e))
 
-# 5. Rotar
 def update_rotate_preview(file, angle_label):
     if not file: return None
+    # Parsear "90° (Derecha)" -> 90
     angle = 0
     if "90" in angle_label: angle = 90
     elif "180" in angle_label: angle = 180
     elif "270" in angle_label: angle = 270
+    
     return engine.get_rotated_preview(file, angle)
 
 def process_rotate(file, angle_label):
@@ -77,83 +97,118 @@ def process_rotate(file, angle_label):
     if "90" in angle_label: angle = 90
     elif "180" in angle_label: angle = 180
     elif "270" in angle_label: angle = 270
-    if angle == 0: return file.name
+    
+    if angle == 0: return file.name # Devolver original si es 0
     try: return engine.rotate_pdf(file, angle)
     except Exception as e: raise gr.Error(str(e))
 
-# 6. Imágenes
-def interface_p2i(file):
-    if not file: return None
-    try: return engine.pdf_to_images_zip(file)
+def process_protect(file, pwd):
+    if not file or not pwd: return None
+    try: return engine.protect_pdf(file, pwd)
     except Exception as e: raise gr.Error(str(e))
 
-def interface_i2p(files):
-    if not files: return None
-    try: return engine.images_to_pdf(files)
+def process_extract_text(file):
+    if not file: return None
+    try: return engine.extract_text(file)
     except Exception as e: raise gr.Error(str(e))
 
-# 7. Extras
-def interface_meta(file, t, a, s):
+def process_metadata(file, t, a, s):
     if not file: return None
     try: return engine.update_metadata(file, t, a, s)
     except Exception as e: raise gr.Error(str(e))
 
-def interface_text(file):
+def process_p2i(file):
     if not file: return None
-    try: return engine.extract_text(file)
+    try: return engine.pdf_to_images_zip(file)
     except Exception as e: raise gr.Error(str(e))
 
-# 8. Proteger
-def interface_protect(file, pwd):
-    if not file or not pwd: return None
-    try: return engine.protect_pdf(file, pwd)
+def process_i2p(files):
+    if not files: return None
+    try: return engine.images_to_pdf(files)
     except Exception as e: raise gr.Error(str(e))
 
 
-# --- CONSTRUCCIÓN DE LA UI ---
+# --- CONSTRUCCIÓN DE LA INTERFAZ (LAYOUT) ---
+
 with gr.Blocks(title=config.APP_TITLE, theme=gr.themes.Soft()) as demo:
     gr.Markdown(f"# {config.APP_TITLE}")
     gr.Markdown(config.APP_DESCRIPTION)
     
-    # Pestañas aplanadas y ordenadas por uso
     with gr.Tabs():
         
-        # 1. UNIR
-        with gr.TabItem("Unir PDF"):
+        # ---------------------------------------------------------
+        # TAB 1: UNIR (PRIORIDAD ALTA)
+        # ---------------------------------------------------------
+        with gr.TabItem("Unir"):
             gr.Markdown(config.TXT_MERGE)
             with gr.Row():
                 with gr.Column(scale=1):
-                    m_files = gr.File(file_count="multiple", label="Archivos", file_types=[".pdf"])
+                    m_files = gr.File(file_count="multiple", label="Archivos PDF", file_types=[".pdf"])
                 with gr.Column(scale=2):
                     m_table = gr.Dataframe(headers=["ID", "Archivo"], interactive=False)
-                    m_order = gr.Textbox(label="Orden", placeholder="Ej: 0, 2, 1")
+                    m_order = gr.Textbox(label="Orden de unión (IDs)", placeholder="Ej: 0, 2, 1")
                     m_btn = gr.Button("Unir Archivos", variant="primary")
-            m_out = gr.File(label="Resultado")
+            m_out = gr.File(label="Resultado Unido")
+            
+            # Eventos
             m_files.change(update_file_list, m_files, [m_table, m_order])
             m_btn.click(process_merge, [m_files, m_order], m_out)
 
-        # 2. DIVIDIR
-        with gr.TabItem("Dividir PDF"):
-            gr.Markdown(config.TXT_SPLIT)
+        # ---------------------------------------------------------
+        # TAB 2: DIVIDIR Y REORDENAR (GESTIÓN DE PÁGINAS)
+        # ---------------------------------------------------------
+        with gr.TabItem("Dividir / Reordenar"):
             with gr.Row():
-                with gr.Column():
-                    s_file = gr.File(label="PDF", file_types=[".pdf"])
-                    s_info = gr.Markdown("")
-                    s_pages = gr.State(0)
-                    s_range = gr.Textbox(label="Rangos", placeholder="Ej: 1-3, 5")
-                    with gr.Row():
-                        s_prev_btn = gr.Button("Ver Hitos")
-                        s_btn = gr.Button("Dividir (ZIP)", variant="primary")
-                with gr.Column():
-                    s_gal = gr.Gallery(label="Vista Previa", height=300, object_fit="contain", columns=3)
-                    s_out = gr.File(label="ZIP Descargable", file_types=[".zip"])
-            s_file.change(load_pdf_info, s_file, [s_info, s_pages, s_out])
-            s_prev_btn.click(update_split_preview, [s_file, s_range, s_pages], s_gal)
-            s_btn.click(process_split, [s_file, s_range], s_out)
-
-        # 3. COMPRIMIR (NUEVO)
-        with gr.TabItem("Comprimir PDF"):
-            gr.Markdown("Reduce el tamaño de tu archivo PDF optimizando imágenes y recursos.")
+                # Panel izquierdo: Carga de archivo
+                with gr.Column(scale=1):
+                    dr_file = gr.File(label="PDF Origen", file_types=[".pdf"])
+                    dr_info = gr.Markdown("")
+                    dr_pages = gr.State(0) # Guarda num páginas
+                
+                # Panel derecho: Herramientas
+                with gr.Column(scale=2):
+                    with gr.Tab("Extraer (ZIP)"):
+                        gr.Markdown("Separa páginas en archivos individuales.")
+                        s_range = gr.Textbox(label="Rangos a extraer", placeholder="Ej: 1-3, 5")
+                        with gr.Row():
+                            s_prev_btn = gr.Button("Vista Previa")
+                            s_btn = gr.Button("Dividir y Descargar ZIP", variant="primary")
+                        s_gal = gr.Gallery(label="Preview", height=200, columns=4, object_fit="contain")
+                        s_out = gr.File(label="Archivo ZIP")
+                        
+                        s_prev_btn.click(update_split_preview, [dr_file, s_range, dr_pages], s_gal)
+                        s_btn.click(process_split, [dr_file, s_range], s_out)
+
+                    with gr.Tab("Reordenar (PDF)"):
+                        gr.Markdown("Crea un nuevo PDF cambiando el orden de las páginas.")
+                        r_order = gr.Textbox(label="Nuevo Orden", placeholder="Ej: 3, 1, 2, 4-10")
+                        r_btn = gr.Button("Reordenar Documento", variant="primary")
+                        r_out = gr.File(label="PDF Reordenado")
+                        
+                        r_btn.click(process_reorder, [dr_file, r_order], r_out)
+
+            # Carga de info automática
+            dr_file.change(load_pdf_info, dr_file, [dr_info, dr_pages, s_out])
+
+        # ---------------------------------------------------------
+        # TAB 3: COMPARAR (NUEVO)
+        # ---------------------------------------------------------
+        with gr.TabItem("Comparar PDFs"):
+            gr.Markdown("Sube dos versiones para detectar diferencias visuales (marcadas en **Magenta**).")
+            with gr.Row():
+                cmp_a = gr.File(label="Versión A (Original)", file_types=[".pdf"])
+                cmp_b = gr.File(label="Versión B (Modificada)", file_types=[".pdf"])
+            
+            cmp_btn = gr.Button("Generar Informe de Diferencias", variant="primary")
+            cmp_out = gr.File(label="PDF Comparativo")
+            
+            cmp_btn.click(process_compare, [cmp_a, cmp_b], cmp_out)
+
+        # ---------------------------------------------------------
+        # TAB 4: COMPRIMIR
+        # ---------------------------------------------------------
+        with gr.TabItem("Comprimir"):
+            gr.Markdown("Optimiza el tamaño del archivo.")
             with gr.Row():
                 with gr.Column():
                     c_file = gr.File(label="PDF Original", file_types=[".pdf"])
@@ -162,81 +217,79 @@ with gr.Blocks(title=config.APP_TITLE, theme=gr.themes.Soft()) as demo:
                         label="Nivel de Compresión", 
                         value="Media (Recomendado - eBook)"
                     )
-                    c_btn = gr.Button("Comprimir Ahora", variant="primary")
+                    c_btn = gr.Button("Comprimir", variant="primary")
                 with gr.Column():
                     c_out = gr.File(label="PDF Comprimido")
-            c_btn.click(interface_compress, [c_file, c_level], c_out)
+            
+            c_btn.click(process_compress, [c_file, c_level], c_out)
 
-        # 4. PDF A WORD
-        with gr.TabItem("PDF a Word"):
-            gr.Markdown("Convierte documentos a formato DOCX editable (Beta).")
-            with gr.Row():
-                w_file = gr.File(label="PDF", file_types=[".pdf"])
-                w_btn = gr.Button("Convertir a Word", variant="primary")
-                w_out = gr.File(label="Archivo Word")
-            w_btn.click(interface_word, w_file, w_out)
-
-        # 5. ROTAR
-        with gr.TabItem("Rotar PDF"):
-            gr.Markdown("Rota el documento permanentemente.")
-            with gr.Row():
-                with gr.Column():
-                    r_file = gr.File(label="PDF", file_types=[".pdf"])
-                    r_angle = gr.Radio(["0° (Original)", "90° (Derecha)", "180° (Invertir)", "270° (Izquierda)"], label="Rotación", value="0° (Original)")
-                    r_btn = gr.Button("Aplicar Rotación", variant="primary")
-                with gr.Column():
-                    r_prev = gr.Image(label="Preview", height=350, type="filepath")
-                    r_out = gr.File(label="PDF Rotado")
-            r_file.change(update_rotate_preview, [r_file, r_angle], r_prev)
-            r_angle.change(update_rotate_preview, [r_file, r_angle], r_prev)
-            r_btn.click(process_rotate, [r_file, r_angle], r_out)
-
-        # 6. IMAGENES
-        with gr.TabItem("Imágenes"):
-            gr.Markdown("Extraer imágenes de PDF o crear PDF desde imágenes.")
-            with gr.Row():
-                with gr.Column():
-                    gr.Markdown("### PDF -> JPG")
-                    p2i_f = gr.File(label="PDF")
-                    p2i_b = gr.Button("Convertir (ZIP)")
-                    p2i_o = gr.File(label="ZIP")
-                    p2i_b.click(interface_p2i, p2i_f, p2i_o)
-                with gr.Column():
-                    gr.Markdown("### JPG -> PDF")
-                    i2p_f = gr.File(label="Imágenes", file_count="multiple")
-                    i2p_b = gr.Button("Unir en PDF")
-                    i2p_o = gr.File(label="PDF")
-                    i2p_b.click(interface_i2p, i2p_f, i2p_o)
-
-        # 7. TEXTO Y METADATOS
-        with gr.TabItem("Texto/Info"):
-            gr.Markdown("Edita información interna o extrae texto plano.")
-            with gr.Row():
-                with gr.Column():
-                    gr.Markdown("### Extraer Texto")
-                    t_file = gr.File(label="PDF")
-                    t_btn = gr.Button("Obtener .txt")
-                    t_out = gr.File()
-                    t_btn.click(interface_text, t_file, t_out)
-                with gr.Column():
-                    gr.Markdown("### Editar Metadatos")
-                    meta_f = gr.File(label="PDF")
-                    meta_t = gr.Textbox(label="Título")
-                    meta_a = gr.Textbox(label="Autor")
-                    meta_s = gr.Textbox(label="Asunto")
-                    meta_b = gr.Button("Guardar")
-                    meta_o = gr.File()
-                    meta_b.click(interface_meta, [meta_f, meta_t, meta_a, meta_s], meta_o)
-
-        # 8. PROTEGER
-        with gr.TabItem("Proteger"):
-            gr.Markdown("Cifra tu documento con contraseña.")
-            with gr.Row():
-                p_file = gr.File(label="PDF")
-                p_pwd = gr.Textbox(label="Contraseña", type="password")
-                p_btn = gr.Button("Encriptar", variant="primary")
-                p_out = gr.File(label="PDF Protegido")
-            p_btn.click(interface_protect, [p_file, p_pwd], p_out)
+        # ---------------------------------------------------------
+        # TAB 5: CONVERTIR
+        # ---------------------------------------------------------
+        with gr.TabItem("Convertir"):
+            with gr.Tab("PDF a Word"):
+                gr.Markdown("Convierte a formato editable DOCX (Beta).")
+                w_file = gr.File(label="PDF")
+                w_btn = gr.Button("Convertir a Word")
+                w_out = gr.File(label="Documento Word")
+                w_btn.click(process_word, w_file, w_out)
+            
+            with gr.Tab("PDF <-> Imágenes"):
+                with gr.Row():
+                    with gr.Column():
+                        gr.Markdown("### PDF a JPG")
+                        p2i_f = gr.File(label="PDF")
+                        p2i_b = gr.Button("Obtener Imágenes (ZIP)")
+                        p2i_o = gr.File(label="ZIP")
+                        p2i_b.click(process_p2i, p2i_f, p2i_o)
+                    with gr.Column():
+                        gr.Markdown("### JPG a PDF")
+                        i2p_f = gr.File(label="Imágenes", file_count="multiple", file_types=["image"])
+                        i2p_b = gr.Button("Crear PDF")
+                        i2p_o = gr.File(label="PDF")
+                        i2p_b.click(process_i2p, i2p_f, i2p_o)
+
+        # ---------------------------------------------------------
+        # TAB 6: HERRAMIENTAS EXTRA
+        # ---------------------------------------------------------
+        with gr.TabItem("Extras"):
+            with gr.Tab("Rotar"):
+                with gr.Row():
+                    with gr.Column():
+                        rot_file = gr.File(label="PDF", file_types=[".pdf"])
+                        rot_ang = gr.Radio(["0° (Original)", "90° (Derecha)", "180° (Invertir)", "270° (Izquierda)"], label="Rotación", value="0° (Original)")
+                        rot_btn = gr.Button("Aplicar Rotación")
+                    with gr.Column():
+                        rot_prev = gr.Image(label="Vista Previa", height=300)
+                        rot_out = gr.File(label="PDF Rotado")
+                
+                rot_file.change(update_rotate_preview, [rot_file, rot_ang], rot_prev)
+                rot_ang.change(update_rotate_preview, [rot_file, rot_ang], rot_prev)
+                rot_btn.click(process_rotate, [rot_file, rot_ang], rot_out)
+
+            with gr.Tab("Proteger"):
+                with gr.Row():
+                    p_file = gr.File(label="PDF")
+                    p_pwd = gr.Textbox(label="Contraseña", type="password")
+                    p_btn = gr.Button("Encriptar")
+                    p_out = gr.File(label="Protegido")
+                p_btn.click(process_protect, [p_file, p_pwd], p_out)
+            
+            with gr.Tab("Metadatos y Texto"):
+                with gr.Row():
+                    with gr.Column():
+                        mt_file = gr.File(label="PDF")
+                        mt_t = gr.Textbox(label="Título")
+                        mt_a = gr.Textbox(label="Autor")
+                        mt_s = gr.Textbox(label="Asunto")
+                        mt_btn = gr.Button("Guardar Metadatos")
+                        mt_out = gr.File(label="PDF Editado")
+                        mt_btn.click(process_metadata, [mt_file, mt_t, mt_a, mt_s], mt_out)
+                    with gr.Column():
+                        gr.Markdown("### Extraer Texto Plano")
+                        txt_btn = gr.Button("Extraer .txt")
+                        txt_out = gr.File(label="Archivo de Texto")
+                        txt_btn.click(process_extract_text, mt_file, txt_out)
 
 if __name__ == "__main__":
     demo.launch()