#!/bin/bash
set -e

echo "Starting DF-GAN Bird Image Generator setup..."

# Install NLTK data
echo "Setting up NLTK data..."
python nltk_setup.py

# Run the download_models.py script to get the models
echo "Downloading model files..."
python download_models.py || {
    echo "Warning: Some model files may not have downloaded correctly."
    echo "The application will attempt to continue with fallback models."
}

# Start the Gradio app
echo "Starting the web application..."
exec python app.py