# Simple Model Download Script for Windows
# This script will help you download OmniAvatar models even if Python isn't in PATH

Write-Host "🎭 OmniAvatar Model Download Assistant" -ForegroundColor Green
Write-Host "=====================================" -ForegroundColor Green
Write-Host ""

Write-Host "❌ Current Status: No video models found" -ForegroundColor Red
Write-Host "🎯 Result: App runs in TTS-only mode (audio output only)" -ForegroundColor Yellow
Write-Host ""
Write-Host "To enable video generation, you need to download ~30GB of models:" -ForegroundColor Cyan
Write-Host "  📦 Wan2.1-T2V-14B (~28GB) - Base text-to-video model" -ForegroundColor White
Write-Host "  📦 OmniAvatar-14B (~2GB) - Avatar animation weights" -ForegroundColor White
Write-Host "  📦 wav2vec2-base-960h (~360MB) - Audio encoder" -ForegroundColor White
Write-Host ""

Write-Host "🚀 Download Options:" -ForegroundColor Green
Write-Host ""
Write-Host "1. 🐍 Using Python (Recommended)" -ForegroundColor Yellow
Write-Host "   - Open Command Prompt or PowerShell as Administrator" -ForegroundColor Gray
Write-Host "   - Navigate to this directory" -ForegroundColor Gray
Write-Host "   - Run: python setup_omniavatar.py" -ForegroundColor Gray
Write-Host ""

Write-Host "2. 🌐 Manual Download" -ForegroundColor Yellow
Write-Host "   - Visit: https://huggingface.co/OmniAvatar/OmniAvatar-14B" -ForegroundColor Gray
Write-Host "   - Click 'Files and versions' tab" -ForegroundColor Gray
Write-Host "   - Download all files to: pretrained_models/OmniAvatar-14B/" -ForegroundColor Gray
Write-Host "   - Repeat for other models (see MODEL_DOWNLOAD_GUIDE.md)" -ForegroundColor Gray
Write-Host ""

Write-Host "3. 🔧 Git LFS (If available)" -ForegroundColor Yellow
Write-Host "   git lfs clone https://huggingface.co/OmniAvatar/OmniAvatar-14B pretrained_models/OmniAvatar-14B" -ForegroundColor Gray
Write-Host ""

Write-Host "📋 After downloading models:" -ForegroundColor Cyan
Write-Host "  ✅ Restart your app: python app.py" -ForegroundColor White
Write-Host "  ✅ Check logs for 'full functionality enabled'" -ForegroundColor White
Write-Host "  ✅ API will return video URLs instead of audio-only" -ForegroundColor White
Write-Host ""

# Check if any Python executable might exist in common locations
$commonPythonPaths = @(
    "C:\Python*\python.exe",
    "C:\Users\$env:USERNAME\AppData\Local\Programs\Python\Python*\python.exe",
    "C:\Program Files\Python*\python.exe"
)

Write-Host "🔍 Scanning for Python installations..." -ForegroundColor Yellow
$foundPython = $false

foreach ($pattern in $commonPythonPaths) {
    $pythonExes = Get-ChildItem -Path $pattern -ErrorAction SilentlyContinue
    foreach ($exe in $pythonExes) {
        Write-Host "   Found: $($exe.FullName)" -ForegroundColor Green
        $foundPython = $true
    }
}

if ($foundPython) {
    Write-Host ""
    Write-Host "💡 Try running the setup script with full path to Python:" -ForegroundColor Cyan
    Write-Host "   C:\Path\To\Python\python.exe setup_omniavatar.py" -ForegroundColor Gray
} else {
    Write-Host "   No Python installations found in common locations" -ForegroundColor Gray
}

Write-Host ""
Write-Host "📖 For detailed instructions, see: MODEL_DOWNLOAD_GUIDE.md" -ForegroundColor Cyan