Update app.py

app.py

@@ -48,14 +48,21 @@ def procesar_video(video):
         if frame_count == 0:
             raise gr.Error("No se pudieron extraer fotogramas del video")
 
-        # Crear collage de 4 fotogramas aleatorios
+        # Crear collage de 4 fotogramas aleatorios con separación
         selected_frames = random.sample(frame_paths, min(4, len(frame_paths)))
         images = [Image.open(img) for img in selected_frames]
-        collage = Image.new('RGB', (images[0].width * 2, images[0].height * 2))
+        margin = 10
+        collage_width = images[0].width * 2 + margin
+        collage_height = images[0].height * 2 + margin
+        collage = Image.new('RGB', (collage_width, collage_height), color=(255,255,255))
+        positions = [
+            (0, 0),
+            (images[0].width + margin, 0),
+            (0, images[0].height + margin),
+            (images[0].width + margin, images[0].height + margin)
+        ]
         for i, img in enumerate(images):
-            row = i // 2
-            col = i % 2
-            collage.paste(img, (col * images[0].width, row * images[0].height))
+            collage.paste(img, positions[i])
         collage_path = os.path.join(temp_dir, "collage.jpg")
         collage.save(collage_path)
 
@@ -134,7 +141,8 @@ with gr.Blocks(title="Extracción de Fotogramas Forenses") as demo:
     
     with gr.Row():
         with gr.Column():
-            video_input = gr.Video(label="Subir Video (solo archivos de video)", interactive=True)
+            # Desactivar opciones de webcam y edición (tijera) estableciendo source="upload"
+            video_input = gr.Video(source="upload", label="Subir Video (solo archivos de video)", interactive=True)
             # Ambos botones inician desactivados
             procesar_btn = gr.Button("Procesar Fotogramas", interactive=False)
         with gr.Column():
@@ -186,9 +194,12 @@ with gr.Blocks(title="Extracción de Fotogramas Forenses") as demo:
         outputs=[gallery_output, download_file, temp_dir_state, zip_path_state, download_btn, procesar_btn],
     )
     
-    # Al pulsar "DESCARGAR FOTOGRAMAS": se envía la ruta del ZIP generado.
+    # Función para activar la descarga: actualiza el componente File con la ruta del ZIP y lo hace visible
+    def trigger_download(zip_path):
+        return gr.update(value=zip_path, visible=True)
+    
     download_btn.click(
-        fn=lambda zip_path: zip_path,
+        fn=trigger_download,
         inputs=zip_path_state,
         outputs=download_file,
     )