Upload 12 files

.env.example

@@ -5,10 +5,6 @@
 # 管理员密钥（必需，用于登录管理面板）
 # 明文示例：
 ADMIN_KEY=your-admin-secret-key
-# Hash 示例（SHA256）：ADMIN_KEY=sha256:你的32字节hex
-
-# API密钥（可选，用于API端点认证，优先级：环境变量 > settings.yaml）
-# API_KEY=your-api-key
 
 # 服务端口（可选，默认 7860）
 # PORT=7860
@@ -34,24 +30,7 @@ ADMIN_KEY=your-admin-secret-key
 # ============================================
 # 账户配置
 # ============================================
-# 方式1：使用 accounts.json 文件（推荐）
+# 使用 accounts.json 文件
 # 账户配置保存在 accounts.json 文件中
 # 首次启动时会自动创建空配置
-# 请在管理面板中添加账户，或直接编辑 accounts.json
-#
-# 方式2：使用环境变量（适用于容器化部署，如抱脸等平台）
-# 设置 ACCOUNTS_CONFIG 环境变量，值为 JSON 数组字符串
-# 优先级：环境变量 > 文件
-#
-# ACCOUNTS_CONFIG='[{"id":"account_1","secure_c_ses":"your-cookie","csesidx":"your-idx","config_id":"your-config","expires_at":"2025-12-31 23:59:59"}]'
-#
-# 账户配置格式示例：
-# [
-#   {
-#     "id": "account_1",
-#     "secure_c_ses": "your-cookie-here",
-#     "csesidx": "your-idx",
-#     "config_id": "your-config",
-#     "expires_at": "2025-12-31 23:59:59"
-#   }
-# ]
+# 请在管理面板中添加账户，或直接编辑 accounts.json

.gitignore

@@ -39,7 +39,9 @@ ENV/
 # Generated files
 data/
 logs/
+static/
 
 # OS
 .DS_Store
 Thumbs.db
+old_version.py

deploy.bat

@@ -0,0 +1,121 @@
+@echo off
+REM Gemini Business2API Deployment Script for Windows
+REM This script automates the initial deployment process
+
+setlocal enabledelayedexpansion
+
+echo ==========================================
+echo Gemini Business2API Deployment Script
+echo ==========================================
+echo.
+
+REM Check if git is installed
+where git >nul 2>nul
+if %errorlevel% neq 0 (
+    echo [ERROR] Git is not installed. Please install git first.
+    exit /b 1
+)
+
+REM Check if python is installed
+where python >nul 2>nul
+if %errorlevel% neq 0 (
+    echo [ERROR] Python is not installed. Please install Python 3.11+ first.
+    exit /b 1
+)
+
+REM Check if npm is installed
+where npm >nul 2>nul
+if %errorlevel% neq 0 (
+    echo [ERROR] npm is not installed. Please install Node.js and npm first.
+    exit /b 1
+)
+
+REM Step 1: Build frontend
+echo [STEP] Step 1: Building frontend...
+if exist "frontend" (
+    cd frontend
+    echo [INFO] Installing frontend dependencies...
+    call npm install
+    if %errorlevel% neq 0 (
+        echo [ERROR] Failed to install frontend dependencies
+        exit /b 1
+    )
+
+    echo [INFO] Building frontend...
+    call npm run build
+    if %errorlevel% neq 0 (
+        echo [ERROR] Failed to build frontend
+        exit /b 1
+    )
+
+    echo [SUCCESS] Frontend built successfully
+    cd ..
+) else (
+    echo [ERROR] Frontend directory not found. Are you in the project root?
+    exit /b 1
+)
+
+REM Step 2: Create virtual environment
+echo [STEP] Step 2: Setting up Python virtual environment...
+if exist ".venv" (
+    echo [INFO] Virtual environment already exists, skipping creation
+) else (
+    echo [INFO] Creating virtual environment...
+    python -m venv .venv
+    if %errorlevel% neq 0 (
+        echo [ERROR] Failed to create virtual environment
+        exit /b 1
+    )
+    echo [SUCCESS] Virtual environment created
+)
+
+REM Activate virtual environment
+echo [INFO] Activating virtual environment...
+call .venv\Scripts\activate.bat
+
+REM Step 3: Install Python dependencies
+echo [STEP] Step 3: Installing Python dependencies...
+python -m pip install --upgrade pip
+python -m pip install -r requirements.txt
+if %errorlevel% neq 0 (
+    echo [ERROR] Failed to install Python dependencies
+    exit /b 1
+)
+echo [SUCCESS] Python dependencies installed
+
+REM Step 4: Setup .env file
+echo [STEP] Step 4: Setting up configuration...
+if exist ".env" (
+    echo [INFO] .env file already exists, skipping
+) else (
+    if exist ".env.example" (
+        copy /Y ".env.example" ".env" >nul
+        echo [SUCCESS] .env file created from .env.example
+    ) else (
+        echo [ERROR] .env.example not found
+        exit /b 1
+    )
+)
+
+REM Step 5: Show completion message
+echo.
+echo ==========================================
+echo [SUCCESS] Deployment completed successfully!
+echo ==========================================
+echo.
+echo [INFO] Next steps:
+echo.
+echo   1. Edit .env file and set your ADMIN_KEY:
+echo      notepad .env
+echo.
+echo   2. Start the service:
+echo      python main.py
+echo.
+echo   3. Access the admin panel:
+echo      http://localhost:7860/
+echo.
+echo [INFO] Optional: To activate virtual environment later, run:
+echo   .venv\Scripts\activate.bat
+echo.
+
+endlocal

