Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import openai
-from api_usage import check_gpt4_availability, get_subscription, check_key_availability #, get_usage
+from api_usage import check_gpt4_availability, check_gpt4_32k_availability, get_subscription, check_key_availability #, get_usage
 
 def get_key_info(key):
     # Return a dictionary containing key information
@@ -9,22 +9,31 @@ def get_key_info(key):
     info_dict = {"account_name": "",
                  "key_availability": key_avai,
                  "gpt4_availability": "",
-                 "has_payment_method": "",
+                 "gpt4_32k_availability": "",
+                 "rpm": "",
+                 "organization": ""
+                 "quota": ""}
+#                 "has_payment_method": "",
 #                 "used": "",
-                 "plan": "",
-                 "soft_limit": "",
-                 "hard_limit": ""}
+#                 "plan": "",
+#                 "soft_limit": "",
+#                 "hard_limit": ""}
     if key_avai:
         info = get_subscription(key)
 #        used = get_usage(key)
         gpt4_avai = check_gpt4_availability()
-        info_dict["account_name"] = info["account_name"]
+        gpt4_32k_avai = check_gpt4_32k_availability()
+#        info_dict["account_name"] = info["account_name"]
         info_dict["gpt4_availability"] = gpt4_avai
-        info_dict["has_payment_method"] = info["has_payment_method"]
+        info_dict["gpt4_32k_availability"] = gpt4_32k_avai
+        info_dict["rpm"] = info["rpm"] + " (gpt4)" if gpt4_avai else info["rpm"] + " (turbo)"
+        info_dict["organization"] = info["organization"]
+        info_dict["quota"] = info["quota"]
+#        info_dict["has_payment_method"] = info["has_payment_method"]
 #        info_dict["used"] = used
-        info_dict["plan"] = info["plan"]
-        info_dict["soft_limit"] = info["soft_limit_usd"]
-        info_dict["hard_limit"] = info["hard_limit_usd"]
+#        info_dict["plan"] = info["plan"]
+#        info_dict["soft_limit"] = info["soft_limit_usd"]
+#        info_dict["hard_limit"] = info["hard_limit_usd"]
     return info_dict