Upload 53 files

config.py

@@ -24,24 +24,30 @@ TOKEN_EXPIRY_DAYS = 30
 PLATFORMS = [
     {"id": "anthropic", "name": "Anthropic"},
     {"id": "openai", "name": "OpenAI"},
-    {"id": "google", "name": "Google"}
+    {"id": "google", "name": "Google"},
+    {"id": "deepseek", "name": "Deepseek"}
 ]
 
 # 平台标签样式
 PLATFORM_STYLES = {
     "anthropic": {
-        "background-color": "rgba(236, 72, 153, 0.1)",
-        "border-color": "rgba(236, 72, 153, 0.3)",
-        "color": "#be185d"
+        "background-color": "rgba(217, 119, 87, 0.1)",
+        "border-color": "rgba(217, 119, 87, 0.3)",
+        "color": "#c25032"
     },
     "openai": {
-        "background-color": "rgba(16, 185, 129, 0.1)",
-        "border-color": "rgba(16, 185, 129, 0.3)",
-        "color": "#047857"
+        "background-color": "rgba(16, 163, 127, 0.1)",
+        "border-color": "rgba(16, 163, 127, 0.3)",
+        "color": "#0e8c6b"
     },
     "google": {
-        "background-color": "rgba(59, 130, 246, 0.1)",
-        "border-color": "rgba(59, 130, 246, 0.3)",
-        "color": "#1d4ed8"
+        "background-color": "rgba(28, 125, 255, 0.1)",
+        "border-color": "rgba(28, 125, 255, 0.3)",
+        "color": "#0051c3"
+    },
+    "deepseek": {
+        "background-color": "rgba(77, 107, 254, 0.1)",
+        "border-color": "rgba(77, 107, 254, 0.3)",
+        "color": "#3246d3"
     }
 }

core/api/anthropic.py

@@ -0,0 +1,85 @@
+"""
+Anthropic API相关功能模块 - 提供Anthropic Claude API的调用和验证功能
+"""
+import requests
+
+def validate_api_key(api_key):
+    """
+    验证Anthropic API密钥是否有效
+    
+    Args:
+        api_key (str): 需要验证的Anthropic API密钥
+    
+    Returns:
+        dict: 包含验证结果的字典，包括:
+            - success (bool): API密钥验证成功为True，失败为False
+            - return_message (str): 验证响应的具体消息，成功时为"200: Success!"，失败时为错误详情
+            - balance (float): 账户余额，固定为0
+            - states (str): 账户状态，付费版或免费版
+    """
+    url = "https://api.anthropic.com/v1/messages"
+    
+    headers = {
+        "x-api-key": api_key,
+        "content-type": "application/json",
+        "anthropic-version": "2023-06-01"
+    }
+    
+    payload = {
+        "model": "claude-3-5-sonnet-20241022",
+        "max_tokens": 1,
+        "messages": [
+            {"role": "user", "content": "Hello"}
+        ]
+    }
+    
+    try:
+        response = requests.post(url, headers=headers, json=payload)
+        
+        if response.status_code == 200:
+            states = ""
+            rateLimit = response.headers.get("anthropic-ratelimit-input-tokens-limit")
+            if rateLimit:
+                tokens = int(rateLimit)
+                if tokens == 40000:
+                    states = "Tier1"
+                elif tokens == 80000:
+                    states = "Tier2"
+                elif tokens == 160000:
+                    states = "Tier3"
+                elif tokens == 400000:
+                    states = "Tier4"
+                else:
+                    states = "Unknown"
+            
+            return {
+                "success": True,
+                "return_message": "200: Success!",
+                "balance": 0,
+                "states": states
+            }
+        else:
+            return_message = ""
+            try:
+                error_data = response.json()
+                if "error" in error_data:
+                    return_message = error_data["error"].get("message", "")
+                    return_message = f"{response.status_code}: {return_message}"
+            except:
+                return_message = f"{response.status_code}: {response.text or f'HTTP错误'}"
+                
+            return {
+                "success": False,
+                "return_message": return_message,
+                "balance": 0,
+                "states": ""
+            }
+    except Exception as e:
+        error_str = str(e)
+        print(f"验证API密钥时出错: {error_str}")
+        return {
+            "success": False,
+            "return_message": f"500: 请求异常: {error_str}",
+            "balance": 0,
+            "states": ""
+        }

core/api/deepseek.py

