#!/usr/bin/env bash
set -euo pipefail

: "${APP_DIR:=/app}"

export PYTORCH_CUDA_ALLOC_CONF="expandable_segments:True,max_split_size_mb:128"
export CUDA_MODULE_LOADING="LAZY"

echo "======================================================="
echo " LTXQ8 - Start (LTX-Q8, 8x L40S)"
echo "======================================================="


# 1) Builder (garante Apex/Flash e deps CUDA)
#echo "🛠️  Iniciando o builder.sh para compilar/instalar dependências CUDA..."
#if [ -f "/app/builder.sh" ]; then
#  /bin/bash /app/builder.sh
#  echo "✅ Builder finalizado."
#else
#  echo "⚠️  Aviso: builder.sh não encontrado. Pulando etapa de compilação de dependências."
#fi


python3 - <<'PY'
import os
from services.ltx_server import ltx_server_singleton as server
try:
  server.setup_dependencies()
except Exception as e:
  print("Erro no setup")
PY


export OUTPUT_ROOT="${OUTPUT_ROOT:-/app/outputs}"
export INPUT_ROOT="${INPUT_ROOT:-/app/inputs}"

mkdir -p "$OUTPUT_ROOT" "$INPUT_ROOT"
echo "[ltx][start] Verificando ambiente como usuário: $(whoami)"

# Env da UI
export GRADIO_SERVER_NAME="0.0.0.0"
export GRADIO_SERVER_PORT="${PORT:-7860}"
export GRADIO_ENABLE_QUEUE="True"

tree -L 4 /app
tree -L 4 /data


echo "[ltx][start] Lançando app_ltx.py..."
# Executa diretamente o python.
exec python /app/app_ltx.py