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
+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
 
 async def sort_key(key, rate_limit, claude_opus):
     _key = key.strip()
@@ -32,10 +32,13 @@ 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:
         return get_key_mistral_info(_key)
         
+    if re.match(re.compile(r"[\w\-]+:[\w\-@\.]+:.+\\n"), _key):
+        return await get_key_gcp_info(_key)
+        
     return not_supported(_key)
 
 def get_key_oai_info(key):
@@ -212,7 +215,20 @@ def get_key_openrouter_info(key):
     else:
         info_dict['usage'] = key_avai[1]
     return info_dict
-  
+
+async def get_key_gcp_info(key):    
+    key_avai = await check_gcp_anthropic(key)
+    info_dict = {#"account_name": "",
+                 "key_type": "Vertex AI (GCP)",
+                 "key_availability": key_avai[0],
+                 "status": "",
+                 "enabled_region": ""}
+    if key_avai[0]:
+        info_dict['enabled_region'] = key_avai[2]
+    else:
+        info_dict['status'] = key_avai[1]
+    return info_dict
+    
 def not_supported(key):
     info_dict = {#"account_name": "",
                      "key_type": "Not supported",
@@ -224,19 +240,21 @@ def clear_inputs(text):
 
 with gr.Blocks() as demo:
     gr.Markdown('''
-    # OpenAI/Anthropic/Gemini/Azure/Mistral/Replicate/AWS Claude/OpenRouter API Key Status Checker
+    # OpenAI/Anthropic/Gemini/Azure/Mistral/Replicate/AWS Claude/OpenRouter/Vertex AI(GCP Anthropic) API Key Status Checker
     
     *(Based on shaocongma, CncAnon1, su, Drago, kingbased key checkers)*
 
     AWS credential's format: AWS_ACCESS_KEY_ID:AWS_SECRET_ACCESS_KEY (root might not be accurate)
     
     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)
     ''')
 
     with gr.Row():
         with gr.Column():
-            key = gr.Textbox(lines=1, max_lines=2, label="API Key")
-            rate_limit = gr.Checkbox(label="Check concurrent rate limit (API Claude) (experimental)")
+            key = gr.Textbox(lines=1, max_lines=5, label="API Key")
+            rate_limit = gr.Checkbox(label="Check concurrent rate limit (API Claude)")
             claude_opus = gr.Checkbox(label="Use claude-3-opus for filter_response check? (default: claude-3-haiku)")
             with gr.Row():
                 clear_button = gr.Button("Clear")
@@ -246,4 +264,4 @@ with gr.Blocks() as demo:
 
     clear_button.click(fn=clear_inputs, inputs=[key], outputs=[key])
     submit_button.click(fn=sort_key, inputs=[key, rate_limit, claude_opus], outputs=[info], api_name="sort_key")
-demo.launch()
+demo.launch()