#!/bin/bash

# Z-Image-Turbo API Wrapper - Quick Start Script
# This script sets up and runs the API wrapper

set -e

echo "========================================"
echo "Z-Image-Turbo API Wrapper - Quick Setup"
echo "========================================"
echo ""

# Check Python version
if ! command -v python3 &> /dev/null; then
    echo "❌ Python 3 is not installed"
    echo "Install it with: sudo apt-get install python3 python3-pip"
    exit 1
fi

PYTHON_VERSION=$(python3 --version | cut -d' ' -f2)
echo "✓ Python $PYTHON_VERSION found"

# Create virtual environment
echo ""
echo "Setting up virtual environment..."
if [ ! -d "venv" ]; then
    python3 -m venv venv
    echo "✓ Virtual environment created"
else
    echo "✓ Virtual environment already exists"
fi

# Activate virtual environment
source venv/bin/activate
echo "✓ Virtual environment activated"

# Install dependencies
echo ""
echo "Installing dependencies..."
pip install -q --upgrade pip
pip install -q -r requirements.txt
echo "✓ Dependencies installed"

# Check if gunicorn is needed for production
echo ""
echo "========================================"
echo "Setup Complete!"
echo "========================================"
echo ""
echo "To start the development server:"
echo "  source venv/bin/activate"
echo "  python app.py"
echo ""
echo "To start the production server:"
echo "  source venv/bin/activate"
echo "  gunicorn -w 4 -b 0.0.0.0:5000 --timeout 300 app:app"
echo ""
echo "To test the API:"
echo "  curl http://localhost:5000/health"
echo "  curl -X POST http://localhost:5000/api/generate -H 'Content-Type: application/json' -d '{\"prompt\": \"A sunset\"}'"
echo ""