#!/bin/bash

# Fix ChromaDB Installation Issue
# Run this on production Linux server: /workspace/Foundations

set -e

echo "=========================================="
echo "Fixing ChromaDB Installation"
echo "=========================================="

# Navigate to FoundationsAI directory
cd /workspace/Foundations

# Activate virtual environment
echo "Activating virtual environment..."
source venv/bin/activate

# Uninstall corrupted chromadb packages
echo "Removing corrupted chromadb installation..."
pip uninstall -y chromadb chromadb-client 2>/dev/null || true

# Clear pip cache for chromadb
echo "Clearing pip cache..."
pip cache remove chromadb 2>/dev/null || true

# Reinstall chromadb with specific compatible version
echo "Installing chromadb 0.4.22..."
pip install chromadb==0.4.22

# Verify installation
echo "Verifying installation..."
python3 -c "import chromadb; print(f'ChromaDB version: {chromadb.__version__}')"

echo ""
echo "=========================================="
echo "✓ ChromaDB fixed successfully!"
echo "=========================================="
echo ""
echo "Now restart FoundationsAI:"
echo "  cd /workspace/Foundations"
echo "  source venv/bin/activate"
echo "  python3 LLMAPI.py"