Update app.py

app.py

@@ -9,19 +9,15 @@ import subprocess
 # Crear carpetas necesarias
 INPUT_DIR = "batch_images"
 OUTPUT_DIR = "batch_results"
-RESULTS_DIR = "comparaciones"
 os.makedirs(INPUT_DIR, exist_ok=True)
 os.makedirs(OUTPUT_DIR, exist_ok=True)
-os.makedirs(RESULTS_DIR, exist_ok=True)
 
 def face_swap_batch(rb_img, lote_imgs):
     # Limpiar carpetas anteriores
     shutil.rmtree(INPUT_DIR, ignore_errors=True)
     shutil.rmtree(OUTPUT_DIR, ignore_errors=True)
-    shutil.rmtree(RESULTS_DIR, ignore_errors=True)
     os.makedirs(INPUT_DIR, exist_ok=True)
     os.makedirs(OUTPUT_DIR, exist_ok=True)
-    os.makedirs(RESULTS_DIR, exist_ok=True)
 
     # Guardar rostro base
     rb_path = os.path.join(INPUT_DIR, "RB.jpg")
@@ -49,23 +45,15 @@ def face_swap_batch(rb_img, lote_imgs):
             "--frame-processor", "face_swapper", "face_enhancer"
         ])
 
-    # Comparaciones lado a lado
+    # Devolver solo las imágenes generadas
     resultados = []
     for path in img_paths:
         filename = os.path.basename(path)
-        original = Image.open(path).resize((256, 256))
         generado_path = os.path.join(OUTPUT_DIR, f"swapped_{filename}")
-        if not os.path.exists(generado_path):
-            continue
-        generado = Image.open(generado_path).resize((256, 256))
-        combinado = Image.new("RGB", (512, 256))
-        combinado.paste(original, (0, 0))
-        combinado.paste(generado, (256, 0))
-        out_path = os.path.join(RESULTS_DIR, f"comparado_{filename}")
-        combinado.save(out_path)
-        resultados.append(out_path)
+        if os.path.exists(generado_path):
+            resultados.append(generado_path)
 
-    # Devolver resultado: zip o imagen única
+    # ZIP o imagen suelta
     if len(resultados) == 1:
         return resultados[0]
     else:
@@ -82,7 +70,7 @@ gr.Interface(
         gr.Image(label="🧠 Rostro base", type="pil"),
         gr.File(label="🖼️ Imágenes por lote", file_types=["image"], file_count="multiple")
     ],
-    outputs=gr.File(label="📦 Descargar ZIP o imagen"),
+    outputs=gr.File(label="📦 Descargar resultado"),
     title="Deep Fake Lote - Yepo Hz",
-    description="Sube una imagen de rostro base y varias imágenes objetivo. El sistema aplicará el cambio de rostro en lote. Resultado visual lado a lado."
+    description="Sube una imagen de rostro base y varias imágenes objetivo. El sistema aplicará el cambio de rostro en lote y te permitirá descargar las imágenes finales."
 ).launch()