#!/bin/bash
# Script de téléchargement ACE-Step 1.5 Turbo
# Usage: bash download-ace-step.sh

set -e

MODEL_DIR="/home/folzo/Documents/OpenCode/Projets/Effi/comfyui/models/checkpoints/ace-step"

echo "🎵 ACE-Step 1.5 Turbo - Téléchargement"
echo "================================"
echo ""

# Créer le dossier si nécessaire
mkdir -p "$MODEL_DIR"

# Vérifier si wget est disponible
if ! command -v wget &> /dev/null; then
    echo "❌ wget non trouvé. Installation..."
    sudo apt update && sudo apt install wget
fi

cd "$MODEL_DIR"

# Vérifier si le modèle existe déjà
if [ -f "acestep-v15-turbo.safetensors" ]; then
    echo "✅ Le modèle existe déjà :"
    ls -lh acestep-v15-turbo.safetensors
    echo ""
    echo "Pour re-télécharger, supprimez d'abord le fichier :"
    echo "rm $MODEL_DIR/acestep-v15-turbo.safetensors"
    exit 0
fi

echo "📥 Téléchargement du modèle ACE-Step 1.5 Turbo..."
echo "📦 Destination: $MODEL_DIR"
echo ""
echo "⏳ Le fichier fait environ 3-4GB, soyez patient..."
echo ""

# Télécharger le modèle
wget --progress=bar:force \
  "https://huggingface.co/ACE-Step/Ace-Step1.5/resolve/main/acestep-v15-turbo/model.safetensors" \
  -O acestep-v15-turbo.safetensors

echo ""
echo "✅ Téléchargement terminé !"
echo ""
echo "📁 Fichier téléchargé :"
ls -lh acestep-v15-turbo.safetensors

echo ""
echo "================================"
echo "🎉 Prochaine étape :"
echo "1. Redémarrer ComfyUI: docker restart comfyui"
echo "2. Configurer le tool dans OpenWebUI"
echo "3. Tester avec un prompt musical !"
echo "================================"