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,24 +37,27 @@ with st.sidebar:
                 st.warning("Please enter an API key.")
             else:
                 try:
-                    # 1️⃣ Initialize client
-                    client = genai.Client(
+                    # ✅ CORRECT: Configure SDK globally
+                    genai.configure(
                         api_key=st.session_state.api_key_input
                     )
 
-                    # 2️⃣ REAL validation call (this is the key fix)
-                    test_response = client.models.generate_content(
+                    # Create client AFTER configure
+                    client = genai.Client()
+
+                    # ✅ Minimal real inference validation
+                    client.models.generate_content(
                         model="gemini-2.5-flash",
-                        contents="Say OK"
+                        contents="OK"
                     )
 
-                    # 3️⃣ Only after success, list models
+                    # Fetch available models
                     models = [
                         m.name for m in client.models.list()
                         if "gemini" in m.name.lower()
                     ]
 
-                    # 4️⃣ Persist validated state
+                    # Persist validated state
                     st.session_state.api_key = st.session_state.api_key_input
                     st.session_state.client = client
                     st.session_state.models = models
@@ -70,7 +73,7 @@ with st.sidebar:
                     st.error("❌ Invalid or restricted API key")
                     st.caption(str(e))
 
-    # ---------- AFTER VALIDATION ----------
+    # -------- AFTER VALIDATION --------
     else:
         st.success("🔒 API Key validated (hidden)")