Update default category

src/streamlit_app.py

@@ -262,6 +262,18 @@ df_mod = load_private_tsv("imageaction__captcha@SPEC-1de6b70ae2f0.tsv")
 st.session_state.dataset = df_base
 st.session_state.dataset_modified = df_mod
 st.session_state.categories = sorted(df_base["answer_norm"].unique())
+
+# Session defaults
+if "provider" not in st.session_state:
+    st.session_state.provider = BaseAdapter.OPENAI   # default provider = OpenAI
+
+if "model" not in st.session_state:
+    st.session_state.model = "gpt-5-2025-08-07"      # default OpenAI model
+
+if "target_category" not in st.session_state:
+    st.session_state.target_category = "bus"    
+
+
 # Sidebar
 with st.sidebar:
     st.subheader("Challenge Settings")
@@ -271,7 +283,7 @@ with st.sidebar:
 
         DEFAULT_CAT = "bus"  # normalized label
 
-        if cats and DEFAULT_CAT in cats:
+        if cats and DEFAULT_CAT in st.session_state.categories:
             default_idx = cats.index(DEFAULT_CAT)
         else:
             default_idx = 0  # fallback
@@ -291,31 +303,49 @@ with st.sidebar:
     st.markdown("---")
     st.subheader("Solver")
 
-    # 1) Provider first (include Manual + all providers from your dict)
+    # 1) Provider first (Manual + all providers)
     provider_options = ["Manual"] + list(MODEL_PROVIDERS.keys())
-    try:
-        provider_idx = provider_options.index(st.session_state.provider)
-    except ValueError:
-        provider_idx = 0  # fallback to Manual if prior value is missing
 
-    st.session_state.provider = st.selectbox("Provider", provider_options, index=provider_idx)
+    # ensure current provider is valid; otherwise default to OpenAI
+    if st.session_state.provider not in provider_options:
+        st.session_state.provider = BaseAdapter.OPENAI
+
+    provider_idx = provider_options.index(st.session_state.provider)
+
+    st.session_state.provider = st.selectbox(
+        "Provider",
+        provider_options,
+        index=provider_idx,
+    )
 
-    # 2) Model (enabled only when provider != Manual)
+    # 2) Model: only when provider != Manual
     if st.session_state.provider == "Manual":
         st.session_state.model = None
         st.selectbox("Model", ["(not required in Manual mode)"], index=0, disabled=True)
         st.caption("Manual mode: click tiles to select. No model needed.")
     else:
         models_for_provider = MODEL_PROVIDERS.get(st.session_state.provider, [])
-        # Keep previously selected model if still valid; otherwise default to first/empty
+
+        # if provider is OpenAI and our default gpt-5 is in the list, prefer that
+        if st.session_state.provider == BaseAdapter.OPENAI and "gpt-5-2025-08-07" in models_for_provider:
+            if st.session_state.model not in models_for_provider:
+                st.session_state.model = "gpt-5-2025-08-07"
+        else:
+            # generic fallback for other providers
+            if st.session_state.model not in models_for_provider and models_for_provider:
+                st.session_state.model = models_for_provider[0]
+
         if not models_for_provider:
             st.session_state.model = None
-            st.selectbox("Model", ["(no models available for this provider)"], index=0, disabled=True)
+            st.selectbox("Model", ["(no models for this provider)"], index=0, disabled=True)
         else:
-            if st.session_state.model not in models_for_provider:
-                st.session_state.model = models_for_provider[0]
             model_idx = models_for_provider.index(st.session_state.model)
-            st.session_state.model = st.selectbox("Model", models_for_provider, index=model_idx)
+            st.session_state.model = st.selectbox(
+                "Model",
+                models_for_provider,
+                index=model_idx,
+            )
+
 
 
 # Generate new challenge