The user settings are now stored in BrowserState, so the data survives restarting the server, and survives reloading the page.

app.py

@@ -7,7 +7,7 @@ TEST-COMMIT: to make hugging face spaces rebuild. Again.
 if __name__ == "__main__":
     # from src.huggingface_spaces.print_gradio_info import print_gradio_info
     # print_gradio_info()
-    # from src.plan.app_text2plan import run_app_text2plan
-    # run_app_text2plan()
-    from src.huggingface_spaces.app_state4 import demo
-    demo.launch()
+    from src.plan.app_text2plan import run_app_text2plan
+    run_app_text2plan()
+    # from src.huggingface_spaces.app_state4 import demo
+    # demo.launch()

src/plan/app_text2plan.py

@@ -12,6 +12,7 @@ import time
 import sys
 import threading
 import logging
+import json
 from dataclasses import dataclass
 from math import ceil
 from src.llm_factory import get_available_llms
@@ -134,9 +135,6 @@ class SessionState:
     """
     In a multi-user environment (e.g. Hugging Face Spaces), this class hold each users state.
     In a single-user environment, this class is used to hold the state of that lonely user.
-
-    IDEA: Persist the user settings for longer. The settings survive a page refresh, but not a server restart.
-    The browser has a local storage. Can Gradio access that, so that the settings are remembered between sessions?
     """
     def __init__(self):
         # Settings: the user's OpenRouter API key.
