Update app.py

app.py

@@ -24,7 +24,7 @@ for k, v in defaults.items():
 with st.sidebar:
     st.header("🔐 API Configuration")
 
-    # -------- BEFORE VALIDATION --------
+    # ---------- BEFORE VALIDATION ----------
     if not st.session_state.api_validated:
         st.text_input(
             "Enter Google API Key",
@@ -37,35 +37,40 @@ with st.sidebar:
                 st.warning("Please enter an API key.")
             else:
                 try:
+                    # 1️⃣ Initialize client
                     client = genai.Client(
                         api_key=st.session_state.api_key_input
                     )
 
+                    # 2️⃣ REAL validation call (this is the key fix)
+                    test_response = client.models.generate_content(
+                        model="gemini-2.5-flash",
+                        contents="Say OK"
+                    )
+
+                    # 3️⃣ Only after success, list models
                     models = [
                         m.name for m in client.models.list()
                         if "gemini" in m.name.lower()
                     ]
 
-                    if not models:
-                        st.error("API key valid, but no Gemini models available.")
-                    else:
-                        # ✅ Persist validated state
-                        st.session_state.api_key = st.session_state.api_key_input
-                        st.session_state.client = client
-                        st.session_state.models = models
-                        st.session_state.selected_model = models[0]
-                        st.session_state.api_validated = True
+                    # 4️⃣ Persist validated state
+                    st.session_state.api_key = st.session_state.api_key_input
+                    st.session_state.client = client
+                    st.session_state.models = models
+                    st.session_state.selected_model = models[0]
+                    st.session_state.api_validated = True
 
-                        # 🔒 Clear visible input immediately
-                        st.session_state.api_key_input = ""
+                    # 🔒 Clear visible input
+                    st.session_state.api_key_input = ""
 
-                        st.success("API key validated")
+                    st.success("✅ API key validated successfully")
 
                 except Exception as e:
-                    st.error("Invalid API key")
+                    st.error("❌ Invalid or restricted API key")
                     st.caption(str(e))
 
-    # -------- AFTER VALIDATION --------
+    # ---------- AFTER VALIDATION ----------
     else:
         st.success("🔒 API Key validated (hidden)")