#!/bin/bash
set -e

echo "=== Setting up InfiniteTalk ==="

echo "Repository structure:"
ls -la InfiniteTalk/

# Add to Python path
export PYTHONPATH="/home/user/app/InfiniteTalk:$PYTHONPATH"

# Install requirements using pip3
echo "Installing Python requirements..."
pip3 install -r requirements.txt

# Test imports
echo "Testing imports..."
python3 -c "
import sys
sys.path.append('./InfiniteTalk')
try:
    from wan.configs import SIZE_CONFIGS, SUPPORTED_SIZES, WAN_CONFIGS
    print('✓ SUCCESS: All wan imports work!')
except ImportError as e:
    print(f'✗ FAILED: {e}')
"

echo "=== Setup completed ==="