Update app.py

app.py

@@ -1,5 +1,6 @@
 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
 
 def sort_key(key):
@@ -29,7 +30,8 @@ def sort_key(key):
 
 def get_key_oai_info(key):
     # Return a dictionary containing key information
-    key_avai = check_key_availability(key)
+    session = requests.Session()
+    key_avai = check_key_availability(session, key)
     info_dict = {#"account_name": "",
                  "key_type": "OpenAI",
                  "key_availability": True if key_avai else False,
@@ -44,7 +46,7 @@ def get_key_oai_info(key):
                  #"tokens_per_minute_left": "",
                  "quota": ""}
     if key_avai:
-        info = get_subscription(key, key_avai)
+        info = get_subscription(key, session, key_avai)
         info_dict["gpt4_availability"] = info["has_gpt4"]
         info_dict["gpt4_32k_availability"] = info["has_gpt4_32k"]
         info_dict["default_org"] = info["default_org"]