Upload 10 files

.gitignore

@@ -44,4 +44,4 @@ static/
 # OS
 .DS_Store
 Thumbs.db
-old_version.py
+old_version.py

Dockerfile

@@ -1,31 +1,62 @@
-FROM python:3.11-slim
-WORKDIR /app
-
-ENV PYTHONDONTWRITEBYTECODE=1 \
-    PYTHONUNBUFFERED=1
-
-# Install Python dependencies
-COPY requirements.txt .
-RUN apt-get update && apt-get install -y --no-install-recommends \
-    gcc \
-    && pip install --no-cache-dir -r requirements.txt \
-    && apt-get purge -y gcc \
-    && apt-get autoremove -y \
-    && rm -rf /var/lib/apt/lists/*
-
-# Copy backend code
-COPY main.py .
-COPY core ./core
-COPY util ./util
-
-# Copy prebuilt static assets
-COPY static ./static
-
-# Create data directory (local + HF Spaces Pro)
-RUN mkdir -p ./data
-
-# Declare data volume
-VOLUME ["/app/data"]
-
-# Start service
-CMD ["python", "-u", "main.py"]
+# Stage 1: 构建前端
+FROM node:20-slim AS frontend-builder
+WORKDIR /app/frontend
+
+# 先复制 package 文件利用 Docker 缓存
+COPY frontend/package.json frontend/package-lock.json ./
+RUN npm install --silent
+
+# 复制前端源码并构建
+COPY frontend/ ./
+RUN npm run build
+
+# Stage 2: 最终运行时镜像
+FROM python:3.11-slim
+WORKDIR /app
+
+ENV PYTHONDONTWRITEBYTECODE=1 \
+    PYTHONUNBUFFERED=1 \
+    TZ=Asia/Shanghai
+
+# 安装 Python 依赖和浏览器依赖（合并为单一 RUN 指令以减少层数）
+COPY requirements.txt .
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+        gcc \
+        curl \
+        tzdata \
+        chromium chromium-driver \
+        dbus dbus-x11 \
+        libglib2.0-0 libnss3 libnspr4 libatk1.0-0 libatk-bridge2.0-0 \
+        libcups2 libdrm2 libxkbcommon0 libxcomposite1 libxdamage1 \
+        libxfixes3 libxrandr2 libgbm1 libasound2 libpango-1.0-0 \
+        libcairo2 fonts-liberation fonts-noto-cjk && \
+    ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone && \
+    pip install --no-cache-dir -r requirements.txt && \
+    apt-get purge -y gcc && \
+    apt-get autoremove -y && \
+    rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*
+
+# 复制后端代码
+COPY main.py .
+COPY core ./core
+COPY util ./util
+
+# 从 builder 阶段只复制构建好的静态文件
+COPY --from=frontend-builder /app/static ./static
+
+# 创建数据目录
+RUN mkdir -p ./data
+
+# 声明数据卷
+VOLUME ["/app/data"]
+
+# 声明端口
+EXPOSE 7860
+
+# 健康检查
+HEALTHCHECK --interval=30s --timeout=10s --start-period=10s --retries=3 \
+    CMD curl -f http://localhost:7860/admin/health || exit 1
+
+# 启动服务
+CMD ["python", "-u", "main.py"]

LICENSE

@@ -18,4 +18,4 @@ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.
+SOFTWARE.

docker-compose.yml

@@ -1,6 +1,7 @@
 services:
   gemini-api:
     build: .
+    container_name: gemini-business2api
     ports:
       - "7860:7860"
     volumes:
@@ -8,3 +9,16 @@ services:
     env_file:
       - .env
     restart: unless-stopped
+    # 健康检查
+    healthcheck:
+      test: ["CMD", "curl", "-f", "http://localhost:7860/admin/health"]
+      interval: 30s
+      timeout: 10s
+      retries: 3
+      start_period: 10s
+    # 日志限制
+    logging:
+      driver: json-file
+      options:
+        max-size: "10m"
+        max-file: "3"

main.py

@@ -428,6 +428,11 @@ async def serve_logo():
         return FileResponse(logo_path)
     raise HTTPException(404, "Not Found")
 
+@app.get("/admin/health")
+async def health_check():
+    """健康检查端点，用于 Docker HEALTHCHECK"""
+    return {"status": "ok"}
+
 # ---------- Session 中间件配置 ----------
 from starlette.middleware.sessions import SessionMiddleware
 app.add_middleware(

requirements.txt

@@ -1,6 +1,6 @@
 fastapi==0.110.0
 uvicorn[standard]==0.29.0
-httpx==0.27.0
+httpx[socks]==0.27.0
 pydantic==2.7.0
 aiofiles==24.1.0
 python-dotenv==1.0.1
@@ -8,7 +8,7 @@ itsdangerous==2.1.2
 python-multipart==0.0.6
 pyyaml>=6.0
 jinja2>=3.1.0
-requests==2.32.3
+requests[socks]==2.32.3
 DrissionPage==4.0.5.6
 undetected-chromedriver>=3.5.5
 selenium>=4.15.0

setup.bat

@@ -0,0 +1,185 @@
+@echo off
+REM Gemini Business2API Setup Script
+REM Handles both installation and updates automatically
+REM Uses uv for Python environment management
+REM Usage: setup.bat
+
+setlocal enabledelayedexpansion
+
+echo ==========================================
+echo Gemini Business2API Setup Script
+echo ==========================================
+echo.
+
+REM Color codes for output (using echo instead of ANSI codes for better Windows compatibility)
+set GREEN=[92m
+set RED=[91m
+set YELLOW=[93m
+set BLUE=[94m
+set NC=[0m
+
+REM Function to print colored messages (simplified for Windows)
+set "PRINT_SUCCESS=echo [SUCCESS]"
+set "PRINT_ERROR=echo [ERROR]"
+set "PRINT_INFO=echo [INFO]"
+set "PRINT_STEP=echo [STEP]"
+
+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 Step 1: Install or update uv
+echo [STEP] Step 1: Installing/Updating uv...
+
+where uv >nul 2>nul
+if %errorlevel% neq 0 (
+    echo [INFO] uv not found, installing...
+    REM Install uv using pipx or pip
+    pipx install uv 2>nul
+    if %errorlevel% neq 0 (
+        pip install --user uv 2>nul
+        if %errorlevel% neq 0 (
+            REM Fallback: download and install uv binary
+            curl -LsSf https://astral.sh/uv/install.bat | cmd
+        )
+    )
+    if %errorlevel% equ 0 (
+        echo [SUCCESS] uv installed successfully
+    ) else (
+        echo [ERROR] Failed to install uv
+        exit /b 1
+    )
+) else (
+    echo [INFO] Updating uv to latest version...
+    uv pip install --upgrade uv
+    echo [SUCCESS] uv updated
+)
+echo.
+
+REM Step 2: Ensure Python 3.11 is available
+echo [STEP] Step 2: Ensuring Python 3.11 is available...
+uv python list | findstr /C:"3.11" >nul
+if %errorlevel% neq 0 (
+    echo [INFO] Python 3.11 not found, installing...
+    uv python install 3.11
+    if %errorlevel% neq 0 (
+        echo [ERROR] Failed to install Python 3.11
+        exit /b 1
+    )
+    echo [SUCCESS] Python 3.11 installed
+) else (
+    echo [SUCCESS] Python 3.11 is already available
+)
+echo.
+
+REM Step 3: Pull latest code from git
+echo [STEP] Step 3: Syncing code from repository...
+echo [INFO] Fetching latest changes...
+git fetch origin
+
+echo [INFO] Pulling latest code...
+git pull origin main 2>nul || git pull origin master 2>nul
+if %errorlevel% equ 0 (
+    echo [SUCCESS] Code synchronized successfully
+) else (
+    echo [INFO] No remote changes to pull
+)
+echo.
+
+REM Step 4: Setup .env file if it doesn't exist
+echo [STEP] Step 4: Checking configuration...
+if exist .env (
+    echo [INFO] .env file exists
+) else (
+    if exist .env.example (
+        copy .env.example .env >nul
+        echo [SUCCESS] .env file created from .env.example
+        echo [INFO] Please edit .env and configure your ADMIN_KEY
+    ) else (
+        echo [ERROR] .env.example not found
+        exit /b 1
+    )
+)
+echo.
+
+REM Step 5: Setup Python virtual environment
+echo [STEP] Step 5: Setting up Python environment...
+if exist .venv (
+    echo [INFO] Virtual environment already exists
+) else (
+    echo [INFO] Creating virtual environment with Python 3.11...
+    uv venv --python 3.11 .venv
+    if %errorlevel% neq 0 (
+        echo [ERROR] Failed to create virtual environment
+        exit /b 1
+    )
+    echo [SUCCESS] Virtual environment created
+)
+echo.
+
+REM Step 6: Install/Update Python dependencies
+echo [STEP] Step 6: Installing Python dependencies...
+echo [INFO] Using uv to install dependencies (this may take a moment)...
+.venv\Scripts\python.exe -m pip install --upgrade pip --quiet
+uv pip install -r requirements.txt
+if %errorlevel% neq 0 (
+    echo [ERROR] Failed to install Python dependencies
+    exit /b 1
+)
+echo [SUCCESS] Python dependencies installed
+echo.
+
+REM Step 7: Setup frontend
+echo [STEP] Step 7: Setting up frontend...
+if exist frontend (
+    cd frontend
+
+    REM Check if npm is installed
+    where npm >nul 2>nul
+    if %errorlevel% equ 0 (
+        echo [INFO] Installing dependencies...
+        npm install
+
+        echo [INFO] Building frontend...
+        npm run build
+        echo [SUCCESS] Frontend built successfully
+    ) else (
+        echo [ERROR] npm is not installed. Please install Node.js and npm first.
+        cd ..
+        exit /b 1
+    )
+
+    cd ..
+) else (
+    echo [ERROR] Frontend directory not found. Are you in the project root?
+    exit /b 1
+)
+echo.
+
+REM Step 8: Show completion message
+echo ==========================================
+echo [SUCCESS] Setup completed successfully!
+echo ==========================================
+echo.
+
+if exist .env (
+    echo [INFO] Next steps:
+    echo.
+    echo   1. Edit .env file if needed:
+    echo      notepad .env
+    echo.
+    echo   2. Start the service:
+    echo      .venv\Scripts\python.exe main.py
+    echo.
+    echo   3. Access the admin panel:
+    echo      http://localhost:7860/
+    echo.
+    echo [INFO] To activate virtual environment later, run:
+    echo   .venv\Scripts\activate.bat
+)
+echo.
+
+endlocal

setup.sh

@@ -0,0 +1,172 @@
+#!/bin/bash
+
+# Gemini Business2API Setup Script
+# Handles both installation and updates automatically
+# Uses uv for Python environment management
+# Usage: ./setup.sh
+
+set -e  # Exit on error
+
+echo "=========================================="
+echo "Gemini Business2API Setup 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}✓ $1${NC}"
+}
+
+print_error() {
+    echo -e "${RED}✗ $1${NC}"
+}
+
+print_info() {
+    echo -e "${YELLOW}→ $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
+
+# Step 1: Install or update uv
+print_step "Step 1: Installing/Updating uv..."
+if ! command -v uv &> /dev/null; then
+    print_info "uv not found, installing..."
+    # Install uv using pipx or pip
+    if command -v pipx &> /dev/null; then
+        pipx install uv
+    elif command -v pip &> /dev/null; then
+        pip install --user uv
+    else
+        # Fallback: download and install uv binary
+        curl -LsSf https://astral.sh/uv/install.sh | sh
+        export PATH="$HOME/.local/bin:$PATH"
+    fi
+    print_success "uv installed successfully"
+else
+    print_info "Updating uv to latest version..."
+    uv pip install --upgrade uv
+    print_success "uv updated"
+fi
+echo ""
+
+# Step 2: Ensure Python 3.11 is available
+print_step "Step 2: Ensuring Python 3.11 is available..."
+if ! uv python list | grep -q "3.11"; then
+    print_info "Python 3.11 not found, installing..."
+    uv python install 3.11
+    print_success "Python 3.11 installed"
+else
+    print_success "Python 3.11 is already available"
+fi
+echo ""
+
+# Step 3: Pull latest code from git
+print_step "Step 3: Syncing code from repository..."
+print_info "Fetching latest changes..."
+git fetch origin
+
+print_info "Pulling latest code..."
+if git pull origin main 2>/dev/null || git pull origin master 2>/dev/null; then
+    print_success "Code synchronized successfully"
+else
+    print_info "No remote changes to pull"
+fi
+echo ""
+
+# Step 4: Setup .env file if it doesn't exist
+print_step "Step 4: Checking configuration..."
+if [ -f ".env" ]; then
+    print_info ".env file exists"
+else
+    if [ -f ".env.example" ]; then
+        cp .env.example .env
+        print_success ".env file created from .env.example"
+        print_info "Please edit .env and configure your ADMIN_KEY"
+    else
+        print_error ".env.example not found"
+        exit 1
+    fi
+fi
+echo ""
+
+# Step 5: Setup Python virtual environment
+print_step "Step 5: Setting up Python environment..."
+if [ -d ".venv" ]; then
+    print_info "Virtual environment already exists"
+else
+    print_info "Creating virtual environment with Python 3.11..."
+    uv venv --python 3.11 .venv
+    print_success "Virtual environment created"
+fi
+echo ""
+
+# Step 6: Install/Update Python dependencies
+print_step "Step 6: Installing Python dependencies..."
+print_info "Using uv to install dependencies (this may take a moment)..."
+uv pip install --python .venv/bin/python -r requirements.txt --system
+print_success "Python dependencies installed"
+echo ""
+
+# Step 7: Setup frontend
+print_step "Step 7: Setting up frontend..."
+if [ -d "frontend" ]; then
+    cd frontend
+
+    # Check if npm is installed
+    if command -v npm &> /dev/null; then
+        print_info "Installing dependencies..."
+        npm install
+
+        print_info "Building frontend..."
+        npm run build
+        print_success "Frontend built successfully"
+    else
+        print_error "npm is not installed. Please install Node.js and npm first."
+        cd ..
+        exit 1
+    fi
+
+    cd ..
+else
+    print_error "Frontend directory not found. Are you in the project root?"
+    exit 1
+fi
+echo ""
+
+# Step 8: Show completion message
+echo "=========================================="
+print_success "Setup completed successfully!"
+echo "=========================================="
+echo ""
+
+if [ -f ".env" ]; then
+    print_info "Next steps:"
+    echo ""
+    echo "  1. Edit .env file if needed:"
+    echo "     ${BLUE}nano .env${NC}  or  ${BLUE}vim .env${NC}"
+    echo ""
+    echo "  2. Start the service:"
+    echo "     ${BLUE}uv run python main.py${NC}"
+    echo ""
+    echo "  3. Access the admin panel:"
+    echo "     ${BLUE}http://localhost:7860/${NC}"
+    echo ""
+    print_info "To activate virtual environment later, run:"
+    echo "  ${BLUE}source .venv/bin/activate${NC}"
+fi
+echo ""