@@ -162,12 +160,48 @@ class SessionState:
         """
         return self
 
-def run_planner(submit_or_retry_button, plan_prompt, session_state: SessionState):
+def initialize_browser_settings(browser_state, session_state: SessionState):
+    try:
+        settings = json.loads(browser_state) if browser_state else {}
+    except Exception:
+        settings = {}
+    openrouter_api_key = settings.get("openrouter_api_key_text", "")
+    model = settings.get("model_radio", available_model_names[0])
+    speedvsdetail = settings.get("speedvsdetail_radio", SpeedVsDetailEnum.ALL_DETAILS_BUT_SLOW)
+    session_state.openrouter_api_key = openrouter_api_key
+    session_state.llm_model = model
+    session_state.speedvsdetail = speedvsdetail
+    return openrouter_api_key, model, speedvsdetail, browser_state, session_state
+
+def update_browser_settings_callback(openrouter_api_key, model, speedvsdetail, browser_state, session_state: SessionState):
+    try:
+        settings = json.loads(browser_state) if browser_state else {}
+    except Exception:
+        settings = {}
+    settings["openrouter_api_key_text"] = openrouter_api_key
+    settings["model_radio"] = model
+    settings["speedvsdetail_radio"] = speedvsdetail
+    updated_browser_state = json.dumps(settings)
+    session_state.openrouter_api_key = openrouter_api_key
+    session_state.llm_model = model
+    session_state.speedvsdetail = speedvsdetail
+    return updated_browser_state, openrouter_api_key, model, speedvsdetail, session_state
+
+def run_planner(submit_or_retry_button, plan_prompt, browser_state, session_state: SessionState):
     """
     Generator function for launching the pipeline process and streaming updates.
     The session state is carried in a SessionState instance.
     """
 
+    # Sync persistent settings from BrowserState into session_state
+    try:
+        settings = json.loads(browser_state) if browser_state else {}
+    except Exception:
+        settings = {}
+    session_state.openrouter_api_key = settings.get("openrouter_api_key_text", session_state.openrouter_api_key)
+    session_state.llm_model = settings.get("model_radio", session_state.llm_model)
+    session_state.speedvsdetail = settings.get("speedvsdetail_radio", session_state.speedvsdetail)
+
     # Clear any previous stop signal.
     session_state.stop_event.clear()
 
@@ -360,28 +394,6 @@ def open_output_dir(session_state: SessionState):
     except Exception as e:
         return f"Failed to open directory: {e}", session_state
 
-def update_openrouter_api_key(openrouter_api_key, session_state: SessionState):
-    """Updates the OpenRouter API key in the session state."""
-    session_state.openrouter_api_key = openrouter_api_key
-    return openrouter_api_key, session_state
-
-def update_model_radio(llm_model, session_state: SessionState):
-    """Updates the llm_model in the session state"""
-    session_state.llm_model = llm_model
-    return llm_model, session_state
-
-def update_speedvsdetail_radio(speedvsdetail, session_state: SessionState):
-    """Updates the speedvsdetail in the session state"""
-    session_state.speedvsdetail = speedvsdetail
-    return speedvsdetail, session_state
-
-def initialize_settings(session_state: SessionState):
-    """Initializes the settings from session_state, if available."""
-    return (session_state.openrouter_api_key,
-            session_state.llm_model,
-            session_state.speedvsdetail,
-            session_state)
-
 def check_api_key(session_state: SessionState):
     """Checks if the API key is provided and returns a warning if not."""
     if CONFIG.visible_openrouter_api_key_textbox and (not session_state.openrouter_api_key or len(session_state.openrouter_api_key) == 0):
@@ -451,23 +463,24 @@ with gr.Blocks(title="PlanExe") as demo_text2plan:
     
     # Manage the state of the current user
     session_state = gr.State(SessionState())
+    browser_state = gr.BrowserState("", storage_key="Test", secret="Test")
 
     # Submit and Retry buttons call run_planner and update the state.
     submit_btn.click(
         fn=run_planner,
-        inputs=[submit_btn, prompt_input, session_state],
+        inputs=[submit_btn, prompt_input, browser_state, session_state],
         outputs=[output_markdown, download_output, session_state]
     ).then(
-        fn=check_api_key,  # Check after submitting.
+        fn=check_api_key,
         inputs=[session_state],
         outputs=[api_key_warning]
     )
     retry_btn.click(
         fn=run_planner,
-        inputs=[retry_btn, prompt_input, session_state],
+        inputs=[retry_btn, prompt_input, browser_state, session_state],
         outputs=[output_markdown, download_output, session_state]
     ).then(
-        fn=check_api_key,  # Check after retrying.
+        fn=check_api_key,
         inputs=[session_state],
         outputs=[api_key_warning]
     )
@@ -476,6 +489,10 @@ with gr.Blocks(title="PlanExe") as demo_text2plan:
         fn=stop_planner,
         inputs=session_state,
         outputs=[status_markdown, session_state]
+    ).then(
+        fn=check_api_key,
+        inputs=[session_state],
+        outputs=[api_key_warning]
     )
     # Open Output Dir button.
     open_dir_btn.click(
@@ -484,33 +501,44 @@ with gr.Blocks(title="PlanExe") as demo_text2plan:
         outputs=[status_markdown, session_state]
     )
 
+    # Unified change callbacks for settings.
     openrouter_api_key_text.change(
-        fn=update_openrouter_api_key,
-        inputs=[openrouter_api_key_text, session_state],
-        outputs=[openrouter_api_key_text, session_state]
+        fn=update_browser_settings_callback,
+        inputs=[openrouter_api_key_text, model_radio, speedvsdetail_radio, browser_state, session_state],
+        outputs=[browser_state, openrouter_api_key_text, model_radio, speedvsdetail_radio, session_state]
     ).then(
-        fn=check_api_key,  # Check when the API key changes.
+        fn=check_api_key,
         inputs=[session_state],
         outputs=[api_key_warning]
     )
+
     model_radio.change(
-        fn=update_model_radio,
-        inputs=[model_radio, session_state],
-        outputs=[model_radio, session_state]
+        fn=update_browser_settings_callback,
+        inputs=[openrouter_api_key_text, model_radio, speedvsdetail_radio, browser_state, session_state],
+        outputs=[browser_state, openrouter_api_key_text, model_radio, speedvsdetail_radio, session_state]
+    ).then(
+        fn=check_api_key,
+        inputs=[session_state],
+        outputs=[api_key_warning]
     )
+
     speedvsdetail_radio.change(
-        fn=update_speedvsdetail_radio,
-        inputs=[speedvsdetail_radio, session_state],
-        outputs=[speedvsdetail_radio, session_state]
+        fn=update_browser_settings_callback,
+        inputs=[openrouter_api_key_text, model_radio, speedvsdetail_radio, browser_state, session_state],
+        outputs=[browser_state, openrouter_api_key_text, model_radio, speedvsdetail_radio, session_state]
+    ).then(
+        fn=check_api_key,
+        inputs=[session_state],
+        outputs=[api_key_warning]
     )
 
-
+    # Initialize settings on load from persistent browser_state.
     demo_text2plan.load(
-        fn=initialize_settings,
-        inputs=[session_state],
-        outputs=[openrouter_api_key_text, model_radio, speedvsdetail_radio, session_state]
+        fn=initialize_browser_settings,
+        inputs=[browser_state, session_state],
+        outputs=[openrouter_api_key_text, model_radio, speedvsdetail_radio, browser_state, session_state]
     ).then(
-        fn=check_api_key,  # Check on initial load.
+        fn=check_api_key,
         inputs=[session_state],
         outputs=[api_key_warning]
     )