Refactor config, add multi-SSO support and admin panel

- Move configuration loading to `src/config.py` and utilize environment variables.
- Add support for multiple SSO tokens loaded from the `SSO` environment variable.
- Implement logic to cycle through available SSO tokens for API requests.
- Introduce a basic password-protected admin panel (`/admin`) for managing SSO tokens.
- Update README with detailed configuration instructions using `.env`.
- Change default application port to 7860.
- Add `.env` to `.gitignore`.
- Include support for various environment variables (proxy, CF cookie, image keys, etc.).
- Add basic retry mechanism for API calls.

.env.example

@@ -0,0 +1,40 @@
+# .env 示例文件
+
+# 是否使用临时对话模式 (默认为 true)
+# IS_TEMP_CONVERSATION=true
+
+# 是否使用自定义 SSO 令牌模式 (默认为 false)
+# IS_CUSTOM_SSO=false
+
+# Grok API 的基础 URL (默认为 https://grok.com)
+# API_BASE_URL=https://grok.com
+
+# 用于访问本程序 API 的密钥 (默认为 sk-123456)
+# API_KEY=your_api_key_here
+
+# PicGo 图床的 API 密钥 (可选)
+# PICGO_KEY=
+
+# Tumy 图床的 API 密钥 (可选)
+# TUMY_KEY=
+
+# HTTP/HTTPS 代理地址 (可选)
+# PROXY=
+
+# 是否开启管理页面 (可选)
+# MANAGER_SWITCH= # 是否开启管理页面 (可选，默认为关闭)
+
+# 管理页面的访问密码 (可选，开启管理页面时需要设置)
+# ADMINPASSWORD=
+
+# Cloudflare 的 cf_clearance cookie 值 (可选)
+# CF_CLEARANCE=
+
+# 应用程序监听的端口 (默认为 7860)
+# PORT=7860
+
+# Flask 应用程序的 Secret Key，用于会话管理 (如果未设置，将自动生成)
+# FLASK_SECRET_KEY=
+
+# Grok SSO 令牌，多个令牌使用逗号分隔
+# SSO=sso_token_1,sso_token_2

.gitignore

@@ -16,4 +16,7 @@ venv/
 *.swp
 
 # Hugging Face Space nested directory (if it exists locally)
-g2api-test/
+g2api-test/
+
+# Environment variables
+.env

README.md

@@ -12,7 +12,7 @@ pinned: false
 
 ## 项目简介
 
