Update app.py

app.py

@@ -1,4 +1,4 @@
-# Versión 1.2: Corrección de UI y Preview Avanzada
+# Versión 1.3: UI con descarga de ZIP
 import gradio as gr
 import pandas as pd
 import config
@@ -6,7 +6,8 @@ from core import PDFEngine
 
 engine = PDFEngine()
 
-# --- FUNCIONES UNIR ---
+# --- FUNCIONES AUXILIARES ---
+# (Se mantienen igual que v1.2: update_file_list, process_merge_ordered, load_pdf_info, update_preview)
 def update_file_list(files):
     if not files: return pd.DataFrame(), ""
     data = [[i, f.split("/")[-1]] for i, f in enumerate(files)]
@@ -18,10 +19,8 @@ def process_merge_ordered(files, order_str):
     try:
         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: {str(e)}")
+    except Exception as e: raise gr.Error(f"Error: {str(e)}")
 
-# --- FUNCIONES DIVIDIR ---
 def load_pdf_info(file):
     if not file: return None, 0, gr.update(visible=False)
     info = engine.get_pdf_info(file)
@@ -29,37 +28,27 @@ def load_pdf_info(file):
 
 def update_preview(file, range_str, total_pages):
     if not file or not range_str: return None
-    
-    # NUEVO: Obtenemos los hitos clave (inicio/fin) de cada rango
     key_pages = engine.get_preview_indices_from_string(range_str, total_pages)
-    
     if not key_pages: return None
-    
-    # Limitamos a 8 previews para no saturar
     key_pages = key_pages[:8]
-    
     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}"))
-            
+        if img_path: gallery_data.append((img_path, f"Pág {page_num}"))
     return gallery_data
 
 def process_split_range(file, range_str):
     if not file: return None
     try:
+        # Ahora devuelve la ruta de un .zip
         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))
+    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:
@@ -74,14 +63,14 @@ with gr.Blocks(title=config.APP_TITLE, theme=gr.themes.Soft()) as demo:
                     merge_files = gr.File(file_count="multiple", label="Archivos", file_types=[".pdf"])
                 with gr.Column(scale=2):
                     file_table = gr.Dataframe(headers=["ID", "Archivo"], interactive=False)
-                    order_input = gr.Textbox(label="Orden (IDs separados por coma)", placeholder="Ej: 0, 2, 1")
+                    order_input = gr.Textbox(label="Orden (IDs)", placeholder="Ej: 0, 2, 1")
                     merge_btn = gr.Button("Unir PDF", variant="primary")
-            merge_output = gr.File(label="Resultado")
+            merge_output = gr.File(label="PDF Unido")
             
             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 (Mejorado)
+        # TAB 2: DIVIDIR (Salida ZIP)
         with gr.TabItem("Dividir PDF"):
             gr.Markdown(config.TXT_SPLIT)
             with gr.Row():
@@ -89,22 +78,22 @@ with gr.Blocks(title=config.APP_TITLE, theme=gr.themes.Soft()) as demo:
                     split_file = gr.File(label="PDF Origen", file_types=[".pdf"])
                     info_txt = gr.Markdown("")
                     page_count = gr.State(0)
-                    range_in = gr.Textbox(label="Rangos", placeholder="Ej: 1-3, 5, 8-10")
+                    range_in = gr.Textbox(label="Rangos a separar", placeholder="Ej: 1-3, 5, 8-9")
                     
                     with gr.Row():
-                        prev_btn = gr.Button("Ver Páginas Clave")
-                        split_btn = gr.Button("Extraer", variant="primary")
+                        prev_btn = gr.Button("Ver Hitos")
+                        split_btn = gr.Button("Dividir y Descargar ZIP", variant="primary")
                 
                 with gr.Column():
-                    # 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")
+                    preview_gal = gr.Gallery(label="Vista Previa", columns=3, height=350, object_fit="contain")
+                    # Cambiamos etiqueta para clarificar
+                    split_out = gr.File(label="Archivo ZIP con divisiones", file_types=[".zip"])
 
             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 (Restaurado)
+        # TAB 3: PROTEGER
         with gr.TabItem("Proteger PDF"):
             gr.Markdown(config.TXT_PROTECT)
             with gr.Row():