Update app.py

app.py

@@ -24,7 +24,6 @@ for k, v in defaults.items():
 with st.sidebar:
     st.header("🔐 API Configuration")
 
-    # ---------- BEFORE VALIDATION ----------
     if not st.session_state.api_validated:
         with st.form("api_form", clear_on_submit=False):
             api_key_input = st.text_input(
@@ -39,10 +38,8 @@ with st.sidebar:
                     st.warning("Please enter an API key.")
                 else:
                     try:
-                        # Initialize client
                         client = genai.Client(api_key=api_key_input)
 
-                        # Attempt to list models (authentication test)
                         try:
                             models = [
                                 m.name for m in client.models.list()
@@ -51,7 +48,6 @@ with st.sidebar:
                         except Exception:
                             models = []
 
-                        # Persist validated state
                         st.session_state.api_key = api_key_input
                         st.session_state.client = client
                         st.session_state.models = models or ["gemini-2.5-flash"]
@@ -68,8 +64,7 @@ with st.sidebar:
                             st.warning("⚠️ API key accepted, but model access may be restricted")
                             st.caption(str(e))
 
-    # ---------- AFTER VALIDATION ----------
-    if st.session_state.api_validated:
+    else:
         st.success("🔒 API Key validated (hidden)")
 
         st.selectbox(
@@ -78,11 +73,9 @@ with st.sidebar:
             key="selected_model"
         )
 
-        # Allow changing API key
         if st.button("🔄 Change API Key"):
             for k in defaults:
                 st.session_state[k] = defaults[k]
-
 # ---------------- Chat History ----------------
 for msg in st.session_state.messages:
     with st.chat_message(msg["role"]):