Update app.py

app.py

@@ -1,5 +1,4 @@
 import gradio as gr
-import openai
 import anthropic
 from api_usage import get_subscription, check_key_availability, check_key_ant_availability
 
@@ -12,23 +11,29 @@ def sort_key(key):
 
 def get_key_oai_info(key):
     # Return a dictionary containing key information
-    openai.api_key = key
-    key_avai = check_key_availability()
+    key_avai = check_key_availability(key)
     info_dict = {"account_name": "",
                  "key_availability": True if key_avai else False,
                  "gpt4_availability": "",
                  "gpt4_32k_availability": "",
+                 "default_org": "",                 
+                 "org_description": "",
+                 "organization": "",
+                 "models": "",
                  "requests_per_minute": "",
                  "tokens_per_minute": "",
-                 "organization": "",
+                 #"tokens_per_minute_left": "",
                  "quota": ""}
     if key_avai:
-        info = get_subscription(key, key_avai)
+        info = get_subscription(key)
         info_dict["gpt4_availability"] = info["has_gpt4"]
         info_dict["gpt4_32k_availability"] = info["has_gpt4_32k"]
+        info_dict["default_org"] = info["default_org"]        
+        info_dict["org_description"] = info["org_description"]
+        info_dict["organization"] = info["organization"]      
+        info_dict["models"] = info["models"]
         info_dict["requests_per_minute"] = info["rpm"]
         info_dict["tokens_per_minute"] = info["tpm"]
-        info_dict["organization"] = info["organization"]
         info_dict["quota"] = info["quota"]
     return info_dict