Update app.py

app.py

@@ -1,14 +1,14 @@
 import gradio as gr
-import requests
+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
 
 async def sort_key(key, rate_limit, claude_opus):
     _key = key.strip()
     
-    if _key.startswith("sk-or-v1-"):    
+    if _key.startswith("sk-or-v1-") and re.match(re.compile("sk-or-v1-[a-z0-9]{64}"), _key):    
         return get_key_openrouter_info(_key)
         
-    if _key.startswith("sk-ant-"):
+    if _key.startswith("sk-ant-") and len(_key) == 109:
         return await get_key_ant_info(_key, rate_limit, claude_opus)
         
     if _key.startswith("sk-"):    
@@ -31,7 +31,7 @@ async def sort_key(key, rate_limit, claude_opus):
         return get_key_azure_info(endpoint, api_key)
         
     if _key.startswith("AKIA") and len(_key.split(':')[0]) == 20 and _key.split(':')[0].isupper():
-        return get_key_aws_info(_key)
+        return await get_key_aws_info(_key)
         
     if len(_key) == 32:
         return get_key_mistral_info(_key)
@@ -158,8 +158,8 @@ def get_key_replicate_info(key):
         info_dict['hardware_available'] = key_avai[3]
     return info_dict
 
-def get_key_aws_info(key):    
-    key_avai = check_key_aws_availability(key)
+async def get_key_aws_info(key):
+    key_avai = await check_key_aws_availability(key)
     info_dict = {#"account_name": "",
                  "key_type": "Amazon AWS Claude",
                  "key_availability": key_avai[0],
@@ -230,7 +230,7 @@ with gr.Blocks() as demo:
     gr.Markdown('''
     # OpenAI/Anthropic/Gemini/Azure/Mistral/Replicate/AWS Claude/OpenRouter API Key Status Checker
     
-    *(Based on shaocongma, CncAnon1, Drago and kingbased key checkers)*
+    *(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)