#!/bin/bash
# Transform import statements from seed.engine to warbler_cda

cd warbler-cda-package/warbler_cda || exit 1

# Transform all Python files
for file in *.py embeddings/*.py api/*.py utils/*.py; do
    if [ -f "$file" ]; then
        echo "Transforming $file..."
        
        # Transform absolute imports
        sed -i 's/from seed\.engine\./from warbler_cda./g' "$file"
        sed -i 's/import seed\.engine\./import warbler_cda./g' "$file"
        
        # Transform relative imports to absolute
        sed -i 's/from \.embeddings import/from warbler_cda.embeddings import/g' "$file"
        sed -i 's/from \.embeddings\./from warbler_cda.embeddings./g' "$file"
        sed -i 's/from \.anchor_data_classes/from warbler_cda.anchor_data_classes/g' "$file"
        sed -i 's/from \.anchor_memory_pool/from warbler_cda.anchor_memory_pool/g' "$file"
        sed -i 's/from \.base_provider/from warbler_cda.embeddings.base_provider/g' "$file"
        sed -i 's/from \.local_provider/from warbler_cda.embeddings.local_provider/g' "$file"
        sed -i 's/from \.openai_provider/from warbler_cda.embeddings.openai_provider/g' "$file"
        sed -i 's/from \.factory/from warbler_cda.embeddings.factory/g' "$file"
        
        # Transform specific imports in API files
        sed -i 's/from retrieval_api import/from warbler_cda.retrieval_api import/g' "$file"
        sed -i 's/from stat7_rag_bridge import/from warbler_cda.stat7_rag_bridge import/g' "$file"
        sed -i 's/from stat7_experiments import/from warbler_cda.stat7_experiments import/g' "$file"
        
        # Remove privacy hooks imports (not included in package)
        sed -i '/from \.hooks\.privacy_hooks/d' "$file"
        sed -i '/from seed\.engine\.hooks/d' "$file"
        
        echo "  ✓ Transformed $file"
    fi
done

echo ""
echo "✓ All import transformations complete!"