@@ -0,0 +1,110 @@
+"""
+Deepseek API相关功能模块 - 提供Deepseek API的调用和验证功能
+"""
+import requests
+
+def validate_api_key(api_key):
+    """
+    验证Deepseek API密钥是否有效并获取账户余额
+    
+    Args:
+        api_key (str): 需要验证的Deepseek API密钥
+    
+    Returns:
+        dict: 包含验证结果的字典，包括:
+            - success (bool): API密钥验证成功为True，失败为False
+            - return_message (str): 验证响应的具体消息，成功时为"200: Success!"，失败时为错误详情
+            - balance (float): 账户余额，成功时为账户总余额，失败时为0
+            - states (str): 账户状态
+    """
+    url = "https://api.deepseek.com/v1/user/balance"
+    
+    headers = {
+        "Authorization": f"Bearer {api_key}"
+    }
+    
+    try:
+        response = requests.get(url, headers=headers)
+        
+        if response.status_code == 200:
+            response_data = response.json()
+            is_available = response_data.get("is_available", False)
+            
+            if not is_available:
+                return {
+                    "success": False,
+                    "return_message": "429: Insufficient balance",
+                    "balance": 0,
+                    "states": ""
+                }
+             
+            balance_infos = response_data.get("balance_infos", [])
+            
+            balance = 0.0
+            
+            try:
+                exchange_rate_response = requests.get("https://api.exchangerate-api.com/v4/latest/USD")
+                exchange_rate_response.raise_for_status()
+                exchange_rate_data = exchange_rate_response.json()
+                exchange_rate = exchange_rate_data.get("rates", {}).get("CNY")
+                
+                if exchange_rate is None:
+                    exchange_rate = 7.2
+            except Exception as e:
+                exchange_rate = 7.2
+            
+            has_cny = False
+            has_usd = False
+            
+            for balance_info in balance_infos:
+                currency = balance_info.get("currency")
+                total_balance = float(balance_info.get("total_balance", 0))
+                
+                if currency == "CNY":
+                    balance += total_balance
+                    if total_balance > 0:
+                        has_cny = True
+                elif currency == "USD":
+                    balance += total_balance * exchange_rate
+                    if total_balance > 0:
+                        has_usd = True
+            
+            states = ""
+            if has_cny and has_usd:
+                states = "CNY/USD"
+            elif has_cny:
+                states = "CNY"
+            elif has_usd:
+                states = "USD"
+            
+            return {
+                "success": True,
+                "return_message": "200: Success!",
+                "balance": balance,
+                "states": states
+            }
+        else:
+            return_message = ""
+            try:
+                error_data = response.json()
+                if "error" in error_data:
+                    return_message = error_data["error"].get("message", "")
+                    return_message = f"{response.status_code}: {return_message}"
+            except:
+                return_message = f"{response.status_code}: {response.text or f'HTTP错误'}"
+                
+            return {
+                "success": False,
+                "return_message": return_message,
+                "balance": 0,
+                "states": ""
+            }
+    except Exception as e:
+        error_str = str(e)
+        print(f"验证API密钥时出错: {error_str}")
+        return {
+            "success": False,
+            "return_message": f"500: 请求异常: {error_str}",
+            "balance": 0,
+            "states": ""
+        }

core/api/google.py

