Update app.py

app.py

@@ -1,4 +1,4 @@
-# Versión 1.1: UI Avanzada con gestión de estado
+# Versión 1.2: Corrección de UI y Preview Avanzada
 import gradio as gr
 import pandas as pd
 import config
@@ -6,130 +6,116 @@ from core import PDFEngine
 
 engine = PDFEngine()
 
-# --- FUNCIONES AUXILIARES UNIR ---
+# --- FUNCIONES UNIR ---
 def update_file_list(files):
-    """Crea un Dataframe para mostrar los archivos subidos y sus índices."""
-    if not files:
-        return pd.DataFrame(), ""
-    
-    # files es una lista de objetos o rutas dependiendo de la config. 
-    # En Gradio 4, si file_count="multiple", suele ser lista de rutas.
-    data = []
-    for idx, f_path in enumerate(files):
-        name = f_path.split("/")[-1] # Obtener nombre simple
-        data.append([idx, name])
-    
+    if not files: return pd.DataFrame(), ""
+    data = [[i, f.split("/")[-1]] for i, f in enumerate(files)]
     default_order = ",".join([str(i) for i in range(len(files))])
-    return pd.DataFrame(data, columns=["ID", "Nombre del Archivo"]), default_order
+    return pd.DataFrame(data, columns=["ID", "Archivo"]), default_order
 
 def process_merge_ordered(files, order_str):
-    if not files:
-        return None
+    if not files: return None
     try:
-        # Convertir string "0, 2, 1" a lista de enteros [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(f"Error en el orden: {str(e)}")
+        raise gr.Error(f"Error: {str(e)}")
 
-# --- FUNCIONES AUXILIARES DIVIDIR ---
+# --- FUNCIONES DIVIDIR ---
 def load_pdf_info(file):
-    if not file:
-        return None, None, gr.update(visible=False)
-    
+    if not file: return None, 0, gr.update(visible=False)
     info = engine.get_pdf_info(file)
-    msg = f"Documento cargado: {info['name']} ({info['pages']} páginas)."
-    return msg, info['pages'], gr.update(visible=True)
+    return f"Cargado: {info['name']} ({info['pages']} págs).", info['pages'], gr.update(visible=True)
 
 def update_preview(file, range_str, total_pages):
-    if not file or not range_str:
-        return None
+    if not file or not range_str: return None
     
-    # Parseamos para ver qué páginas quiere el usuario
-    # Solo previsualizamos la primera y última del rango para no saturar memoria
-    indices = engine.parse_range_string(range_str, total_pages)
-    if not indices:
-        return None
+    # NUEVO: Obtenemos los hitos clave (inicio/fin) de cada rango
+    key_pages = engine.get_preview_indices_from_string(range_str, total_pages)
     
-    first_idx = indices[0] + 1 # Convertir a 1-based para visualización
-    last_idx = indices[-1] + 1
+    if not key_pages: return None
     
-    images = []
-    # Generar imagen de inicio
-    img1 = engine.generate_preview(file, first_idx)
-    if img1: images.append((img1, f"Página {first_idx}"))
+    # Limitamos a 8 previews para no saturar
+    key_pages = key_pages[:8]
     
-    # Generar imagen de fin si es distinta
-    if last_idx != first_idx:
-        img2 = engine.generate_preview(file, last_idx)
-        if img2: images.append((img2, f"Página {last_idx}"))
-        
-    return images
+    gallery_data = []
+    for page_num in key_pages:
+        img_path = engine.generate_preview(file, page_num)
+        if img_path:
+            gallery_data.append((img_path, f"Pág {page_num}"))
+            
+    return gallery_data
 
 def process_split_range(file, range_str):
-    return engine.split_pdf_custom(file, range_str)
+    if not file: return None
+    try:
+        return engine.split_pdf_custom(file, range_str)
+    except Exception as e:
+        raise gr.Error(str(e))
+
+# --- FUNCIONES PROTEGER ---
+def interface_protect(file, password):
+    if not file or not password: return None
+    try:
+        return engine.protect_pdf(file, password)
+    except Exception as e:
+        raise gr.Error(str(e))
 
 # --- UI PRINCIPAL ---
 with gr.Blocks(title=config.APP_TITLE, theme=gr.themes.Soft()) as demo:
     gr.Markdown(f"# {config.APP_TITLE}")
     
     with gr.Tabs():
-        # TAB 1: UNIR (Con reordenamiento manual)
+        # TAB 1: UNIR
         with gr.TabItem("Unir PDF"):
