Update app.py

app.py

@@ -60,9 +60,6 @@ with st.sidebar:
 
                         st.success("✅ API key validated successfully")
 
-                        # 🔁 Force rerun to show model selector immediately
-                        st.experimental_rerun()
-
                     except Exception as e:
                         error_msg = str(e).lower()
                         if "401" in error_msg or "403" in error_msg or "unauthorized" in error_msg:
@@ -72,7 +69,7 @@ with st.sidebar:
                             st.caption(str(e))
 
     # ---------- AFTER VALIDATION ----------
-    else:
+    if st.session_state.api_validated:
         st.success("🔒 API Key validated (hidden)")
 
         st.selectbox(