@@ -0,0 +1,113 @@
+"""
+Google API相关功能模块 - 提供Google API的调用和验证功能
+"""
+import requests
+
+def validate_api_key(api_key):
+    """
+    验证Google API密钥是否有效
+    
+    Args:
+        api_key (str): 需要验证的Google API密钥
+    
+    Returns:
+        dict: 包含验证结果的字典，包括:
+            - success (bool): API密钥验证成功为True，失败为False
+            - return_message (str): 验证响应的具体消息，成功时为空字符串，失败时为错误详情
+            - balance (float): 账户余额，固定为0
+            - states (str): 账户状态，付费版或免费版
+    """
+    url = "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-pro:generateContent"
+    
+    params = {
+        "key": api_key
+    }
+    
+    payload = {
+        "contents": [
+            {
+                "parts": [{"text": "Hello"}]
+            }
+        ],
+        "generationConfig": {
+            "maxOutputTokens": 1 
+        }
+    }
+    
+    try:
+        response = requests.post(url, params=params, json=payload)
+        
+        if response.status_code == 200:
+            states = check_paid_account(api_key)
+            
+            return {
+                "success": True,
+                "return_message": "200: Success!",
+                "balance": 0,
+                "states": states
+            }
+        else:
+            return_message = ""
+            try:
+                error_data = response.json()
+                if "error" in error_data:
+                    return_message = error_data["error"].get("message", "")
+                    if "details" in error_data["error"]:
+                        for detail in error_data["error"]["details"]:
+                            if detail.get("errorMessage"):
+                                return_message += f" {detail['errorMessage']}"
+                    return_message = f"{response.status_code}: {return_message}"
+            except:
+                return_message = f"{response.status_code}: {response.text or f'HTTP错误'}"
+                
+            return {
+                "success": False,
+                "return_message": return_message,
+                "balance": 0,
+                "states": ""
+            }
+    except Exception as e:
+        error_str = str(e)
+        print(f"验证API密钥时出错: {error_str}")
+        return {
+            "success": False,
+            "return_message": f"500: 请求异常: {error_str}",
+            "balance": 0,
+            "states": ""
+        }
+
+def check_paid_account(api_key):
+    """
+    检查Google API密钥是否为付费账户
+    
+    Args:
+        api_key (str): Google API密钥
+    
+    Returns:
+        str: 账户状态，付费版返回"Paid"，免费版返回"Free"
+    """
+    try:
+        imagen_url = "https://generativelanguage.googleapis.com/v1beta/models/imagen-3.0-generate-002:predict"
+        params = {"key": api_key}
+        payload = {
+            "instances": [{"prompt": "Hi"}]
+        }
+        
+        timeout = 1
+        
+        try:
+            response = requests.post(imagen_url, params=params, json=payload, timeout=timeout)
+            
+            if response.status_code == 200:
+                return "Paid"
+            else:
+                return "Free"
+                
+        except requests.exceptions.Timeout:
+            return "Paid"
+        except Exception:
+            return "Free"
+            
+    except Exception as e:
+        print(f"检查账户类型时出错: {str(e)}")
+        return "Free"

core/api/openai.py

@@ -0,0 +1,86 @@
+"""
+OpenAI API相关功能模块 - 提供OpenAI API的调用和验证功能
+"""
+import requests
+
+def validate_api_key(api_key):
+    """
+    验证OpenAI API密钥是否有效
+    
+    Args:
+        api_key (str): 需要验证的OpenAI API密钥
+    
+    Returns:
+        dict: 包含验证结果的字典，包括:
+            - success (bool): API密钥验证成功为True，失败为False
+            - return_message (str): 验证响应的具体消息，成功时为"200: Success!"，失败时为错误详情
+            - balance (float): 账户余额，固定为0
+            - states (str): 账户状态，对应的Tier等级
+    """
+    url = "https://api.openai.com/v1/chat/completions"
+    
+    headers = {
+        "Authorization": f"Bearer {api_key}",
+        "Content-Type": "application/json"
+    }
+    
+    payload = {
+        "model": "gpt-4o",
+        "max_tokens": 1,
+        "messages": [
+            {"role": "user", "content": "Hello"}
+        ]
+    }
+    
+    try:
+        response = requests.post(url, headers=headers, json=payload)
+        
+        if response.status_code == 200:
+            states = ""
+            rateLimit = response.headers.get("x-ratelimit-limit-tokens")
+            if rateLimit:
+                tokens = int(rateLimit)
+                if tokens == 30000:
+                    states = "Tier1"
+                elif tokens == 450000:
+                    states = "Tier2"
+                elif tokens == 800000:
+                    states = "Tier3"
+                elif tokens == 2000000:
+                    states = "Tier4"
+                elif tokens == 150000000:
+                    states = "Tier5"
+                else:
+                    states = "Unknown"
+            
+            return {
+                "success": True,
+                "return_message": "200: Success!",
+                "balance": 0,
+                "states": states
+            }
+        else:
+            return_message = ""
+            try:
+                error_data = response.json()
+                if "error" in error_data:
+                    return_message = error_data["error"].get("message", "")
+                    return_message = f"{response.status_code}: {return_message}"
+            except:
+                return_message = f"{response.status_code}: {response.text or f'HTTP错误'}"
+                
+            return {
+                "success": False,
+                "return_message": return_message,
+                "balance": 0,
+                "states": ""
+            }
+    except Exception as e:
+        error_str = str(e)
+        print(f"验证API密钥时出错: {error_str}")
+        return {
+            "success": False,
+            "return_message": f"500: 请求异常: {error_str}",
+            "balance": 0,
+            "states": ""
+        }