-            gr.Markdown("Sube los archivos, revisa sus IDs en la tabla y escribe el orden deseado (ej: 2,0,1).")
-            
+            gr.Markdown(config.TXT_MERGE)
             with gr.Row():
                 with gr.Column(scale=1):
                     merge_files = gr.File(file_count="multiple", label="Archivos", file_types=[".pdf"])
                 with gr.Column(scale=2):
-                    file_table = gr.Dataframe(headers=["ID", "Nombre del Archivo"], interactive=False)
-                    order_input = gr.Textbox(label="Orden de unión (IDs separados por coma)")
-                    merge_btn = gr.Button("Unir en este orden", variant="primary")
-            
+                    file_table = gr.Dataframe(headers=["ID", "Archivo"], interactive=False)
+                    order_input = gr.Textbox(label="Orden (IDs separados por coma)", placeholder="Ej: 0, 2, 1")
+                    merge_btn = gr.Button("Unir PDF", variant="primary")
             merge_output = gr.File(label="Resultado")
-
-            # Eventos
+            
             merge_files.change(fn=update_file_list, inputs=merge_files, outputs=[file_table, order_input])
             merge_btn.click(fn=process_merge_ordered, inputs=[merge_files, order_input], outputs=merge_output)
 
-        # TAB 2: DIVIDIR (Con Rangos y Preview)
-        with gr.TabItem("Extraer/Dividir PDF"):
-            gr.Markdown("Extrae páginas específicas. Ejemplos de rango: '1-5', '1,3,5', '1-3, 8'.")
-            
+        # TAB 2: DIVIDIR (Mejorado)
+        with gr.TabItem("Dividir PDF"):
+            gr.Markdown(config.TXT_SPLIT)
             with gr.Row():
                 with gr.Column():
                     split_file = gr.File(label="PDF Origen", file_types=[".pdf"])
-                    info_text = gr.Markdown("Esperando archivo...")
-                    # Estado oculto para guardar número total de páginas
-                    page_count_state = gr.State(0) 
-                    
-                    range_input = gr.Textbox(label="Rango de páginas", placeholder="Ej: 1-3, 5")
-                    preview_btn = gr.Button("Previsualizar Selección")
-                    split_btn = gr.Button("Extraer Páginas", variant="primary")
+                    info_txt = gr.Markdown("")
+                    page_count = gr.State(0)
+                    range_in = gr.Textbox(label="Rangos", placeholder="Ej: 1-3, 5, 8-10")
                     
+                    with gr.Row():
+                        prev_btn = gr.Button("Ver Páginas Clave")
+                        split_btn = gr.Button("Extraer", variant="primary")
+                
                 with gr.Column():
-                    preview_gallery = gr.Gallery(label="Vista previa (Inicio/Fin)", columns=2, height=300, object_fit="contain")
-                    split_output = gr.File(label="PDF Extraído")
+                    # Galería dinámica para mostrar múltiples páginas
+                    preview_gal = gr.Gallery(label="Hitos del Rango", columns=3, height=350, object_fit="contain")
+                    split_out = gr.File(label="PDF Resultante")
 
-            # Eventos
-            split_file.change(fn=load_pdf_info, inputs=split_file, outputs=[info_text, page_count_state, split_output])
-            
-            preview_btn.click(
-                fn=update_preview, 
-                inputs=[split_file, range_input, page_count_state], 
-                outputs=preview_gallery
-            )
-            
-            split_btn.click(
-                fn=process_split_range, 
-                inputs=[split_file, range_input], 
-                outputs=split_output
-            )
+            split_file.change(fn=load_pdf_info, inputs=split_file, outputs=[info_txt, page_count, split_out])
+            prev_btn.click(fn=update_preview, inputs=[split_file, range_in, page_count], outputs=preview_gal)
+            split_btn.click(fn=process_split_range, inputs=[split_file, range_in], outputs=split_out)
 
-        # TAB 3: PROTEGER (Igual que antes)
+        # TAB 3: PROTEGER (Restaurado)
         with gr.TabItem("Proteger PDF"):
-            # (Mantener el código de la UI de proteger de la versión 1.0)
-            pass 
+            gr.Markdown(config.TXT_PROTECT)
+            with gr.Row():
+                with gr.Column():
+                    prot_file = gr.File(label="PDF a proteger", file_types=[".pdf"])
+                    pass_in = gr.Textbox(label="Contraseña", type="password")
+                    prot_btn = gr.Button("Encriptar", variant="primary")
+                with gr.Column():
+                    prot_out = gr.File(label="PDF Encriptado")
+            
+            prot_btn.click(fn=interface_protect, inputs=[prot_file, pass_in], outputs=prot_out)
 
 if __name__ == "__main__":
     demo.launch()