Update app.py

app.py

@@ -1,5 +1,4 @@
 import gradio as gr
-import anthropic
 import requests
 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
 
@@ -60,13 +59,15 @@ def get_key_oai_info(key):
 
 def get_key_ant_info(key):
     # Return a dictionary containing key information
-    ant = anthropic.Anthropic(api_key=key)
-    key_avai = check_key_ant_availability(ant)
+    key_avai = check_key_ant_availability(key)
     info_dict = {#"account_name": "",
                  "key_type": "Anthropic Claude",
                  "key_availability": key_avai[0],
                  "status": key_avai[1],
-                 "filter_response": key_avai[2]}
+                 "filter_response": key_avai[2],
+                 "requests_per_minute": key_avai[3] + ("" if key_avai[3] == "" else f" ({key_avai[4]} left)"),
+                 "tokens_per_minute": key_avai[5] + ("" if key_avai[5] == "" else f" ({key_avai[6]} left)"),
+                 "tier": key_avai[7]}
     return info_dict
 
 def get_key_gemini_info(key):
@@ -156,6 +157,7 @@ def get_key_aws_info(key):
         info_dict['admin'] = key_avai[3]
         info_dict['quarantine'] = key_avai[4] 
         info_dict['iam_full_access'] = key_avai[5]
+        info_dict['billing'] = key_avai[8]
         info_dict['claude_v2_enabled_region'] = key_avai[6]
         info_dict['claude_sonnet_enabled_region'] = key_avai[7]
         info_dict['cost_and_usage'] = key_avai[8]