install.sh

#!/bin/bash
set -e

echo "🚀 Setting up Multimodal AI Environment"
echo "--------------------------------------"

# Create virtual environment
python -m venv .venv
source .venv/bin/activate

# Install core requirements
pip install --upgrade pip
pip install -r requirements.txt

# Verify critical installations
echo "✅ Installation Complete"
echo "Installed Packages:"
pip list | grep -E "torch|transformers|diffusers|gradio"

# Space-specific optimizations
if [ "$SPACES" = "1" ]; then
    echo "🔧 Applying Space Optimizations"
    mkdir -p model_cache
    mkdir -p tmp
    chmod -R 777 model_cache
    chmod -R 777 tmp
fi

echo "🌐 Ready to launch!"