Update app.py

app.py

@@ -151,7 +151,7 @@ def get_key_gemini_info(key):
 
 def get_key_azure_info(endpoint, api_key):
     key_avai = check_key_azure_availability(endpoint, api_key)
-    info_dict = {#"account_name": "",
+    info_dict = {
              "key_type": "Microsoft Azure OpenAI",
              "key_availability": key_avai[0],
              "gpt35_availability": "",
@@ -160,19 +160,42 @@ def get_key_azure_info(endpoint, api_key):
              "dall_e_3_availability": "",             
              "moderation_status": "",
              "models": "",
-             "deployments": ""}
+             "deployments": "",
+             "rate_limits": {}  # New field for per-deployment limits
+    }
+    
     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_32k_availability"] = status[3]
-        info_dict["dall_e_3_availability"] = status[4]
-        info_dict["moderation_status"] = status[0]
-        info_dict["models"] = key_avai[1]
-        info_dict["deployments"] = azure_deploy
+        
+        # Check rate limits for each deployment
+        for model, deployment in azure_deploy.items():
+            url = f"https://{endpoint}/openai/deployments/{deployment}/chat/completions?api-version=2024-02-15-preview"
+            headers = {"api-key": api_key}
+            response = requests.post(url, headers=headers, json={
+                "messages": [{"role": "user", "content": "."}],
+                "max_tokens": 1
+            })
+            
+            info_dict["rate_limits"][model] = {
+                "remaining_requests": response.headers.get("x-ratelimit-remaining-requests", ""),
+                "remaining_tokens": response.headers.get("x-ratelimit-remaining-tokens", ""),
+                "limit_tokens": response.headers.get("x-ratelimit-limit-tokens", "")
+            }
+
+        info_dict.update({
+            "gpt35_availability": status[1],
+            "gpt4_availability": status[2], 
+            "gpt4_32k_availability": status[3],
+            "dall_e_3_availability": status[4],
+            "moderation_status": status[0],
+            "models": key_avai[1],
+            "deployments": azure_deploy
+        })
     return info_dict
 
+
+
 def get_key_mistral_info(key):
     key_avai = check_key_mistral_availability(key)
     info_dict = {#"account_name": "",