Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import anthropic
-from api_usage import get_subscription, check_key_availability, check_key_ant_availability, check_key_gemini_availability, check_key_azure_availability, get_azure_status, get_azure_deploy, check_key_mistral_availability, check_mistral_quota
+from api_usage import get_subscription, check_key_availability, check_key_ant_availability, 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
 
 def sort_key(key):
     _key = key.strip()
@@ -9,7 +9,9 @@ def sort_key(key):
     elif _key.startswith("sk-"):    
         return get_key_oai_info(_key)        
     elif _key.startswith("AIzaSy"):
-        return get_key_gemini_info(_key)        
+        return get_key_gemini_info(_key)
+    elif (_key.startswith("r8_") and len(_key) == 40) or (_key.islower() and len(_key) == 40):
+        return get_key_replicate_info(_key)
     elif len(_key.split(':')) == 2 and _key.split(':')[1].islower() and len(_key.split(':')[1]) == 32 and "openai.azure.com" not in _key.split(':')[1]:
         endpoint = f"{_key.split(':')[0]}.openai.azure.com"
         api_key = _key.split(':')[1]
@@ -19,9 +21,9 @@ def sort_key(key):
         api_key = _key.split(';')[1]
         return get_key_azure_info(endpoint, api_key)        
     elif _key.startswith("AKIA") and len(_key.split(':')[0]) == 20 and _key.split(':')[0].isupper():
-        return get_key_aws_info(_key)        
+        return get_key_aws_info(_key)
     elif len(_key) == 32:
-        return get_key_mistral_info(_key)        
+        return get_key_mistral_info(_key)    
     else:
         return not_supported(_key)
 
@@ -115,10 +117,38 @@ def get_key_mistral_info(key):
         info_dict['has_quota'] = check_mistral_quota(key)
     return info_dict
 
-def get_key_aws_info(key):
+def get_key_replicate_info(key):
+    key_avai = check_key_replicate_availability(key)
+    info_dict = {#"account_name": "",
+                 "key_type": "Replicate",
+                 "key_availability": key_avai[0],
+                 "account_name": "",
+                 "type": "",
+                 "hardware_available": ""}
+    if key_avai[0]:
+        info_dict['account_name'] = key_avai[1]['username']
+        info_dict['type'] = key_avai[1]['type']
+        info_dict['hardware_available'] = key_avai[2]
+    return info_dict
+
+def get_key_aws_info(key):    
+    key_avai = check_key_aws_availability(key)
     info_dict = {#"account_name": "",
                  "key_type": "Amazon AWS Claude",
-                 "status": " Soon™ "}
+                 "key_availability": key_avai[0],
+                 "username": "",
+                 "root": "",
+                 "admin": "",
+                 "quarantine": "",
+                 "claude_enabled_region": key_avai[5],
+                 "cost_and_usage": ""}
+    if key_avai[0]:
+        info_dict['username'] = key_avai[1]
+        info_dict['root'] = key_avai[2]
+        info_dict['admin'] = key_avai[3]
+        info_dict['quarantine'] = key_avai[4] 
+        info_dict['claude_enabled_region'] = key_avai[5]
+        info_dict['cost_and_usage'] = key_avai[6]
     return info_dict
   
 def not_supported(key):
@@ -132,11 +162,13 @@ def clear_inputs(text):
 
 with gr.Blocks() as demo:
     gr.Markdown('''
-    # OpenAI/Anthropic/Google Gemini/Microsoft Azure/Mistral AI API Key Status Checker
+    # OpenAI/Anthropic/Gemini/Azure/Mistral/Replicate/AWS Claude API Key Status Checker
     
     *(Based on shaocongma, CncAnon1 and Drago 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    
+    Azure endpoint's format: YOUR_RESOURCE_NAME:YOUR_API_KEY or (https://)YOUR_RESOURCE_NAME.openai.azure.com;YOUR_API_KEY
     ''')
 
     with gr.Row():
@@ -146,7 +178,7 @@ with gr.Blocks() as demo:
                 clear_button = gr.Button("Clear")
                 submit_button = gr.Button("Submit", variant="primary")
         with gr.Column():
-            info = gr.JSON(label="OpenAI/Anthropic/Gemini/Azure/Mistral API Key Information")
+            info = gr.JSON(label="API Key Information")
 
     clear_button.click(fn=clear_inputs, inputs=[key], outputs=[key])
     submit_button.click(fn=sort_key, inputs=[key], outputs=[info], api_name="sort_key")