feat: 优化密钥状态页面交互体验，添加分组折叠和刷新功能

- Mounted static files directory to serve PWA assets like manifest.json and ServiceWorker scripts.
- Updated `auth.html` and `keys_status.html` templates:
  - Added `<link>` for manifest and icons to support Progressive Web App (PWA) features.
  - Added meta tags for theme color and Apple web app capabilities.
  - Integrated ServiceWorker registration script for offline capabilities.

app/api/gemini_routes.py

@@ -47,7 +47,7 @@ async def list_models(_=Depends(security_service.verify_key),
 
 @router.post("/models/{model_name}:generateContent")
 @router_v1beta.post("/models/{model_name}:generateContent")
-@RetryHandler(max_retries=3, key_manager=Depends(get_key_manager), key_arg="api_key")
+@RetryHandler(max_retries=3, key_arg="api_key")
 async def generate_content(
         model_name: str,
         request: GeminiRequest,
@@ -77,7 +77,7 @@ async def generate_content(
 
 @router.post("/models/{model_name}:streamGenerateContent")
 @router_v1beta.post("/models/{model_name}:streamGenerateContent")
-@RetryHandler(max_retries=3, key_manager=Depends(get_key_manager), key_arg="api_key")
+@RetryHandler(max_retries=3, key_arg="api_key")
 async def stream_generate_content(
         model_name: str,
         request: GeminiRequest,

app/api/openai_routes.py

@@ -46,7 +46,7 @@ async def list_models(
 
 @router.post("/v1/chat/completions")
 @router.post("/hf/v1/chat/completions")
-@RetryHandler(max_retries=3, key_manager=Depends(get_key_manager), key_arg="api_key")
+@RetryHandler(max_retries=3, key_arg="api_key")
 async def chat_completion(
     request: ChatRequest,
     _=Depends(security_service.verify_authorization),

app/main.py

@@ -2,6 +2,7 @@ from fastapi import FastAPI, Request
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import HTMLResponse, RedirectResponse
 from fastapi.templating import Jinja2Templates
+from fastapi.staticfiles import StaticFiles
 from app.core.logger import get_main_logger
 from app.core.security import verify_auth_token
 from app.services.key_manager import get_key_manager_instance
@@ -19,6 +20,9 @@ app = FastAPI()
 # 配置Jinja2模板
 templates = Jinja2Templates(directory="app/templates")
 
+# 配置静态文件
+app.mount("/static", StaticFiles(directory="app/static"), name="static")
+
 # 创建 KeyManager 实例
 key_manager = None
 

app/services/chat/response_handler.py

@@ -232,7 +232,9 @@ def _extract_text(response: Dict[str, Any], model: str, stream: bool = False) ->
                     else:
                         text = candidate["content"]["parts"][0]["text"]
             else:
-                text = candidate["content"]["parts"][0]["text"]
+                text = ""
+                for part in candidate["content"]["parts"]:
+                    text += part["text"]
             text = _add_search_link_text(model, candidate, text)
         else:
             text = "暂无返回"

app/services/chat/retry_handler.py

@@ -3,7 +3,6 @@
 from typing import TypeVar, Callable
 from functools import wraps
 from app.core.logger import get_retry_logger
-from app.services.key_manager import KeyManager
 
 T = TypeVar('T')
 logger = get_retry_logger()
@@ -12,9 +11,8 @@ logger = get_retry_logger()
 class RetryHandler:
     """重试处理装饰器"""
 
-    def __init__(self, max_retries: int = 3, key_manager: KeyManager = None, key_arg: str = "api_key"):
+    def __init__(self, max_retries: int = 3, key_arg: str = "api_key"):
         self.max_retries = max_retries
-        self.key_manager = key_manager
         self.key_arg = key_arg
 
     def __call__(self, func: Callable[..., T]) -> Callable[..., T]:
@@ -29,9 +27,11 @@ class RetryHandler:
                     last_exception = e
                     logger.warning(f"API call failed with error: {str(e)}. Attempt {attempt + 1} of {self.max_retries}")
 
-                    if self.key_manager:
+                    # 从函数参数中获取 key_manager
+                    key_manager = kwargs.get('key_manager')
+                    if key_manager:
                         old_key = kwargs.get(self.key_arg)
-                        new_key = await self.key_manager.handle_api_failure(old_key)
+                        new_key = await key_manager.handle_api_failure(old_key)
                         kwargs[self.key_arg] = new_key
                         logger.info(f"Switched to new API key: {new_key}")
 

app/static/manifest.json

@@ -0,0 +1,17 @@
+{
+  "name": "Gemini Balance",
+  "short_name": "GBalance",
+  "description": "Gemini API密钥管理工具",
+  "start_url": "/",
+  "display": "standalone",
+  "background_color": "#667eea",
+  "theme_color": "#764ba2",
+  "icons": [
+    {
+      "src": "/static/icons/icon-192x192.png",
+      "sizes": "192x192",
+      "type": "image/png",
+      "purpose": "any maskable"
+    }
+  ]
+}

app/static/service-worker.js

@@ -0,0 +1,43 @@
+const CACHE_NAME = 'gbalance-cache-v1';
+const urlsToCache = [
+  '/',
+  '/static/manifest.json',
+  '/static/icons/icon-192x192.png'
+];
+
+self.addEventListener('install', event => {
+  event.waitUntil(
+    caches.open(CACHE_NAME)
+      .then(cache => {
+        console.log('Opened cache');
+        return cache.addAll(urlsToCache);
+      })
+  );
+});
+
+self.addEventListener('fetch', event => {
+  event.respondWith(
+    caches.match(event.request)
+      .then(response => {
+        if (response) {
+          return response;
+        }
+        return fetch(event.request);
+      })
+  );
+});
+
+self.addEventListener('activate', event => {
+  const cacheWhitelist = [CACHE_NAME];
+  event.waitUntil(
+    caches.keys().then(cacheNames => {
+      return Promise.all(
+        cacheNames.map(cacheName => {
+          if (cacheWhitelist.indexOf(cacheName) === -1) {
+            return caches.delete(cacheName);
+          }
+        })
+      );
+    })
+  );
+});

app/templates/auth.html

@@ -4,6 +4,12 @@
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
     <title>验证页面</title>
+    <link rel="manifest" href="/static/manifest.json">
+    <meta name="theme-color" content="#764ba2">
+    <meta name="apple-mobile-web-app-capable" content="yes">
+    <meta name="apple-mobile-web-app-status-bar-style" content="black">
+    <meta name="apple-mobile-web-app-title" content="GBalance">
+    <link rel="icon" href="/static/icons/icon-192x192.png">
     <link href="https://fonts.googleapis.com/css2?family=Roboto:wght@300;400;700&display=swap" rel="stylesheet">
     <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.0.0/css/all.min.css">
     <style>
@@ -28,6 +34,64 @@
             backdrop-filter: blur(10px);
             transition: all 0.4s cubic-bezier(0.4, 0, 0.2, 1);
         }
+
+        @media (max-width: 768px) {
+            .container {
+                width: 85%;
+                padding: 30px;
+            }
+            .logo i {
+                font-size: 40px;
+            }
+            h2 {
+                font-size: 22px;
+            }
+            input {
+                padding: 10px 10px 10px 35px;
+                font-size: 15px;
+            }
+            .input-group i {
+                font-size: 16px;
+            }
+            button {
+                padding: 12px;
+                font-size: 15px;
+            }
+        }
+
+        @media (max-width: 480px) {
+            .container {
+                width: 90%;
+                padding: 25px;
+            }
+            .logo i {
+                font-size: 36px;
+            }
+            h2 {
+                font-size: 20px;
+                margin-bottom: 25px;
+            }
+            form {
+                gap: 15px;
+            }
+            input {
+                padding: 10px 10px 10px 32px;
+                font-size: 14px;
+            }
+            .input-group i {
+                font-size: 15px;
+                left: 10px;
+            }
+            button {
+                padding: 10px;
+                font-size: 14px;
+            }
+            .error-message {
+                font-size: 14px;
+                padding: 8px;
+                margin-top: 12px;
+            }
+        }
         .container:hover {
             transform: translateY(-5px);
             box-shadow: 0 20px 40px rgba(0,0,0,0.25);
@@ -167,5 +231,18 @@
         <p class="error-message">{{ error }}</p>
         {% endif %}
     </div>
+    <script>
+        if ('serviceWorker' in navigator) {
+            window.addEventListener('load', () => {
+                navigator.serviceWorker.register('/static/service-worker.js')
+                    .then(registration => {
+                        console.log('ServiceWorker注册成功:', registration.scope);
+                    })
+                    .catch(error => {
+                        console.log('ServiceWorker注册失败:', error);
+                    });
+            });
+        }
+    </script>
 </body>
 </html>

app/templates/keys_status.html

@@ -4,6 +4,12 @@
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
     <title>API密钥状态</title>
+    <link rel="manifest" href="/static/manifest.json">
+    <meta name="theme-color" content="#764ba2">
+    <meta name="apple-mobile-web-app-capable" content="yes">
+    <meta name="apple-mobile-web-app-status-bar-style" content="black">
+    <meta name="apple-mobile-web-app-title" content="GBalance">
+    <link rel="icon" href="/static/icons/icon-192x192.png">
     <link href="https://fonts.googleapis.com/css2?family=Roboto:wght@300;400;700&display=swap" rel="stylesheet">
     <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.0.0/css/all.min.css">
     <style>
@@ -26,6 +32,74 @@
             position: relative;
             margin: 20px auto;
         }
+
+        @media (max-width: 768px) {
+            .container {
+                width: 100%;
+                padding: 20px;
+                margin: 10px auto;
+            }
+            body {
+                padding: 10px;
+            }
+            h1 {
+                font-size: 24px;
+            }
+            .key-list h2 {
+                font-size: 1.2em;
+                flex-direction: column;
+                gap: 10px;
+                align-items: flex-start;
+            }
+            .key-info {
+                flex-direction: column;
+                align-items: flex-start;
+                gap: 8px;
+            }
+            li {
+                flex-direction: column;
+                gap: 10px;
+            }
+            .copy-btn {
+                width: 100%;
+                justify-content: center;
+            }
+            .key-text {
+                word-break: break-all;
+            }
+            .scroll-buttons {
+                right: 10px;
+                bottom: 10px;
+            }
+            .scroll-btn {
+                width: 35px;
+                height: 35px;
+                font-size: 16px;
+            }
+        }
+
+        @media (max-width: 480px) {
+            .container {
+                padding: 15px;
+            }
+            h1 {
+                font-size: 20px;
+            }
+            .key-list {
+                padding: 15px;
+            }
+            .status-badge {
+                padding: 3px 8px;
+                font-size: 0.8em;
+            }
+            .fail-count {
+                font-size: 0.8em;
+            }
+            .total {
+                font-size: 1em;
+                padding: 12px 20px;
+            }
+        }
         h1 {
             color: #2c3e50;
             text-align: center;
@@ -67,6 +141,26 @@
             font-size: 1.5em;
             padding-bottom: 10px;
             border-bottom: 2px solid rgba(0,0,0,0.1);
+            cursor: pointer;
+        }
+        
+        .key-list h2 .toggle-icon {
+            margin-right: 10px;
+            transition: transform 0.3s ease;
+        }
+        
+        .key-list h2 .toggle-icon.collapsed {
+            transform: rotate(-90deg);
+        }
+        
+        .key-list .key-content {
+            transition: max-height 0.3s ease-out;
+            overflow: hidden;
+            max-height: 2000px;
+        }
+        
+        .key-list .key-content.collapsed {
+            max-height: 0;
         }
         ul {
             list-style-type: none;
@@ -222,23 +316,80 @@
         .scroll-btn:active {
             transform: scale(0.95);
         }
+        .refresh-btn {
+            position: absolute;
+            top: 20px;
+            right: 20px;
+            background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+            color: #fff;
+            border: none;
+            padding: 10px 20px;
+            border-radius: 25px;
+            cursor: pointer;
+            font-size: 14px;
+            font-weight: bold;
+            transition: all 0.3s ease;
+            display: flex;
+            align-items: center;
+            justify-content: center;
+            gap: 8px;
+            box-shadow: 0 5px 15px rgba(0, 0, 0, 0.1);
+            z-index: 10;
+        }
+        
+        .refresh-btn:hover {
+            transform: scale(1.05);
+            box-shadow: 0 8px 20px rgba(118, 75, 162, 0.3);
+            background: linear-gradient(135deg, #764ba2 0%, #667eea 100%);
+        }
+        
+        .refresh-btn:active {
+            transform: scale(0.95);
+        }
+
+        .refresh-btn i {
+            transition: transform 0.5s ease;
+        }
+
+        .refresh-btn.loading i {
+            animation: spin 1s linear infinite;
+        }
+
+        @keyframes spin {
+            from { transform: rotate(0deg); }
+            to { transform: rotate(360deg); }
+        }
+
+        @media (max-width: 768px) {
+            .refresh-btn {
+                top: 10px;
+                right: 10px;
+                padding: 8px 16px;
+                font-size: 12px;
+            }
+        }
     </style>
 </head>
 <body>
     <div class="container">
+        <button class="refresh-btn" onclick="refreshPage(this)">
+            <i class="fas fa-sync-alt"></i> 刷新
+        </button>
         <h1>API密钥状态</h1>
         <div class="key-list">
-            <h2>
+            <h2 onclick="toggleSection(this, 'validKeys')">
                 <span>
+                    <i class="fas fa-chevron-down toggle-icon"></i>
                     <i class="fas fa-check-circle" style="color: #27ae60;"></i>
                     有效密钥
                 </span>
-                <button class="copy-btn" onclick="copyKeys('valid')">
+                <button class="copy-btn" onclick="event.stopPropagation(); copyKeys('valid')">
                     <i class="fas fa-copy"></i>
                     批量复制
                 </button>
             </h2>
-            <ul id="validKeys">
+            <div class="key-content">
+                <ul id="validKeys">
                 {% for key, fail_count in valid_keys.items() %}
                     <li>
                         <div class="key-info">
@@ -257,20 +408,23 @@
                         </button>
                     </li>
                 {% endfor %}
-            </ul>
+                </ul>
+            </div>
         </div>
         <div class="key-list">
-            <h2>
+            <h2 onclick="toggleSection(this, 'invalidKeys')">
                 <span>
+                    <i class="fas fa-chevron-down toggle-icon"></i>
                     <i class="fas fa-times-circle" style="color: #e74c3c;"></i>
                     无效密钥
                 </span>
-                <button class="copy-btn" onclick="copyKeys('invalid')">
+                <button class="copy-btn" onclick="event.stopPropagation(); copyKeys('invalid')">
                     <i class="fas fa-copy"></i>
                     批量复制
                 </button>
             </h2>
-            <ul id="invalidKeys">
+            <div class="key-content">
+                <ul id="invalidKeys">
                 {% for key, fail_count in invalid_keys.items() %}
                     <li>
                         <div class="key-info">
@@ -289,7 +443,8 @@
                         </button>
                     </li>
                 {% endfor %}
-            </ul>
+                </ul>
+            </div>
         </div>
         <div class="total">
             <i class="fas fa-key"></i> 总密钥数：{{ total }}
@@ -392,6 +547,36 @@
                 scrollButtons.style.display = 'none';
             }
         });
+        function refreshPage(button) {
+            button.classList.add('loading');
+            button.disabled = true;
+            
+            // 添加延迟以显示加载动画
+            setTimeout(() => {
+                window.location.reload();
+            }, 300);
+        }
+    </script>
+    <script>
+        function toggleSection(header, sectionId) {
+            const toggleIcon = header.querySelector('.toggle-icon');
+            const content = header.nextElementSibling;
+            
+            toggleIcon.classList.toggle('collapsed');
+            content.classList.toggle('collapsed');
+        }
+
+        if ('serviceWorker' in navigator) {
+            window.addEventListener('load', () => {
+                navigator.serviceWorker.register('/static/service-worker.js')
+                    .then(registration => {
+                        console.log('ServiceWorker注册成功:', registration.scope);
+                    })
+                    .catch(error => {
+                        console.log('ServiceWorker注册失败:', error);
+                    });
+            });
+        }
     </script>
 </body>
 </html>