#!/bin/bash

# Streamlit UI 启动脚本
echo "🚀 启动沙发智能咨询助手 - Streamlit UI"
echo "=================================="

# 检查是否在正确的目录
if [ ! -f "chat_ui.py" ]; then
    echo "❌ 错误: 未找到 chat_ui.py 文件"
    echo "请确保在项目根目录下运行此脚本"
    exit 1
fi

# 检查虚拟环境是否存在
if [ ! -d ".venv" ]; then
    echo "⚠️  警告: 未找到 .venv 虚拟环境目录"
    echo "正在创建虚拟环境..."
    python3 -m venv .venv
fi

# 激活虚拟环境
echo "🔧 激活虚拟环境..."
source .venv/bin/activate

# 检查是否成功激活虚拟环境
if [ "$VIRTUAL_ENV" != "" ]; then
    echo "✅ 虚拟环境已激活: $VIRTUAL_ENV"
else
    echo "❌ 虚拟环境激活失败"
    exit 1
fi

# 安装/更新依赖
echo "📦 检查依赖包..."
if [ -f "requirements.txt" ]; then
    pip install -r requirements.txt --quiet
elif [ -f "pyproject.toml" ]; then
    pip install -e . --quiet
fi

# 检查streamlit是否已安装
if ! python -c "import streamlit" 2>/dev/null; then
    echo "📦 安装 Streamlit..."
    pip install streamlit
fi

# 启动Streamlit应用
echo "🌐 启动 Streamlit 应用..."
echo "浏览器将自动打开 http://localhost:8501"
echo "按 Ctrl+C 停止应用"
echo "=================================="

streamlit run chat_ui.py
