Update app.py

app.py

@@ -10,8 +10,8 @@ def sort_key(key):
         return get_key_oai_info(_key)        
     elif _key.startswith("AIzaSy"):
         return get_key_gemini_info(_key)        
-    elif len(_key.split(':')) == 2 and _key.split(':')[1].islower() and len(_key.split(':')[1]) == 32:
-        endpoint = _key.split(':')[0]
+    elif len(_key.split(':')) == 2 and _key.split(':')[1].islower() and len(_key.split(':')[1]) == 32 and "openai.azure.com" not in _key.split(':')[1]:
+        endpoint = f"{_key.split(':')[0]}.openai.azure.com"
         api_key = _key.split(':')[1]
         return get_key_azure_info(endpoint, api_key)        
     elif "openai.azure.com" in _key.split(';')[0]:
@@ -25,7 +25,6 @@ def sort_key(key):
     else:
         return not_supported(_key)
 
-
 def get_key_oai_info(key):
     # Return a dictionary containing key information
     key_avai = check_key_availability(key)
@@ -91,18 +90,18 @@ def get_key_azure_info(endpoint, api_key):
              "gpt4_availability": "",
              "gpt4_turbo_availability": "",
              "gpt4_32k_availability": "",             
-             "moderation_response": "",
-             "deployments": "",
-             "models": ""}
+             "moderation_status": "",
+             "models": "",
+             "deployments": ""}
     if key_avai[0]:        
         azure_deploy = get_azure_deploy(endpoint, api_key)
         status = get_azure_status(endpoint, api_key, azure_deploy)
-        info_dict["gpt35_availability"] = status[1],
-        info_dict["gpt4_availability"] = status[2],
-        info_dict["gpt4_turbo_availability"] = status[3],
-        info_dict["gpt4_32k_availability"] = status[4],
-        info_dict["moderation_response"] = status[0],
-        info_dict["models"] = key_avai[1],
+        info_dict["gpt35_availability"] = status[1]
+        info_dict["gpt4_availability"] = status[2]
+        info_dict["gpt4_turbo_availability"] = status[3]
+        info_dict["gpt4_32k_availability"] = status[4]
+        info_dict["moderation_status"] = status[0]
+        info_dict["models"] = key_avai[1]
         info_dict["deployments"] = azure_deploy
     return info_dict
 
@@ -139,7 +138,10 @@ with gr.Blocks() as demo:
     
     *(Based on shaocongma, CncAnon1 and Drago key checkers)*
     
-    (Azure endpoint's format: (https://)YOUR_RESOURCE_NAME.openai.azure.com;YOUR_API_KEY)
+    *Azure endpoint's format:*
+    *(https://)YOUR_RESOURCE_NAME.openai.azure.com;YOUR_API_KEY*
+    *YOUR_RESOURCE_NAME:YOUR_API_KEY*
+    
     ''')
 
     with gr.Row():