Deploy SmartPaper R3-Free UI State Fix

managed_auth.py

@@ -29,6 +29,34 @@ from smartpaper_restore_bridge import edge_bridge
 EDGE_ASSERTION_COOKIE_NAME = "smartpaper_edge_assertion"
 EDGE_ASSERTION_AUDIENCE = "smartpaper-streamlit"
 ASSERTION_CLOCK_SKEW_SECONDS = 30
+ACTIVE_GENERATION_STATUSES = frozenset({"created", "running", "resumable", "cancelling"})
+KNOWN_GENERATION_STATUSES = ACTIVE_GENERATION_STATUSES | frozenset({"completed", "failed", "cancelled"})
+
+
+def _select_generation_task_for_session(
+    active_task: Any, latest_task: Any, tracked_task_id: str
+) -> dict[str, Any] | None:
+    """Choose only a genuinely current task for the Streamlit session.
+
+    ``latest_task`` is historical data.  It may update the terminal state of a
+    task that this Streamlit session already tracked, but it must never create a
+    current-task UI state on a fresh login/session by itself.
+    """
+    if isinstance(active_task, dict) and active_task.get("task_id"):
+        status = str(active_task.get("status") or "").strip().lower()
+        if status in ACTIVE_GENERATION_STATUSES:
+            return dict(active_task)
+
+    tracked = str(tracked_task_id or "").strip()
+    if not tracked or not isinstance(latest_task, dict):
+        return None
+    latest_id = str(latest_task.get("task_id") or "").strip()
+    latest_status = str(latest_task.get("status") or "").strip().lower()
+    if latest_id == tracked and latest_status in KNOWN_GENERATION_STATUSES:
+        return dict(latest_task)
+    return None
+
+
 def _runtime_value(name: str) -> str:
     value = os.environ.get(name, "")
     if not value:
@@ -129,7 +157,7 @@ def _verify_edge_assertion(token: str) -> dict[str, Any]:
 
 def _clear_generation_session_state() -> None:
     for key in (
-        "sp_generation_task", "sp_current_node",
+        "sp_generation_task", "sp_tracked_task_id", "sp_current_node",
         "sp_generation_idempotency_key", "zip_buffer", "zip_filename",
         "is_running", "start_pipeline", "resume_pipeline", "run_mode",
         "rag_engine", "up_amj_ref", "up_amj_data", "up_asq_ref",
@@ -241,10 +269,12 @@ def get_account_snapshot(*, block: bool = False) -> dict[str, Any] | None:
         st.session_state["sp_user"] = dict(user)
     task = result.get("task")
     latest_task = result.get("latest_task")
-    if isinstance(task, dict) and task.get("task_id"):
-        st.session_state["sp_generation_task"] = dict(task)
-    elif isinstance(latest_task, dict) and latest_task.get("task_id"):
-        st.session_state["sp_generation_task"] = dict(latest_task)
+    selected_task = _select_generation_task_for_session(
+        task, latest_task, str(st.session_state.get("sp_tracked_task_id") or "")
+    )
+    if selected_task is not None:
+        st.session_state["sp_generation_task"] = selected_task
+        st.session_state["sp_tracked_task_id"] = str(selected_task.get("task_id") or "")
     else:
         st.session_state["sp_generation_task"] = None
     return result
@@ -270,6 +300,9 @@ def start_generation_task(*, idempotency_key: str, mode: str, interface_language
     if not result.get("ok"):
         _raise_edge_error(result, "task_start_failed")
     _merge_quota_into_session(result)
+    started_task = result.get("task")
+    if isinstance(started_task, dict) and started_task.get("task_id"):
+        st.session_state["sp_tracked_task_id"] = str(started_task["task_id"])
     return dict(result)
 
 
@@ -297,6 +330,8 @@ def generation_task_status(task_id: str) -> dict[str, Any]:
     task = result.get("task")
     if isinstance(task, dict):
         st.session_state["sp_generation_task"] = dict(task)
+        if task.get("task_id"):
+            st.session_state["sp_tracked_task_id"] = str(task["task_id"])
     return dict(result)