deploy.sh

@@ -0,0 +1,123 @@
+#!/bin/bash
+
+# Gemini Business2API Deployment Script for Linux/macOS
+# This script automates the initial deployment process
+
+set -e  # Exit on error
+
+echo "=========================================="
+echo "Gemini Business2API Deployment Script"
+echo "=========================================="
+echo ""
+
+# Color codes for output
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+YELLOW='\033[1;33m'
+BLUE='\033[0;34m'
+NC='\033[0m' # No Color
+
+# Function to print colored messages
+print_success() {
+    echo -e "${GREEN}[SUCCESS] $1${NC}"
+}
+
+print_error() {
+    echo -e "${RED}[ERROR] $1${NC}"
+}
+
+print_info() {
+    echo -e "${YELLOW}[INFO] $1${NC}"
+}
+
+print_step() {
+    echo -e "${BLUE}[STEP] $1${NC}"
+}
+
+# Check if git is installed
+if ! command -v git &> /dev/null; then
+    print_error "Git is not installed. Please install git first."
+    exit 1
+fi
+
+# Check if python3 is installed
+if ! command -v python3 &> /dev/null; then
+    print_error "Python3 is not installed. Please install Python 3.11+ first."
+    exit 1
+fi
+
+# Check if npm is installed
+if ! command -v npm &> /dev/null; then
+    print_error "npm is not installed. Please install Node.js and npm first."
+    exit 1
+fi
+
+# Step 1: Build frontend
+print_step "Step 1: Building frontend..."
+if [ -d "frontend" ]; then
+    cd frontend
+    print_info "Installing frontend dependencies..."
+    npm install
+    print_info "Building frontend..."
+    npm run build
+    print_success "Frontend built successfully"
+    cd ..
+else
+    print_error "Frontend directory not found. Are you in the project root?"
+    exit 1
+fi
+
+# Step 2: Create virtual environment
+print_step "Step 2: Setting up Python virtual environment..."
+if [ -d ".venv" ]; then
+    print_info "Virtual environment already exists, skipping creation"
+else
+    print_info "Creating virtual environment..."
+    python3 -m venv .venv
+    print_success "Virtual environment created"
+fi
+
+# Activate virtual environment
+print_info "Activating virtual environment..."
+source .venv/bin/activate
+
+# Step 3: Install Python dependencies
+print_step "Step 3: Installing Python dependencies..."
+pip install --upgrade pip
+pip install -r requirements.txt
+print_success "Python dependencies installed"
+
+# Step 4: Setup .env file
+print_step "Step 4: Setting up configuration..."
+if [ -f ".env" ]; then
+    print_info ".env file already exists, skipping"
+else
+    if [ -f ".env.example" ]; then
+        cp .env.example .env
+        print_success ".env file created from .env.example"
+    else
+        print_error ".env.example not found"
+        exit 1
+    fi
+fi
+
+# Step 5: Show completion message
+echo ""
+echo "=========================================="
+print_success "Deployment completed successfully!"
+echo "=========================================="
+echo ""
+print_info "Next steps:"
+echo ""
+echo "  1. Edit .env file and set your ADMIN_KEY:"
+echo "     ${BLUE}nano .env${NC}  or  ${BLUE}vim .env${NC}"
+echo ""
+echo "  2. Start the service:"
+echo "     ${BLUE}python main.py${NC}"
+echo ""
+echo "  3. Access the admin panel:"
+echo "     ${BLUE}http://localhost:7860/${NC}"
+echo ""
+print_info "Optional: To activate virtual environment later, run:"
+echo "  ${BLUE}source .venv/bin/activate${NC}"
+echo ""

main.py

@@ -321,6 +321,49 @@ multi_account_mgr = load_multi_account_config(
     global_stats
 )
 
+# ---------- 自动注册/刷新服务 ----------
+register_service = None
+login_service = None
+
+def _set_multi_account_mgr(new_mgr):
+    global multi_account_mgr
+    multi_account_mgr = new_mgr
+    if register_service:
+        register_service.multi_account_mgr = new_mgr
+    if login_service:
+        login_service.multi_account_mgr = new_mgr
+
+def _get_global_stats():
+    return global_stats
+
+try:
+    from core.register_service import RegisterService
+    from core.login_service import LoginService
+    register_service = RegisterService(
+        multi_account_mgr,
+        http_client,
+        USER_AGENT,
+        ACCOUNT_FAILURE_THRESHOLD,
+        RATE_LIMIT_COOLDOWN_SECONDS,
+        SESSION_CACHE_TTL_SECONDS,
+        _get_global_stats,
+        _set_multi_account_mgr,
+    )
+    login_service = LoginService(
+        multi_account_mgr,
+        http_client,
+        USER_AGENT,
+        ACCOUNT_FAILURE_THRESHOLD,
+        RATE_LIMIT_COOLDOWN_SECONDS,
+        SESSION_CACHE_TTL_SECONDS,
+        _get_global_stats,
+        _set_multi_account_mgr,
+    )
+except Exception as e:
+    logger.warning("[SYSTEM] 自动注册/刷新服务不可用: %s", e)
+    register_service = None
+    login_service = None
+
 # 验证必需的环境变量
 if not ADMIN_KEY:
     logger.error("[SYSTEM] 未配置 ADMIN_KEY 环境变量，请设置后重启")
@@ -534,6 +577,16 @@ async def startup_event():
     elif storage.is_database_enabled():
         logger.info("[SYSTEM] 自动刷新账号功能已禁用（配置为0）")
 
+    # 启动自动登录刷新轮询
+    if login_service:
+        try:
+            asyncio.create_task(login_service.start_polling())
+            logger.info("[SYSTEM] 账户过期检查轮询已启动（间隔: 30分钟）")
+        except Exception as e:
+            logger.error(f"[SYSTEM] 启动登录服务失败: {e}")
+    else:
+        logger.info("[SYSTEM] 自动登录刷新未启用或依赖不可用")
+
 # ---------- 日志脱敏函数 ----------
 def get_sanitized_logs(limit: int = 100) -> list:
     """获取脱敏后的日志列表，按请求ID分组并提取关键事件"""
@@ -943,6 +996,70 @@ async def admin_update_config(request: Request, accounts_data: list = Body(...))
         logger.error(f"[CONFIG] 更新配置失败: {str(e)}")
         raise HTTPException(500, f"更新失败: {str(e)}")
 
+@app.post("/admin/register/start")
+@require_login()
+async def admin_start_register(request: Request, count: Optional[int] = Body(default=None), domain: Optional[str] = Body(default=None)):
+    if not register_service:
+        raise HTTPException(503, "register service unavailable")
+    task = await register_service.start_register(count=count, domain=domain)
+    return task.to_dict()
+
+@app.get("/admin/register/task/{task_id}")
+@require_login()
+async def admin_get_register_task(request: Request, task_id: str):
+    if not register_service:
+        raise HTTPException(503, "register service unavailable")
+    task = register_service.get_task(task_id)
+    if not task:
+        raise HTTPException(404, "task not found")
+    return task.to_dict()
+
+@app.get("/admin/register/current")
+@require_login()
+async def admin_get_current_register_task(request: Request):
+    if not register_service:
+        raise HTTPException(503, "register service unavailable")
+    task = register_service.get_current_task()
+    if not task:
+        return {"status": "idle"}
+    return task.to_dict()
+
+@app.post("/admin/login/start")
+@require_login()
+async def admin_start_login(request: Request, account_ids: List[str] = Body(...)):
+    if not login_service:
+        raise HTTPException(503, "login service unavailable")
+    task = await login_service.start_login(account_ids)
+    return task.to_dict()
+
+@app.get("/admin/login/task/{task_id}")
+@require_login()
+async def admin_get_login_task(request: Request, task_id: str):
+    if not login_service:
+        raise HTTPException(503, "login service unavailable")
+    task = login_service.get_task(task_id)
+    if not task:
+        raise HTTPException(404, "task not found")
+    return task.to_dict()
+
+@app.get("/admin/login/current")
+@require_login()
+async def admin_get_current_login_task(request: Request):
+    if not login_service:
+        raise HTTPException(503, "login service unavailable")
+    task = login_service.get_current_task()
+    if not task:
+        return {"status": "idle"}
+    return task.to_dict()
+
+@app.post("/admin/login/check")
+@require_login()
+async def admin_check_login_refresh(request: Request):
+    if not login_service:
+        raise HTTPException(503, "login service unavailable")
+    await login_service.check_and_refresh()
+    return {"status": "ok"}
+
 @app.delete("/admin/accounts/{account_id}")
 @require_login()
 async def admin_delete_account(request: Request, account_id: str):
@@ -1010,7 +1127,14 @@ async def admin_get_settings(request: Request):
         "basic": {
             "api_key": config.basic.api_key,
             "base_url": config.basic.base_url,
-            "proxy": config.basic.proxy
+            "proxy": config.basic.proxy,
+            "duckmail_base_url": config.basic.duckmail_base_url,
+            "duckmail_api_key": config.basic.duckmail_api_key,
+            "duckmail_verify_ssl": config.basic.duckmail_verify_ssl,
+            "browser_headless": config.basic.browser_headless,
+            "refresh_window_hours": config.basic.refresh_window_hours,
+            "register_default_count": config.basic.register_default_count,
+            "register_domain": config.basic.register_domain,
         },
         "image_generation": {
             "enabled": config.image_generation.enabled,
@@ -1023,7 +1147,8 @@ async def admin_get_settings(request: Request):
             "max_account_switch_tries": config.retry.max_account_switch_tries,
             "account_failure_threshold": config.retry.account_failure_threshold,
             "rate_limit_cooldown_seconds": config.retry.rate_limit_cooldown_seconds,
-            "session_cache_ttl_seconds": config.retry.session_cache_ttl_seconds
+            "session_cache_ttl_seconds": config.retry.session_cache_ttl_seconds,
+            "auto_refresh_accounts_seconds": config.retry.auto_refresh_accounts_seconds
         },
         "public_display": {
             "logo_url": config.public_display.logo_url,
@@ -1041,10 +1166,23 @@ async def admin_update_settings(request: Request, new_settings: dict = Body(...)
     global API_KEY, PROXY, BASE_URL, LOGO_URL, CHAT_URL
     global IMAGE_GENERATION_ENABLED, IMAGE_GENERATION_MODELS
     global MAX_NEW_SESSION_TRIES, MAX_REQUEST_RETRIES, MAX_ACCOUNT_SWITCH_TRIES
-    global ACCOUNT_FAILURE_THRESHOLD, RATE_LIMIT_COOLDOWN_SECONDS, SESSION_CACHE_TTL_SECONDS
+    global ACCOUNT_FAILURE_THRESHOLD, RATE_LIMIT_COOLDOWN_SECONDS, SESSION_CACHE_TTL_SECONDS, AUTO_REFRESH_ACCOUNTS_SECONDS
     global SESSION_EXPIRE_HOURS, multi_account_mgr, http_client
 
     try:
+        basic = dict(new_settings.get("basic") or {})
+        basic.setdefault("duckmail_base_url", config.basic.duckmail_base_url)
+        basic.setdefault("duckmail_api_key", config.basic.duckmail_api_key)
+        basic.setdefault("duckmail_verify_ssl", config.basic.duckmail_verify_ssl)
+        basic.setdefault("browser_headless", config.basic.browser_headless)
+        basic.setdefault("refresh_window_hours", config.basic.refresh_window_hours)
+        basic.setdefault("register_default_count", config.basic.register_default_count)
+        basic.setdefault("register_domain", config.basic.register_domain)
+        if not isinstance(basic.get("register_domain"), str):
+            basic["register_domain"] = ""
+        basic.pop("duckmail_proxy", None)
+        new_settings["basic"] = basic
+
         image_generation = dict(new_settings.get("image_generation") or {})
         output_format = str(image_generation.get("output_format") or config_manager.image_output_format).lower()
         if output_format not in ("base64", "url"):
@@ -1052,6 +1190,10 @@ async def admin_update_settings(request: Request, new_settings: dict = Body(...)
         image_generation["output_format"] = output_format
         new_settings["image_generation"] = image_generation
 
+        retry = dict(new_settings.get("retry") or {})
+        retry.setdefault("auto_refresh_accounts_seconds", config.retry.auto_refresh_accounts_seconds)
+        new_settings["retry"] = retry
+
         # 保存旧配置用于对比
         old_proxy = PROXY
         old_retry_config = {
@@ -1080,6 +1222,7 @@ async def admin_update_settings(request: Request, new_settings: dict = Body(...)
         ACCOUNT_FAILURE_THRESHOLD = config.retry.account_failure_threshold
         RATE_LIMIT_COOLDOWN_SECONDS = config.retry.rate_limit_cooldown_seconds
         SESSION_CACHE_TTL_SECONDS = config.retry.session_cache_ttl_seconds
+        AUTO_REFRESH_ACCOUNTS_SECONDS = config.retry.auto_refresh_accounts_seconds
         SESSION_EXPIRE_HOURS = config.session.expire_hours
 
         # 检查是否需要重建 HTTP 客户端（代理变化）

requirements.txt

@@ -1,14 +1,18 @@
-fastapi==0.110.0
-uvicorn[standard]==0.29.0
-httpx==0.27.0
-pydantic==2.7.0
-aiofiles==24.1.0
-python-dotenv==1.0.1
-itsdangerous==2.1.2
-python-multipart==0.0.6
-pyyaml>=6.0
-jinja2>=3.1.0
-
-# 可选：PostgreSQL 数据库支持（用于 HF Spaces 等无持久化存储的环境）
-# 如需使用，请取消下行注释并设置 DATABASE_URL 环境变量
-asyncpg>=0.29.0
+fastapi==0.110.0
+uvicorn[standard]==0.29.0
+httpx==0.27.0
+pydantic==2.7.0
+aiofiles==24.1.0
+python-dotenv==1.0.1
+itsdangerous==2.1.2
+python-multipart==0.0.6
+pyyaml>=6.0
+jinja2>=3.1.0
+requests==2.32.3
+DrissionPage==4.0.5.6
+undetected-chromedriver>=3.5.5
+selenium>=4.15.0
+
+# 可选：PostgreSQL 数据库支持（用于 HF Spaces 等无持久化存储的环境）
+# 如需使用，请取消下行注释并设置 DATABASE_URL 环境变量
+asyncpg>=0.29.0

update.bat

@@ -0,0 +1,111 @@
+@echo off
+REM Gemini Business2API Update Script for Windows
+REM This script updates the project to the latest version
+
+setlocal enabledelayedexpansion
+
+echo ==========================================
+echo Gemini Business2API Update Script
+echo ==========================================
+echo.
+
+REM Check if git is installed
+where git >nul 2>nul
+if %errorlevel% neq 0 (
+    echo [ERROR] Git is not installed. Please install git first.
+    exit /b 1
+)
+
+REM Check if python is installed
+where python >nul 2>nul
+if %errorlevel% neq 0 (
+    echo [ERROR] Python is not installed. Please install Python 3.11+ first.
+    exit /b 1
+)
+
+REM Step 1: Backup current .env file
+echo [INFO] Backing up .env file...
+if exist ".env" (
+    copy /Y ".env" ".env.backup" >nul
+    echo [SUCCESS] .env backed up to .env.backup
+) else (
+    echo [INFO] No .env file found, skipping backup
+)
+
+REM Step 2: Pull latest code from git
+echo [INFO] Pulling latest code from git...
+git fetch origin
+git pull origin main 2>nul
+if %errorlevel% neq 0 (
+    git pull origin master 2>nul
+    if %errorlevel% neq 0 (
+        echo [ERROR] Failed to pull latest code
+        exit /b 1
+    )
+)
+echo [SUCCESS] Code updated successfully
+
+REM Step 3: Restore .env file
+if exist ".env.backup" (
+    echo [INFO] Restoring .env file...
+    move /Y ".env.backup" ".env" >nul
+    echo [SUCCESS] .env restored
+)
+
+REM Step 4: Update Python dependencies
+echo [INFO] Updating Python dependencies...
+
+REM Check if virtual environment exists
+if exist ".venv\Scripts\activate.bat" (
+    echo [INFO] Virtual environment detected, activating...
+    call .venv\Scripts\activate.bat
+)
+
+python -m pip install -r requirements.txt --upgrade
+if %errorlevel% neq 0 (
+    echo [ERROR] Failed to update Python dependencies
+    exit /b 1
+)
+echo [SUCCESS] Python dependencies updated
+
+REM Step 5: Update frontend dependencies
+if exist "frontend" (
+    echo [INFO] Updating frontend dependencies...
+    cd frontend
+
+    REM Check if npm is installed
+    where npm >nul 2>nul
+    if %errorlevel% equ 0 (
+        call npm install
+        if %errorlevel% equ 0 (
+            call npm run build
+            if %errorlevel% equ 0 (
+                echo [SUCCESS] Frontend dependencies updated and built
+            ) else (
+                echo [ERROR] Frontend build failed
+            )
+        ) else (
+            echo [ERROR] npm install failed
+        )
+    ) else (
+        echo [WARNING] npm is not installed. Skipping frontend update.
+    )
+
+    cd ..
+)
+
+REM Step 6: Show completion message
+echo.
+echo ==========================================
+echo [SUCCESS] Update completed successfully!
+echo ==========================================
+echo.
+echo [INFO] To restart the service, run:
+echo   python main.py
+echo.
+echo [INFO] Or if running as a Windows service:
+echo   net stop gemini-business2api
+echo   net start gemini-business2api
+echo.
+
+endlocal

update.sh

@@ -0,0 +1,104 @@
+#!/bin/bash
+
+# Gemini Business2API Update Script for Linux/macOS
+# This script updates the project to the latest version
+
+set -e  # Exit on error
+
+echo "=========================================="
+echo "Gemini Business2API Update Script"
+echo "=========================================="
+echo ""
+
+# Color codes for output
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+YELLOW='\033[1;33m'
+NC='\033[0m' # No Color
+
+# Function to print colored messages
+print_success() {
+    echo -e "${GREEN}✓ $1${NC}"
+}
+
+print_error() {
+    echo -e "${RED}✗ $1${NC}"
+}
+
+print_info() {
+    echo -e "${YELLOW}→ $1${NC}"
+}
+
+# Check if git is installed
+if ! command -v git &> /dev/null; then
+    print_error "Git is not installed. Please install git first."
+    exit 1
+fi
+
+# Check if python3 is installed
+if ! command -v python3 &> /dev/null; then
+    print_error "Python3 is not installed. Please install Python 3.11+ first."
+    exit 1
+fi
+
+# Step 1: Backup current .env file
+print_info "Backing up .env file..."
+if [ -f ".env" ]; then
+    cp .env .env.backup
+    print_success ".env backed up to .env.backup"
+else
+    print_info "No .env file found, skipping backup"
+fi
+
+# Step 2: Pull latest code from git
+print_info "Pulling latest code from git..."
+git fetch origin
+git pull origin main || git pull origin master
+print_success "Code updated successfully"
+
+# Step 3: Restore .env file
+if [ -f ".env.backup" ]; then
+    print_info "Restoring .env file..."
+    mv .env.backup .env
+    print_success ".env restored"
+fi
+
+# Step 4: Update Python dependencies
+print_info "Updating Python dependencies..."
+if [ -d ".venv" ]; then
+    print_info "Virtual environment detected, activating..."
+    source .venv/bin/activate
+fi
+
+pip install -r requirements.txt --upgrade
+print_success "Python dependencies updated"
+
+# Step 5: Update frontend dependencies
+if [ -d "frontend" ]; then
+    print_info "Updating frontend dependencies..."
+    cd frontend
+
+    # Check if npm is installed
+    if command -v npm &> /dev/null; then
+        npm install
+        npm run build
+        print_success "Frontend dependencies updated and built"
+    else
+        print_error "npm is not installed. Skipping frontend update."
+    fi
+
+    cd ..
+fi
+
+# Step 6: Show completion message
+echo ""
+echo "=========================================="
+print_success "Update completed successfully!"
+echo "=========================================="
+echo ""
+print_info "To restart the service, run:"
+echo "  python main.py"
+echo ""
+print_info "Or if using systemd:"
+echo "  sudo systemctl restart gemini-business2api"
+echo ""