Update ui_components.py

ui_components.py

@@ -34,6 +34,10 @@ def render_sidebar() -> Dict:
     if "api_key" not in st.session_state:
         st.session_state.api_key = ""
 
+    # Initialize available models in session state
+    if "available_models" not in st.session_state:
+        st.session_state.available_models = AVAILABLE_MODELS
+
     api_key_input = st.sidebar.text_input(
         "API Key",
         value=st.session_state.api_key,
@@ -55,17 +59,21 @@ def render_sidebar() -> Dict:
     # Model and Parameters section
     st.sidebar.subheader("🤖 Model & Parameters")
 
+    # Use models from session state
+    available_models = st.session_state.available_models
+
     # Model selection
     model = st.sidebar.selectbox(
         "Model",
-        options=AVAILABLE_MODELS,
-        index=AVAILABLE_MODELS.index(DEFAULT_MODEL) if DEFAULT_MODEL in AVAILABLE_MODELS else 0,
+        options=available_models,
+        index=available_models.index(DEFAULT_MODEL) if DEFAULT_MODEL in available_models else 0,
         help="Select the OpenAI model to use"
     )
 
     # Refresh models button
     if st.sidebar.button("🔄 Refresh Models", help="Fetch available models from your API key"):
         st.session_state.refresh_models = True
+        st.rerun()
 
     # Temperature
     temperature = st.sidebar.slider(