#!/bin/bash
# This script runs during the build phase of the HuggingFace Space

echo "---------------------------------------------------"
echo "Setting up Acoustic Intelligence Suite environment"
echo "---------------------------------------------------"

# Create persistent directories for cached models
mkdir -p /data/cache/huggingface
mkdir -p /data/cache/transformers
mkdir -p /data/cache/nltk_data
mkdir -p /data/cache/torch

# Set environment variables to point to persistent storage
export TRANSFORMERS_CACHE="/data/cache/transformers"
export HF_HOME="/data/cache/huggingface" 
export NLTK_DATA="/data/cache/nltk_data"
export TORCH_HOME="/data/cache/torch"

# Echo the environment variables for logging
echo "Environment variables set:"
echo "- TRANSFORMERS_CACHE: $TRANSFORMERS_CACHE"
echo "- HF_HOME: $HF_HOME"
echo "- NLTK_DATA: $NLTK_DATA"
echo "- TORCH_HOME: $TORCH_HOME"

echo "Running model cache builder script..."
# Use the dedicated Python script for better error handling and structure
python build_model_cache.py

# Install any additional dependencies that might be missing
echo "Checking for additional missing dependencies..."
pip install --no-cache-dir --upgrade google-generativeai moviepy

echo "Setup complete!"
echo "---------------------------------------------------"