Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import openai
-from api_usage import check_gpt4_availability, check_gpt4_32k_availability, get_subscription, check_key_availability #, get_usage
+from api_usage import get_subscription, check_key_availability #, get_usage, check_gpt4_availability, check_gpt4_32k_availability, 
 
 def get_key_info(key):
     # Return a dictionary containing key information
@@ -11,17 +11,16 @@ def get_key_info(key):
                  "gpt4_availability": "",
                  "gpt4_32k_availability": "",
                  "requests_per_minute": "",
+                 "tokens_per_minute": "",
                  "organization": "",
                  "quota": ""}
     if key_avai:
         info = get_subscription(key)
-#        used = get_usage(key)
-        gpt4_avai = check_gpt4_availability()
-        gpt4_32k_avai = check_gpt4_32k_availability()
 #        info_dict["account_name"] = info["account_name"]
-        info_dict["gpt4_availability"] = gpt4_avai
-        info_dict["gpt4_32k_availability"] = gpt4_32k_avai
-        info_dict["requests_per_minute"] = info["rpm"] + " (gpt4)" if gpt4_avai else info["rpm"] + " (turbo)"
+        info_dict["gpt4_availability"] = info["has_gpt4"]
+        info_dict["gpt4_32k_availability"] = info["has_gpt4_32k"]
+        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