Update app_vince.py

app_vince.py

@@ -1,20 +1,41 @@
 #!/usr/bin/env python3
 """
-VINCIE Service UI (Gradio) — in-process (vince_server)
+VINCIE Service UI (Gradio) — in-process
 
-- Bootstrap: garante repositório e checkpoint antes de importar o servidor.
-- UI minimalista: apenas galeria e vídeo; logs e 'ls' no terminal.
+- Importa diretamente o servidor in-process (vince_server).
+- UI minimalista: galeria e vídeo por aba.
 - Opções avançadas: steps, cfg_scale, aspect_ratio, resolution.
+- Logs e 'ls' de diagnóstico apenas no terminal.
+
+Pré-requisitos garantidos pelo start.sh:
+- Repo VINCIE íntegro e no PYTHONPATH (VINCIE_DIR).
+- Symlink ckpt/VINCIE-3B -> snapshot (contém dit.pth, vae.pth, llm14b).
+- Apex/Flash e demais deps instaladas pelo builder.
 """
 
 import os
-import importlib
 from pathlib import Path
 from typing import List, Tuple, Optional
 
 import gradio as gr
 
-# ---------- Bootstrap do repositório/ckpt antes de importar o servidor ----------
+# Servidor persistente (pipeline aquecida)
+from services.vince_server import server
+
+
+# -------- util --------
+
+def _list_media(out_dir: Path, max_images: int = 24) -> Tuple[List[str], Optional[str]]:
+    img_globs = ("*.png", "*.jpg", "*.jpeg", "*.webp")
+    images: List[Path] = []
+    for pat in img_globs:
+        images += list(out_dir.rglob(pat))
+    images = sorted(images, key=lambda p: p.stat().st_mtime) if images else []
+    image_paths = [str(p) for p in images[-max_images:]] if images else []
+    videos = sorted(out_dir.rglob("*.mp4"), key=lambda p: p.stat().st_mtime)
+    video_path = str(videos[-1]) if videos else None
+    return image_paths, video_path
+
 
 def _debug_ls(paths: List[Path], depth: int = 2) -> None:
     try:
@@ -45,46 +66,8 @@ def _debug_ls(paths: List[Path], depth: int = 2) -> None:
     except Exception as e:
         print(f"[debug-ls] falha geral: {e}")
 
-# Garante repo e ckpt; evita ModuleNotFoundError: common
-try:
-    from services.vincie import VincieService  # usa o serviço CLI apenas para preparar o ambiente
-    print("[bootstrap] garantindo repositório e checkpoint...")
-    _svc = VincieService()
-    _svc.ensure_repo()
-    _svc.ensure_model()
-    os.environ["VINCIE_DIR"] = str(_svc.repo_dir)
-    print(f"[bootstrap] repo={_svc.repo_dir} ckpt={_svc.ckpt_dir}")
-except Exception as e:
-    print(f"[bootstrap] falha no ensure_repo/ensure_model: {e}")
-    _debug_ls([Path("/app"), Path("/app/VINCIE"), Path("/app/VINCIE/ckpt"), Path("/data/.cache/huggingface/hub")], depth=2)
-
-# Tenta importar o servidor in-process; fallback via importlib se 'common' ainda não estiver no path
-try:
-    from services.vince_server import server  # instancia singleton internamente
-except Exception as e:
-    print(f"[import] primeira tentativa falhou: {e}")
-    try:
-        server = importlib.import_module("services.vince_server").server
-        print("[import] servidor importado via importlib com sucesso")
-    except Exception as ee:
-        print(f"[import] falha ao importar services.vince_server: {ee}")
-        _debug_ls([Path("/app/VINCIE")], depth=2)
-        raise
 
-# ---------- Utilitários de mídia ----------
-
-def _list_media(out_dir: Path, max_images: int = 24) -> Tuple[List[str], Optional[str]]:
-    img_globs = ("*.png", "*.jpg", "*.jpeg", "*.webp")
-    images: List[Path] = []
-    for pat in img_globs:
-        images += list(out_dir.rglob(pat))
-    images = sorted(images, key=lambda p: p.stat().st_mtime) if images else []
-    image_paths = [str(p) for p in images[-max_images:]] if images else []
-    videos = sorted(out_dir.rglob("*.mp4"), key=lambda p: p.stat().st_mtime)
-    video_path = str(videos[-1]) if videos else None
-    return image_paths, video_path
-
-# ---------- Handlers ----------
+# -------- UI handlers --------
 
 def ui_multi_turn(
     input_image: Optional[str],
@@ -183,7 +166,8 @@ def ui_multi_concept(
         _debug_ls([out_path], depth=2)
     return imgs, vid
 
-# ---------- UI ----------
+
+# -------- UI --------
 
 with gr.Blocks(title="VINCIE Service (in-process)") as demo:
     gr.Markdown(
@@ -196,7 +180,8 @@ with gr.Blocks(title="VINCIE Service (in-process)") as demo:
         )
     )
 
-    with gr.Accordion("Opções Avançadas (para Abas 1 e 2)", open=False):
+    # Opções avançadas (compartilhadas entre abas)
+    with gr.Accordion("Opções Avançadas", open=False):
         steps_input = gr.Slider(label="Passos de Inferência", minimum=10, maximum=100, step=1, value=50)
         cfg_scale_input = gr.Slider(label="Escala de Orientação (CFG)", minimum=1.0, maximum=15.0, step=0.5, value=7.5)
         aspect_ratio_input = gr.Dropdown(
@@ -206,6 +191,7 @@ with gr.Blocks(title="VINCIE Service (in-process)") as demo:
         )
         resolution_input = gr.Slider(label="Resolução (lado menor)", minimum=256, maximum=1024, step=64, value=512)
 
+    # Aba 1 — Multi-turn Editing
     with gr.Tab("Multi-turn Editing"):
         with gr.Row():
             img = gr.Image(type="filepath", label="Imagem de entrada")
@@ -219,6 +205,7 @@ with gr.Blocks(title="VINCIE Service (in-process)") as demo:
             outputs=[out_gallery, out_video],
         )
 
+    # Aba 2 — Multi-concept Composition
     with gr.Tab("Multi-concept Composition"):
         files = gr.File(file_count="multiple", type="filepath", label="Imagens de conceito")
         descs = gr.Textbox(lines=8, label="Descrições (uma por linha, mesma ordem das imagens)")
@@ -233,7 +220,11 @@ with gr.Blocks(title="VINCIE Service (in-process)") as demo:
         )
 
 if __name__ == "__main__":
-    
+    if os.getenv("GRADIO_ENABLE_QUEUE", "True").lower() == "true" and hasattr(demo, "queue"):
+        demo.queue(
+            concurrency_count=int(os.getenv("GRADIO_QUEUE_CONCURRENCY", "2")),
+            max_size=int(os.getenv("GRADIO_QUEUE_MAX", "32")),
+        )
     demo.launch(
         server_name=os.getenv("GRADIO_SERVER_NAME", "0.0.0.0"),
         server_port=int(os.getenv("GRADIO_SERVER_PORT", os.getenv("PORT", "7860"))),