Update app.py

app.py

@@ -1,54 +1,47 @@
+import os
+import zipfile
+import tempfile
+from pathlib import Path
+
 import gradio as gr
-import torch
-import numpy as np
-from PIL import Image, ImageFilter
+from PIL import Image
 from transformers import pipeline
-import zipfile
-import os
 
-# Carregando o modelo de alta precisão
-pipe = pipeline(task="depth-estimation", model="depth-anything/Depth-Anything-V2-Small-hf")
-
-def generate_binary_armor_depth(files):
-    if not files: return None
-    
-    output_dir = "parallax_depths"
-    os.makedirs(output_dir, exist_ok=True)
-    zip_name = "Leva_Parallax_Binary_V5.zip"
-    
-    with zipfile.ZipFile(zip_name, 'w') as zipf:
-        for file in files:
-            img = Image.open(file.name).convert("RGB")
-            
-            # 1. Gera a Profundidade Base
-            depth_raw = pipe(img)["depth"]
-            
-            # --- UPDATE V5.1: BINARY ARMOR (CORRIGIDO) ---
-            # O tamanho do MaxFilter DEVE ser ímpar (3, 5, 7, 9, 11, 13...)
-            # Ajustado para 11 para garantir blindagem total sem erro de sistema.
-            depth_dilated = depth_raw.filter(ImageFilter.MaxFilter(11))
-            
-            # 3. SEM BLUR: 
-            # Mantemos as bordas secas para o Parallax V15 (HTML) não derreter a linha.
-            depth_final = depth_dilated 
-            # ---------------------------------------------
-            
-            # Salva no ZIP
-            base_name = os.path.basename(file.name)
-            mask_name = f"depth_{base_name}"
-            mask_path = os.path.join(output_dir, mask_name)
-            
-            depth_final.save(mask_path)
-            zipf.write(mask_path, mask_name)
-            
-    return zip_name
-
-demo = gr.Interface(
-    fn=generate_binary_armor_depth,
-    inputs=gr.File(file_count="multiple", label="1. Envie suas artes 2.0"),
-    outputs=gr.File(label="2. Baixe o ZIP (Máscaras Blindadas V5.1)"),
-    title="🎥 Eternity Engine v5.1 - Binary Armor Generator",
-    description="CORREÇÃO: Filtro ajustado para tamanho ímpar (11). Gera máscaras blindadas para o Parallax V15."
+# Depth Anything V2 é uma boa escolha para depth monocular.
+# A API do Transformers mostra o uso com pipeline("depth-estimation") e pipe(image)["depth"].
+pipe = pipeline(
+    task="depth-estimation",
+    model="depth-anything/Depth-Anything-V2-Small-hf"
 )
 
-demo.launch()
+def generate_cinematic_depth(files):
+    if not files:
+        return None
+
+    out_dir = Path(tempfile.mkdtemp(prefix="parallax_depth_"))
+    zip_path = out_dir / "Leva_Parallax_Pro.zip"
+
+    with zipfile.ZipFile(zip_path, "w", compression=zipfile.ZIP_DEFLATED) as zipf:
+        for file_obj in files:
+            img = Image.open(file_obj.name).convert("RGB")
+            depth_img = pipe(img)["depth"]
+
+            base_name = Path(file_obj.name).stem
+            depth_name = f"depth_{base_name}.png"
+            depth_path = out_dir / depth_name
+
+            depth_img.save(depth_path)
+            zipf.write(depth_path, arcname=depth_name)
+
+    return str(zip_path)
+
+with gr.Blocks() as demo:
+    gr.Markdown("# 🎥 Eternity Engine v3.0 — Cinematic Depth")
+    gr.Markdown("Envie uma leva de imagens e baixe um ZIP com os mapas de profundidade.")
+    files = gr.File(file_count="multiple", label="1. Envie a leva de imagens")
+    out = gr.File(label="2. Baixe as máscaras de profundidade (ZIP)")
+    btn = gr.Button("Gerar Depth Maps")
+
+    btn.click(generate_cinematic_depth, inputs=files, outputs=out)
+
+demo.launch()