Update app_vince.py

app_vince.py

@@ -1,17 +1,18 @@
 #!/usr/bin/env python3
 import gradio as gr
-import os
 from pathlib import Path
 from typing import List
 from services.vincie import VincieService
 
-# Inicializa serviço e materializa modelo antes da UI
 svc = VincieService()
 try:
-    # Permite forçar caminho direto definindo VINCIE_DIRECT_TO_CKPT=1 no ambiente
     svc.ensure_model()
 except Exception as e:
-    print(f"[app] Falha em ensure_model no boot: {e}")
+    print(f"[app] Falha em ensure_model: {e}")
+
+def _collect_media(outdir: Path):
+    exts = (".png",".jpg",".jpeg",".gif",".mp4")
+    return sorted(str(p) for p in outdir.rglob("*") if p.is_file() and p.suffix.lower() in exts)
 
 def ui_multi_turn(input_image: str, turns: List[str], seed: int = 1, steps: int = 50,
                   cfg_scale: float = 7.5, negative_prompt: str = "", resolution: int = 512,
@@ -29,12 +30,10 @@ def ui_multi_turn(input_image: str, turns: List[str], seed: int = 1, steps: int
             num_gpus=num_gpus,
             use_vae_slicing=True,
         )
-        generated_files = []
-        for ext in (".png", ".jpg", ".jpeg", ".mp4", ".gif"):
-            generated_files.extend(str(p) for p in Path(outdir).rglob(f"*{ext}") if p.is_file())
-        if not generated_files:
+        files = _collect_media(Path(outdir))
+        if not files:
             return gr.Gallery(value=None, label="Nenhum arquivo gerado encontrado. Verifique os logs.")
-        return gr.Gallery(value=generated_files, label="Resultados Gerados")
+        return gr.Gallery(value=files, label="Resultados Gerados")
     except Exception as e:
         print(f"[app] Erro na geração: {e}")
         return gr.Gallery(value=None, label=f"[Erro] {str(e)}")
@@ -56,10 +55,10 @@ def ui_text_to_video(input_image: str, prompt: str, seed: int = 1, steps: int =
             num_gpus=num_gpus,
             use_vae_slicing=True,
         )
-        video_files = [str(p) for p in Path(outdir).rglob("*.mp4") if p.is_file()]
-        if not video_files:
+        videos = [p for p in _collect_media(Path(outdir)) if p.endswith(".mp4")]
+        if not videos:
             return gr.Video(value=None, label="Nenhum vídeo gerado encontrado. Verifique os logs.")
-        return gr.Video(value=video_files[0], label="Vídeo Gerado")
+        return gr.Video(value=videos[0], label="Vídeo Gerado")
     except Exception as e:
         print(f"[app] Erro na geração: {e}")
         return gr.Video(value=None, label=f"[Erro] {str(e)}")