-本项目旨在将现有的 Grok 网页版逆向封装为 API 的 Python Flask 程序进行全面改造。改造目标是构建一个结构专业、功能完善（包含 Web UI）、文档齐全且易于部署的应用。项目严格遵守内存数据模式，所有运行时数据仅存储在内存中，不进行本地持久化。
+本项目基于 [https://github.com/xLmiler/grok2api_python](https://github.com/xLmiler/grok2api_python) 进行二次开发改造。旨在将现有的 Grok 网页版逆向封装为 API 的 Python Flask 程序进行全面改造。改造目标是构建一个结构专业、功能完善（包含 Web UI）、文档齐全且易于部署的应用。项目严格遵守内存数据模式，所有运行时数据仅存储在内存中，不进行本地持久化。
 
 ## 功能列表
 
@@ -44,25 +44,47 @@ pinned: false
    ```
 3. 运行 Docker 容器：
    ```bash
-   docker run -p 5000:5000 grok-api-remodel
+   docker run -p 7860:7860 grok-api-remodel
    ```
-   应用程序将在本地的 5000 端口运行。
+   应用程序将在本地的 7860 端口运行。
 
 ### 部署到 Hugging Face Spaces (使用 Docker)
 
 1. 在 Hugging Face Spaces 上创建一个新的 Space。
 2. 选择 Docker 作为 Space SDK。
 3. 将本项目仓库关联到你的 Hugging Face Space。
-4. Hugging Face Spaces 将会自动检测项目根目录下的 `Dockerfile` 并进行构建和部署。确保你的 `Dockerfile` 配置正确，暴露了 Flask 应用运行的端口（默认为 5000）。
+4. Hugging Face Spaces 将会自动检测项目根目录下的 `Dockerfile` 并进行构建和部署。确保你的 `Dockerfile` 配置正确，暴露了 Flask 应用运行的端口（默认为 7860）。
 
 ## 配置说明
 
-本项目的部分配置通过环境变量进行管理。
-
-- `YOUR_API_KEY`: 用于认证或访问外部服务的 API 密钥（如果需要）。
-- `LISTEN_PORT`: 应用程序监听的端口（默认为 5000）。
+本项目通过环境变量管理大部分配置项。您可以在项目根目录下创建 `.env` 文件来设置这些环境变量。
+
+主要环境变量列表：
+
+- `IS_TEMP_CONVERSATION`: 是否使用临时对话模式（默认为 `true`）。
+- `IS_CUSTOM_SSO`: 是否使用自定义 SSO 令牌模式（默认为 `false`）。
+- `API_BASE_URL`: Grok API 的基础 URL（默认为 `https://grok.com`）。
+- `API_KEY`: 用于访问本程序 API 的密钥（默认为 `sk-123456`）。
+- `PICGO_KEY`: PicGo 图床的 API 密钥（可选）。
+- `TUMY_KEY`: Tumy 图床的 API 密钥（可选）。
+- `PROXY`: HTTP/HTTPS 代理地址（可选）。
+- `MANAGER_SWITCH`: 是否开启管理页面（可选，默认为关闭）。
+- `ADMINPASSWORD`: 管理页面的访问密码（可选，开启管理页面时需要设置）。
+- `CF_CLEARANCE`: Cloudflare 的 cf_clearance cookie 值（可选）。
+- `PORT`: 应用程序监听的端口（默认为 `7860`）。
+- `FLASK_SECRET_KEY`: Flask 应用程序的 Secret Key，用于会话管理（如果未设置，将自动生成）。
+- `SSO`: Grok SSO 令牌，多个令牌使用逗号分隔。
+
+示例 `.env` 文件内容：
+
+```dotenv
+IS_TEMP_CONVERSATION=true
+API_KEY=your_api_key_here
+PORT=7860
+SSO=sso_token_1,sso_token_2
+```
 
-请根据你的部署环境设置相应的环境变量。在 Hugging Face Spaces 上，可以在 Space 的设置页面配置环境变量。
+请根据您的实际需求配置这些环境变量。
 
 ## API 使用指南
 

app.py

@@ -6,77 +6,14 @@ import base64
 import sys
 import inspect
 import secrets
-from dotenv import load_dotenv
 import requests
 from flask import Flask, request, Response, jsonify, stream_with_context, render_template, redirect, session
 from curl_cffi import requests as curl_requests
 from werkzeug.middleware.proxy_fix import ProxyFix
 
+from src.config import CONFIG, DEFAULT_HEADERS # 导入配置项和默认请求头
 from src.core.logger import logger # 从新的位置导入 logger
 
-current_dir = os.path.dirname(os.path.abspath(__file__))
-env_path = os.path.join(current_dir, '.env')
-load_dotenv(env_path)
-
-CONFIG = {
-    "MODELS": {
-        'grok-2': 'grok-latest',
-        'grok-2-imageGen': 'grok-latest',
-        'grok-2-search': 'grok-latest',
-        "grok-3": "grok-3",
-        "grok-3-search": "grok-3",
-        "grok-3-imageGen": "grok-3",
-        "grok-3-deepsearch": "grok-3",
-        "grok-3-reasoning": "grok-3"
-    },
-    "API": {
-        "IS_TEMP_CONVERSATION": os.getenv("IS_TEMP_CONVERSATION", "true").lower() == "true",
-        "IS_CUSTOM_SSO": os.getenv("IS_CUSTOM_SSO", "false").lower() == "true",
-        "BASE_URL": "https://grok.com",
-        "API_KEY": os.getenv("API_KEY", "sk-123456"),
-        "SIGNATURE_COOKIE": None,
-        "PICGO_KEY": os.getenv("PICGO_KEY") or None,
-        "TUMY_KEY": os.getenv("TUMY_KEY") or None,
-        "RETRY_TIME": 1000,
-        "PROXY": os.getenv("PROXY") or None
-    },
-    "ADMIN": {
-        "MANAGER_SWITCH": os.getenv("MANAGER_SWITCH") or None,
-        "PASSWORD": os.getenv("ADMINPASSWORD") or None
-    },
-    "SERVER": {
-        "COOKIE": None,
-        "CF_CLEARANCE":os.getenv("CF_CLEARANCE") or None,
-        "PORT": int(os.getenv("PORT", 5200))
-    },
-    "RETRY": {
-        "RETRYSWITCH": False,
-        "MAX_ATTEMPTS": 2
-    },
-    "SHOW_THINKING": os.getenv("SHOW_THINKING") == "true",
-    "IS_THINKING": False,
-    "IS_IMG_GEN": False,
-    "IS_IMG_GEN2": False,
-    "ISSHOW_SEARCH_RESULTS": os.getenv("ISSHOW_SEARCH_RESULTS", "true").lower() == "true"
-}
-
-DEFAULT_HEADERS = {
-    'Accept': '*/*',
-    'Accept-Language': 'zh-CN,zh;q=0.9',
-    'Accept-Encoding': 'gzip, deflate, br, zstd',
-    'Content-Type': 'text/plain;charset=UTF-8',
-    'Connection': 'keep-alive',
-    'Origin': 'https://grok.com',
-    'Priority': 'u=1, i',
-    'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/133.0.0.0 Safari/537.36',
-    'Sec-Ch-Ua': '"Not(A:Brand";v="99", "Google Chrome";v="133", "Chromium";v="133"',
-    'Sec-Ch-Ua-Mobile': '?0',
-    'Sec-Ch-Ua-Platform': '"macOS"',
-    'Sec-Fetch-Dest': 'empty',
-    'Sec-Fetch-Mode': 'cors',
-    'Sec-Fetch-Site': 'same-origin',
-    'Baggage': 'sentry-public_key=b311e0f2690c81f25e2c4cf6d4f7ce1c'
-}
 
 from src.core.auth_token_manager import AuthTokenManager # 从新的位置导入 AuthTokenManager
 

src/config.py

@@ -0,0 +1,73 @@
+# 应用程序配置项
+import os
+from dotenv import load_dotenv
+
+# 加载 .env 文件
+current_dir = os.path.dirname(os.path.abspath(__file__))
+# 假设 .env 文件在项目根目录，需要调整路径
+# env_path = os.path.join(current_dir, '.env')
+# load_dotenv(env_path)
+# 如果 .env 文件在项目根目录，则直接加载
+load_dotenv()
+
+
+CONFIG = {
+    "MODELS": {
+        'grok-2': 'grok-latest',
+        'grok-2-imageGen': 'grok-latest',
+        'grok-2-search': 'grok-latest',
+        "grok-3": "grok-3",
+        "grok-3-search": "grok-3",
+        "grok-3-imageGen": "grok-3",
+        "grok-3-deepsearch": "grok-3",
+        "grok-3-reasoning": "grok-3"
+    },
+    "API": {
+        "IS_TEMP_CONVERSATION": os.getenv("IS_TEMP_CONVERSATION", "true").lower() == "true",
+        "IS_CUSTOM_SSO": os.getenv("IS_CUSTOM_SSO", "false").lower() == "true",
+        "BASE_URL": os.getenv("API_BASE_URL", "https://grok.com"),
+        "API_KEY": os.getenv("API_KEY", "sk-123456"),
+        "SIGNATURE_COOKIE": None,
+        "PICGO_KEY": os.getenv("PICGO_KEY") or None,
+        "TUMY_KEY": os.getenv("TUMY_KEY") or None,
+        "RETRY_TIME": 1000,
+        "PROXY": os.getenv("PROXY") or None
+    },
+    "ADMIN": {
+        "MANAGER_SWITCH": os.getenv("MANAGER_SWITCH") or None,
+        "PASSWORD": os.getenv("ADMINPASSWORD") or None
+    },
+    "SERVER": {
+        "COOKIE": None,
+        "CF_CLEARANCE":os.getenv("CF_CLEARANCE") or None,
+        "PORT": int(os.getenv("PORT", 7860))
+    },
+    "RETRY": {
+        "RETRYSWITCH": False,
+        "MAX_ATTEMPTS": 2
+    },
+    "SHOW_THINKING": os.getenv("SHOW_THINKING") == "true",
+    "IS_THINKING": False,
+    "IS_IMG_GEN": False,
+    "IS_IMG_GEN2": False,
+    "ISSHOW_SEARCH_RESULTS": os.getenv("ISSHOW_SEARCH_RESULTS", "true").lower() == "true"
+}
+
+# 默认请求头
+DEFAULT_HEADERS = {
+    'Accept': '*/*',
+    'Accept-Language': 'zh-CN,zh;q=0.9',
+    'Accept-Encoding': 'gzip, deflate, br, zstd',
+    'Content-Type': 'text/plain;charset=UTF-8',
+    'Connection': 'keep-alive',
+    'Origin': 'https://grok.com',
+    'Priority': 'u=1, i',
+    'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/133.0.0.0 Safari/537.36',
+    'Sec-Ch-Ua': '"Not(A:Brand";v="99", "Google Chrome";v="133", "Chromium";v="133"',
+    'Sec-Ch-Ua-Mobile': '?0',
+    'Sec-Ch-Ua-Platform': '"macOS"',
+    'Sec-Fetch-Dest': 'empty',
+    'Sec-Fetch-Mode': 'cors',
+    'Sec-Fetch-Site': 'same-origin',
+    'Baggage': 'sentry-public_key=b311e0f2690c81f25e2c4cf6d4f7ce1c'
+}