update.sh

#!/bin/bash

# Update script for AI Agent & TTS API
# This script pulls the latest changes and restarts the service safely.

set -e

# Search for the application directory in common locations
if [ -d "$HOME/aiagent/.git" ]; then
    APP_DIR="$HOME/aiagent"
elif [ -d "/root/aiagent/.git" ]; then
    APP_DIR="/root/aiagent"
elif [ -d "/opt/aiagent/.git" ]; then
    APP_DIR="/opt/aiagent"
else
    # Try to find it relative to scripts if possible, or fail gracefully
    SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd 2>/dev/null || pwd)"
    if [ -d "$SCRIPT_DIR/.git" ]; then
        APP_DIR="$SCRIPT_DIR"
    else
        echo "❌ Lỗi: Không tìm thấy thư mục cài đặt AI Agent (có chứa .git)."
        echo "Hãy chắc chắn bạn đã cài đặt vào ~/aiagent hoặc /root/aiagent."
        exit 1
    fi
fi

cd "$APP_DIR"
echo "🔄 Đang cập nhật AI Agent tại: $APP_DIR"

# 1. Fetch and Reset to avoid local conflicts
git fetch origin
git reset --hard origin/main

# 2. Re-apply permissions
chmod +x *.sh

# 3. Update dependencies in venv if it exists
if [ -d "venv" ]; then
    echo "📦 Đang cập nhật thư viện Python (venv)..."
    ./venv/bin/python3 -m pip install --upgrade pip
    ./venv/bin/python3 -m pip install --ignore-installed -r requirements.txt
else
    echo "📦 Đang cập nhật thư viện Python (System)..."
    pip3 install --break-system-packages -r requirements.txt || pip3 install -r requirements.txt
fi

# 4. Restart the service
echo "⚙️ Đang khởi động lại dịch vụ..."
if [[ "$(uname)" == "Darwin" ]]; then
    # macOS
    ./install_service.sh
    ./manage.sh restart
elif [[ "$(uname)" == "Linux" ]]; then
    # Linux (Systemd)
    sudo systemctl restart aiagent || ./manage.sh restart
fi

echo ""
echo "✅ Cập nhật hoàn tất!"
echo "------------------------------------------------"
echo "🌐 Trạng thái: $(./manage.sh status)"
echo "------------------------------------------------"