Update app.py

app.py

@@ -4,24 +4,36 @@ import requests
 def check_key_gemini_availability(key):
     """
     Checks the availability of a Gemini API key.
-
     Args:
         key: The API key to check.
-
     Returns:
         A tuple containing a boolean indicating whether the key is valid and an error message (or None if the key is valid).
     """
     try:
+        # First, check if the key allows access to the list of models
         url_getListModel = f"https://generativelanguage.googleapis.com/v1beta/models?key={key}"
         rq = requests.get(url_getListModel)
         rq.raise_for_status()  # Raise HTTPError for bad responses (4xx or 5xx)
         result = rq.json()
-        if 'models' in result:
-            return True, None
+        if 'models' not in result:
+            return False, f"Invalid key: 'models' field not found in response from list models endpoint."
+
+        # Second, attempt to generate content to further validate the key
+        url_generateContent = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent?key={key}"
+        # headers = {'Content-Type': 'application/json'}
+        data = {"contents": [{"parts": [{"text": "Say \"this is a test\""}]}]}
+        rq = requests.post(url_generateContent, headers=headers, json=data) #data=json.dumps(data)
+
+        if 400 <= rq.status_code < 500:
+            return False, f"Invalid key: Generate content returned HTTP {rq.status_code}"
+        elif 500 <= rq.status_code < 600:
+             return True, None # consider 5xx as valid, could be temporary issue.
         else:
-            return False, f"Invalid key: 'models' field not found in response."
+            rq.raise_for_status() # Raise other errors for other status codes
+            return True, None
+
     except requests.exceptions.HTTPError as e:
-        return False, f"Invalid key: HTTP {e.response.status_code}"
+        return False, f"Invalid key: HTTP {e.response.status_code} - {e}"
     except Exception as e:
         return False, f"Error key: {e}"