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,43 +37,39 @@ with st.sidebar:
                 st.warning("Please enter an API key.")
             else:
                 try:
-                    # ✅ CORRECT: Configure SDK globally
-                    genai.configure(
-                        api_key=st.session_state.api_key_input
-                    )
-
-                    # Create client AFTER configure
-                    client = genai.Client()
-
-                    # ✅ Minimal real inference validation
-                    client.models.generate_content(
-                        model="gemini-2.5-flash",
-                        contents="OK"
-                    )
-
-                    # Fetch available models
-                    models = [
-                        m.name for m in client.models.list()
-                        if "gemini" in m.name.lower()
-                    ]
+                    # Initialize client
+                    client = genai.Client(api_key=st.session_state.api_key_input)
+
+                    # Attempt to list models (authentication test)
+                    models = []
+                    try:
+                        models = [
+                            m.name for m in client.models.list()
+                            if "gemini" in m.name.lower()
+                        ]
+                    except Exception:
+                        # Listing models may fail even if key is valid
+                        models = []
 
                     # 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.models = models or ["gemini-2.5-flash"]
+                    st.session_state.selected_model = st.session_state.models[0]
                     st.session_state.api_validated = True
-
-                    # 🔒 Clear visible input
                     st.session_state.api_key_input = ""
 
                     st.success("✅ API key validated successfully")
 
                 except Exception as e:
-                    st.error("❌ Invalid or restricted API key")
-                    st.caption(str(e))
-
-    # -------- AFTER VALIDATION --------
+                    error_msg = str(e).lower()
+                    if "401" in error_msg or "403" in error_msg or "unauthorized" in error_msg:
+                        st.error("❌ Invalid API key")
+                    else:
+                        st.warning("⚠️ API key accepted, but model access may be restricted")
+                        st.caption(str(e))
+
+    # ---------- AFTER VALIDATION ----------
     else:
         st.success("🔒 API Key validated (hidden)")