Update app.py

app.py

@@ -26,19 +26,19 @@ with st.sidebar:
 
     # ---------- BEFORE VALIDATION ----------
     if not st.session_state.api_validated:
-        st.text_input(
+        api_key_input = st.text_input(
             "Enter Google API Key",
             type="password",
             key="api_key_input"
         )
 
         if st.button("Validate API Key"):
-            if not st.session_state.api_key_input:
+            if not api_key_input:
                 st.warning("Please enter an API key.")
             else:
                 try:
                     # Initialize client
-                    client = genai.Client(api_key=st.session_state.api_key_input)
+                    client = genai.Client(api_key=api_key_input)
 
                     # Attempt to list models (authentication test)
                     models = []
@@ -52,12 +52,11 @@ with st.sidebar:
                         models = []
 
                     # Persist validated state
-                    st.session_state.api_key = st.session_state.api_key_input
+                    st.session_state.api_key = api_key_input
                     st.session_state.client = client
                     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
-                    st.session_state.api_key_input = ""
 
                     st.success("✅ API key validated successfully")