templates/components/api_key_list.html

@@ -13,7 +13,7 @@
 >
     <!-- 精美的选中数量显示 -->
     <div class="flex items-center space-x-3">
-        <div class="flex items-center justify-center h-8 w-8 rounded-full bg-gradient-to-r from-blue-400 to-indigo-500 text-white font-semibold text-sm shadow-sm">
+        <div class="flex items-center justify-center h-8 min-w-8 px-2 rounded-full bg-gradient-to-r from-blue-400 to-indigo-500 text-white font-semibold text-sm shadow-sm">
             <span x-text="selectedKeys.length"></span>
         </div>
         
@@ -120,19 +120,25 @@
                 </div>
             </div>
             
-            <h2 class="text-lg font-medium flex items-center"
+<h2 class="text-lg font-medium flex items-center"
                     :style="{
                         color: '{{ platform.id }}' in getPlatformStyles() ? getPlatformStyle('{{ platform.id }}').color : '#111827'
                     }"
                 >
+                    <!-- 添加平台图标 -->
+                    <img src="/static/img/{% if platform.id == 'anthropic' %}claude.svg{% elif platform.id == 'google' %}gemini.svg{% else %}{{ platform.id }}.svg{% endif %}" class="w-5 h-5 mr-2" alt="{{ platform.name }} 图标">
                     <span>{{ platform.name }}</span>
-                        <span 
+<span 
                         x-show="getPlatformKeyCount('{{ platform.id }}') > 0"
                         x-text="getPlatformKeyCount('{{ platform.id }}')"
-                        class="ml-2 platform-title-counter rounded-full"
+                        class="ml-2 platform-title-counter"
                         :style="{
                             backgroundColor: '{{ platform.id }}' in getPlatformStyles() ? getPlatformStyle('{{ platform.id }}')['background-color'] : 'rgba(224, 242, 254, 1)',
-                            color: '{{ platform.id }}' in getPlatformStyles() ? getPlatformStyle('{{ platform.id }}').color : '#0369a1'
+                            color: '{{ platform.id }}' in getPlatformStyles() ? getPlatformStyle('{{ platform.id }}').color : '#0369a1',
+                            borderRadius: '12px',
+                            minWidth: '1.6rem',
+                            padding: '0 0.4rem',
+                            width: 'auto'
                         }"
                     ></span>
                 </h2>

templates/components/toolbar.html

@@ -41,6 +41,7 @@
                 class="tag-counter" 
                 :class="allPlatformsSelected ? 'bg-primary-700 text-white' : 'bg-gray-200 text-gray-700'"
                 x-text="totalKeyCount()"
+                style="border-radius: 12px; min-width: 1.6rem; padding: 0 0.4rem; width: auto;"
             ></span>
         </button>
         
@@ -55,13 +56,26 @@
                 }"
                 :style="platformFilters[platform.id] ? getPlatformStyle(platform.id) : {}"
             >
+                <!-- 添加平台图标 -->
+                <img :src="'/static/img/' + (platform.id === 'anthropic' ? 'claude.svg' : platform.id === 'google' ? 'gemini.svg' : platform.id + '.svg')" class="w-4 h-4 mr-1.5" :alt="platform.name + ' 图标'">
                 <span x-text="platform.name"></span>
                 <span 
                     class="tag-counter" 
                     :class="{
                         'bg-gray-200 text-gray-700': !platformFilters[platform.id]
                     }"
-                    :style="platformFilters[platform.id] ? {background: 'rgba(' + getPlatformStyle(platform.id).color.replace('#', '').match(/.{2}/g).map(hex => parseInt(hex, 16)).join(',') + ',0.1)'} : {}"
+                    :style="platformFilters[platform.id] ? {
+                        background: 'rgba(' + getPlatformStyle(platform.id).color.replace('#', '').match(/.{2}/g).map(hex => parseInt(hex, 16)).join(',') + ',0.1)',
+                        borderRadius: '12px',
+                        minWidth: '1.6rem',
+                        padding: '0 0.4rem',
+                        width: 'auto'
+                    } : {
+                        borderRadius: '12px',
+                        minWidth: '1.6rem',
+                        padding: '0 0.4rem',
+                        width: 'auto'
+                    }"
                     x-text="getPlatformKeyCount(platform.id)"
                 ></span>
             </button>