#!/bin/bash
# 安装多模态检索系统依赖包

echo "🚀 开始安装多模态检索系统依赖包..."

# 更新pip
pip install --upgrade pip

# 安装基础依赖
echo "📦 安装基础依赖包..."
pip install torch>=2.0.0 torchvision>=0.15.0
pip install transformers>=4.30.0 accelerate>=0.20.0
pip install numpy>=1.21.0 Pillow>=9.0.0
pip install scikit-learn>=1.3.0 tqdm>=4.65.0
pip install flask>=2.3.0 werkzeug>=2.3.0
pip install psutil>=5.9.0

# 安装百度VDB SDK
echo "🔗 安装百度VDB SDK..."
pip install pymochow

# 安装MongoDB驱动
echo "💾 安装MongoDB驱动..."
pip install pymongo>=4.0.0

# 安装FAISS (备用)
echo "🔍 安装FAISS..."
pip install faiss-cpu>=1.7.4

echo "✅ 依赖包安装完成！"
echo "📋 已安装的主要包："
echo "  - torch (深度学习框架)"
echo "  - transformers (模型库)"
echo "  - pymochow (百度VDB SDK)"
echo "  - flask (Web框架)"
echo "  - pymongo (MongoDB驱动)"
echo ""
echo "🎯 接下来可以运行测试脚本验证安装"