Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import requests, re
-from api_usage import get_subscription, check_key_availability, check_key_ant_availability, check_ant_rate_limit, check_key_gemini_availability, check_key_azure_availability, get_azure_status, get_azure_deploy, check_key_mistral_availability, check_mistral_quota, check_key_replicate_availability, check_key_aws_availability, check_key_or_availability, check_key_or_limits, check_gcp_anthropic
+from api_usage import get_subscription, check_key_availability, check_key_ant_availability, check_ant_rate_limit, check_key_gemini_availability, check_key_azure_availability, get_azure_status, get_azure_deploy, check_key_mistral_availability, check_mistral_quota, check_key_replicate_availability, check_key_aws_availability, check_key_or_availability, check_key_or_limits, check_gcp_anthropic, check_groq_status, check_nai_status, check_elevenlabs_status
 
 async def sort_key(key, rate_limit, claude_opus):
     _key = key.strip()
@@ -16,6 +16,9 @@ async def sort_key(key, rate_limit, claude_opus):
         
     if _key.startswith("AIzaSy"):
         return get_key_gemini_info(_key)
+    
+    if _key.startswith("pst-"):
+        return get_key_nai_info(_key)
         
     if (_key.startswith("r8_") and len(_key) == 40) or (_key.islower() and len(_key) == 40):
         return get_key_replicate_info(_key)
@@ -33,11 +36,20 @@ async def sort_key(key, rate_limit, claude_opus):
     if _key.startswith("AKIA") and len(_key.split(':')[0]) == 20 and _key.split(':')[0].isupper():
         return await get_key_aws_info(_key)
     
-    if len(_key) == 32:
+    if re.match(re.compile(r"[a-f0-9]{32}"), _key) or re.match(re.compile(r"sk_[a-f0-9]{48}"), _key):
+        return get_key_elevenlabs_info(_key)
+    
+    if re.match(re.compile(r"[a-zA-Z0-9]{32}"), _key):
         return get_key_mistral_info(_key)
         
-    if re.match(re.compile(r"[\w\-]+:[\w\-@\.]+:.+\\n"), _key):
-        return await get_key_gcp_info(_key)
+    if re.match(re.compile(r"gsk_[a-zA-Z0-9]{20}WGdyb3FY[a-zA-Z0-9]{24}"), _key):
+        return get_key_groq_info(_key)
+        
+    if re.match(re.compile(r"[\w\-]+:[\w\-@\.]+:[\w-]+:.+"), _key): # 0: refresh token
+        return await get_key_gcp_info(_key, 0)
+        
+    if re.match(re.compile(r"[\w\-]+:[\w\-@\.]+:.+\\n"), _key): # 1: service account
+        return await get_key_gcp_info(_key, 1)
         
     return not_supported(_key)
 
@@ -106,7 +118,8 @@ def get_key_gemini_info(key):
     info_dict = {#"account_name": "",
                  "key_type": "Google Gemini",
                  "key_availability": key_avai[0],
-                 "models": key_avai[1]}
+                 "status": key_avai[1],
+                 "models": key_avai[2]}
     return info_dict
 
 def get_key_azure_info(endpoint, api_key):
@@ -117,6 +130,7 @@ def get_key_azure_info(endpoint, api_key):
              "gpt35_availability": "",
              "gpt4_availability": "",
              "gpt4_32k_availability": "",             
+             "dall_e_3_availability": "",             
              "moderation_status": "",
              "models": "",
              "deployments": ""}
@@ -126,6 +140,7 @@ def get_key_azure_info(endpoint, api_key):
         info_dict["gpt35_availability"] = status[1]
         info_dict["gpt4_availability"] = status[2]
         info_dict["gpt4_32k_availability"] = status[3]
+        info_dict["dall_e_3_availability"] = status[4]
         info_dict["moderation_status"] = status[0]
         info_dict["models"] = key_avai[1]
         info_dict["deployments"] = azure_deploy
@@ -216,8 +231,8 @@ def get_key_openrouter_info(key):
         info_dict['usage'] = key_avai[1]
     return info_dict
 
-async def get_key_gcp_info(key):    
-    key_avai = await check_gcp_anthropic(key)
+async def get_key_gcp_info(key, type):    
+    key_avai = await check_gcp_anthropic(key, type)
     info_dict = {#"account_name": "",
                  "key_type": "Vertex AI (GCP)",
                  "key_availability": key_avai[0],
@@ -228,6 +243,31 @@ async def get_key_gcp_info(key):
     else:
         info_dict['status'] = key_avai[1]
     return info_dict
+
+def get_key_groq_info(key):
+    key_avai = check_groq_status(key)
+    info_dict = {#"account_name": "",
+                 "key_type": "Groq",
+                 "key_availability": True if key_avai else False,
+                 "models": key_avai if key_avai else ""}
+    return info_dict
+
+def get_key_nai_info(key):
+    key_avai = check_nai_status(key)
+    info_dict = {#"account_name": "",
+                 "key_type": "NovelAI",
+                 "key_availability": True if key_avai[0] else False,
+                 "user_info": key_avai[1] if key_avai[0] else ""}
+    return info_dict
+    
+def get_key_elevenlabs_info(key):
+    key_avai = check_elevenlabs_status(key)
+    info_dict = {#"account_name": "",
+                 "key_type": "ElevenLabs",
+                 "key_availability": key_avai[0],
+                 "user_info": key_avai[1],
+                 "voices_info": key_avai[2]}
+    return info_dict
     
 def not_supported(key):
     info_dict = {#"account_name": "",
@@ -240,7 +280,7 @@ def clear_inputs(text):
 
 with gr.Blocks() as demo:
     gr.Markdown('''
-    # OpenAI/Anthropic/Gemini/Azure/Mistral/Replicate/AWS Claude/OpenRouter/Vertex AI(GCP Anthropic) API Key Status Checker
+    # OpenAI/Anthropic/Gemini/Azure/Mistral/Replicate/AWS Claude/OpenRouter/Vertex AI(GCP Anthropic)/Groq/NovelAI/ElevenLabs API Key Status Checker
     
     *(Based on shaocongma, CncAnon1, su, Drago, kingbased key checkers)*
 
@@ -248,7 +288,8 @@ with gr.Blocks() as demo:
     
     Azure endpoint's format: YOUR_RESOURCE_NAME:YOUR_API_KEY or (https://)YOUR_RESOURCE_NAME.openai.azure.com;YOUR_API_KEY
     
-    GCP format: PROJECT_ID:CLIENT_EMAIL:PRIVATE_KEY (copy from service account json)
+    GCP format: PROJECT_ID:CLIENT_EMAIL:PRIVATE_KEY (including \\n)
+     | or refresh token: PROJECT_ID:CLIENT_ID:CLIENT_SECRET:REFRESH_TOKEN
     ''')
 
     with gr.Row():