app_vince.py

#!/usr/bin/env python3
"""
VINCIE Service UI (Gradio) — in-process

- Importa o servidor in-process e mantém uma instância única (GPU quente).
- UI minimalista: galeria e vídeo por aba.
- Opções avançadas: steps, cfg_scale, aspect_ratio, resolution.
"""

import os
from pathlib import Path
from typing import List, Tuple, Optional

import gradio as gr

# Servidor persistente (pipeline aquecida)
from services.vince_server import VinceServer
server = VinceServer()  # instancia única e persistente (evita erro de 'self' e aquece GPU)

# -------- 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

# -------- UI handlers --------
def ui_multi_turn(
    input_image: Optional[str],
    turns_text: Optional[str],
    steps_input: int,
    cfg_scale_input: float,
    aspect_ratio_input: str,
    resolution_input: int,
):
    if not input_image or not str(input_image).strip():
        print("[multi_turn] input_image ausente")
        return [], None
    if not turns_text or not turns_text.strip():
        print("[multi_turn] turns ausentes")
        return [], None

    turns = [ln.strip() for ln in turns_text.splitlines() if ln.strip()]

    try:
        out_dir = server.generate_multi_turn(
            input_image=input_image,
            turns=turns,
            cfg_scale=float(cfg_scale_input) if cfg_scale_input is not None else None,
            aspect_ratio=str(aspect_ratio_input) if aspect_ratio_input else None,
            resolution=int(resolution_input) if resolution_input is not None else None,
            steps=int(steps_input) if steps_input is not None else None,
        )
    except Exception as e:
        print(f"[multi_turn] erro: {e}")
        return [], None

    out_path = Path(out_dir)
    if not out_path.exists():
        print(f"[multi_turn] saída inexistente: {out_path}")
        return [], None

    imgs, vid = _list_media(out_path)
    if not imgs and not vid:
        print(f"[multi_turn] sem mídias em {out_path}")
    return imgs, vid

def ui_multi_concept(
    files: Optional[List[str]],
    descs_text: Optional[str],
    final_prompt: Optional[str],
    steps_input: int,
    cfg_scale_input: float,
    aspect_ratio_input: str,
    resolution_input: int,
):
    if not files:
        print("[multi_concept] files ausentes")
        return [], None
    if not descs_text or not descs_text.strip():
        print("[multi_concept] descs ausentes")
        return [], None
    if not final_prompt or not final_prompt.strip():
        print("[multi_concept] final_prompt ausente")
        return [], None

    descs = [ln.strip() for ln in descs_text.splitlines() if ln.strip()]
    if len(descs) != len(files):
        print(f"[multi_concept] mismatch descs({len(descs)}) vs files({len(files)})")
        return [], None

    try:
        out_dir = server.generate_multi_concept(
            concept_images=files,
            concept_prompts=descs,
            final_prompt=final_prompt,
            cfg_scale=float(cfg_scale_input) if cfg_scale_input is not None else None,
            aspect_ratio=str(aspect_ratio_input) if aspect_ratio_input else None,
            resolution=int(resolution_input) if resolution_input is not None else None,
            steps=int(steps_input) if steps_input is not None else None,
            pad_placeholder=False,
        )
    except Exception as e:
        print(f"[multi_concept] erro: {e}")
        return [], None

    out_path = Path(out_dir)
    if not out_path.exists():
        print(f"[multi_concept] saída inexistente: {out_path}")
        return [], None

    imgs, vid = _list_media(out_path)
    if not imgs and not vid:
        print(f"[multi_concept] sem mídias em {out_path}")
    return imgs, vid

# -------- UI --------
with gr.Blocks(title="VINCIE 🎨") as demo:
    gr.Markdown(
        "\n".join(
            [
                "🎨 VINCIE — Edição Multi-Turn e Composição Multi-Concept",
            ]
        )
    )

    # Opções Avançadas
    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(
            label="Aspect Ratio",
            choices=["keep_ratio", "1:1", "16:9", "9:16", "4:3", "3:4"],
            value="keep_ratio",
        )
        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")
        turns = gr.Textbox(lines=8, label="Instruções (uma por linha)")
        run1 = gr.Button("Gerar")
        out_gallery = gr.Gallery(label="Imagens", columns=4, height="auto")
        #out_video = gr.Video(label="Vídeo (se houver)")
        run1.click(
            ui_multi_turn,
            inputs=[img, turns, steps_input, cfg_scale_input, aspect_ratio_input, resolution_input],
            outputs=[out_gallery],
        )

    # Aba 2 — Multi-concept Composition
    with gr.Tab("Multi-concept Composition"):
        with gr.Row():
            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)")
        final_prompt = gr.Textbox(lines=2, label="Prompt final")
        run2 = gr.Button("Gerar")
        out_gallery2 = gr.Gallery(label="Imagens", columns=4, height="auto")
        #out_video2 = gr.Video(label="Vídeo (se houver)")
        run2.click(
            ui_multi_concept,
            inputs=[files, descs, final_prompt, steps_input, cfg_scale_input, aspect_ratio_input, resolution_input],
            outputs=[out_gallery2],
        )

    
if __name__ == "__main__":
    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"))),
        allowed_paths=["/app/outputs", "/app/ckpt"],
        show_error=True,
    )