#!/bin/bash

# Script to migrate from Google Gemini to OpenAI GPT-4o

echo "Migrating from Google Gemini to OpenAI GPT-4o..."

# 1. Activate the virtual environment
source venv/bin/activate

# 2. Install OpenAI package
echo "Installing OpenAI package..."
pip install openai==1.30.4

# 3. Prompt for OpenAI API key
read -p "Enter your OpenAI API key: " openai_api_key

# 4. Update the .env file
echo "Updating .env file..."
if grep -q "OPENAI_API_KEY" .env; then
  # Replace existing OPENAI_API_KEY
  sed -i "s/OPENAI_API_KEY=.*/OPENAI_API_KEY=$openai_api_key/" .env
else
  # Add new OPENAI_API_KEY entry
  sed -i "/GEMINI_API_KEY/i # OpenAI GPT-4o (new)\nOPENAI_API_KEY=$openai_api_key\n" .env
fi

# 5. Reset and rebuild the ChromaDB
echo "Do you want to reset and rebuild the ChromaDB? (y/n)"
read -p "> " rebuild_db

if [[ $rebuild_db == "y" || $rebuild_db == "Y" ]]; then
  echo "Resetting ChromaDB..."
  ./reset_chromadb.py
  
  echo "Rebuilding database (syncing past 7 days of messages)..."
  python sync_messages.py --force --days 7
fi

echo "Migration completed successfully!"
echo "Please restart your application to apply the changes:"
echo "  1. Stop the current process"
echo "  2. Run ./run_app.sh to start with OpenAI integration"