Update app.py

app.py

@@ -20,7 +20,8 @@ def check_key_gemini_availability(key, ai_model):
 
         # Second, attempt to generate content to further validate the key
         ai_model_ = ai_model.strip() or "gemini-1.5-flash"
-        if ai_model_ not in result['models']:
+        models = [ x["name"].split('/',1)[1] for x in result['models'] ]
+        if ai_model_ not in models:
             return False, f"Specified model {ai_model_} is not in available 'models'."
         url_generateContent = f"https://generativelanguage.googleapis.com/v1beta/models/{ai_model_}:generateContent?key={key}"
         # headers = {'Content-Type': 'application/json'}