Update streamlit_app.py

streamlit_app.py

@@ -54,6 +54,9 @@ st.session_state.setdefault("preferred_model", "gemini-2.0-flash-lite")
 st.session_state.setdefault("compression_threshold_mb", 200)
 st.session_state.setdefault("model_select", st.session_state.get("preferred_model"))
 st.session_state.setdefault("custom_model", "")
+st.session_state.setdefault("url_input", "")
+st.session_state.setdefault("video_password", "")
+st.session_state.setdefault("analysis_prompt", "")
 
 MODEL_OPTIONS = [
     "gemini-2.5-flash",
@@ -191,6 +194,7 @@ def wait_for_processed(file_obj, timeout: int = None, progress_callback=None):
         backoff = min(backoff * 2, 8.0)
 
 def _normalize_genai_response(response):
+    import json
     if response is None:
         return ""
     try:
@@ -372,29 +376,40 @@ st.sidebar.header("Video Input")
 st.sidebar.text_input("Video URL", key="url_input", placeholder="https://", value=st.session_state.get("url_input", ""))
 
 settings_exp = st.sidebar.expander("Settings", expanded=False)
-st.session_state["model_select"] = settings_exp.selectbox("Gemini model", MODEL_OPTIONS, index=MODEL_OPTIONS.index(st.session_state.get("model_select", "gemini-2.0-flash-lite")), key="model_select")
-if st.session_state.get("model_select") == "custom":
-    st.session_state["custom_model"] = settings_exp.text_input("Custom model name", value=st.session_state.get("custom_model", ""), key="custom_model")
-model_input_value = (st.session_state.get("custom_model") or st.session_state.get("model_select")).strip()
 
-st.session_state["api_key"] = settings_exp.text_input("Google API Key", key="api_key_input", value=st.session_state.get("api_key", ""), type="password")
+# compute initial index safely
+initial_index = 0
+try:
+    cur = st.session_state.get("model_select", st.session_state.get("preferred_model"))
+    initial_index = MODEL_OPTIONS.index(cur) if cur in MODEL_OPTIONS else MODEL_OPTIONS.index(st.session_state.get("preferred_model"))
+except Exception:
+    initial_index = 0
+
+# let the widget own st.session_state["model_select"]; do not assign into it later
+with settings_exp:
+    st.selectbox("Gemini model", MODEL_OPTIONS, index=initial_index, key="model_select")
+    if st.session_state.get("model_select") == "custom":
+        st.text_input("Custom model name", value=st.session_state.get("custom_model", ""), key="custom_model")
+    model_input_value = (st.session_state.get("custom_model") or st.session_state.get("model_select")).strip()
+
+    st.text_input("Google API Key", key="api_key_input", value=st.session_state.get("api_key", ""), type="password")
 
-default_prompt = (
-    "Watch the video and provide a detailed behavioral report focusing on human actions, interactions, posture, movement, and apparent intent. Keep language professional. Include a list of observations for notable events."
-)
-st.session_state["analysis_prompt"] = settings_exp.text_area("Enter analysis prompt", value=st.session_state.get("analysis_prompt", default_prompt), height=140, key="analysis_prompt")
+    default_prompt = (
+        "Watch the video and provide a detailed behavioral report focusing on human actions, interactions, posture, movement, and apparent intent. Keep language professional. Include a list of observations for notable events."
+    )
+    st.text_area("Enter analysis prompt", value=st.session_state.get("analysis_prompt", default_prompt), height=140, key="analysis_prompt")
 
-st.session_state["video_password"] = settings_exp.text_input("Video Password (if needed)", key="video_password_input", placeholder="password", type="password")
+    st.text_input("Video Password (if needed)", key="video_password_input", placeholder="password", type="password")
 
-st.session_state["processing_timeout"] = settings_exp.number_input("Processing timeout (s)", min_value=60, max_value=3600, value=st.session_state.get("processing_timeout", 900), step=30, key="processing_timeout_input")
-st.session_state["generation_timeout"] = settings_exp.number_input("Generation timeout (s)", min_value=30, max_value=1800, value=st.session_state.get("generation_timeout", 300), step=10, key="generation_timeout_input")
-st.session_state["compression_threshold_mb"] = settings_exp.number_input("Compression threshold (MB)", min_value=10, max_value=2000, value=st.session_state.get("compression_threshold_mb", 200), step=10, key="compression_threshold_input")
+    st.number_input("Processing timeout (s)", min_value=60, max_value=3600, value=st.session_state.get("processing_timeout", 900), step=30, key="processing_timeout_input")
+    st.number_input("Generation timeout (s)", min_value=30, max_value=1800, value=st.session_state.get("generation_timeout", 300), step=10, key="generation_timeout_input")
+    st.number_input("Compression threshold (MB)", min_value=10, max_value=2000, value=st.session_state.get("compression_threshold_mb", 200), step=10, key="compression_threshold_input")
 
-key_source = "session" if st.session_state.get("api_key") else ".env" if os.getenv("GOOGLE_API_KEY") else "none"
-settings_exp.caption(f"Using API key from: {key_source}")
+    key_source = "session" if st.session_state.get("api_key") else ".env" if os.getenv("GOOGLE_API_KEY") else "none"
+    st.caption(f"Using API key from: {key_source}")
 
-if not get_effective_api_key():
-    settings_exp.warning("No Google API key provided; upload/generation disabled.", icon="⚠️")
+    if not get_effective_api_key():
+        st.warning("No Google API key provided; upload/generation disabled.", icon="⚠️")
 
 col1, col2 = st.columns([1, 3])
 with col1: