Upload app.py

app.py

@@ -499,18 +499,60 @@ Chat with the assistant to brainstorm your course idea. Use the panel on the lef
 
     # Helper to refresh the sidebar session list. Returns an update for session_picker.
     def refresh_sessions():
+        """Refresh the list of chat sessions for the sidebar.
+
+        Each chat will be displayed by its title (or "Untitled" if none) followed by a short
+        portion of its key in parentheses. The actual radio choice value will be this
+        formatted string so that we can extract the chat key later on when the user
+        selects a session. The currently active chat will be pre-selected.
+
+        Returns:
+            A gradio update for the ``session_picker`` radio component.
+        """
         chats = list_chats()
-        choices = [c["key"] for c in chats]
-        # Display titles or truncated key if no title
-        labels = [c["title"] if c["title"] else c["key"][:8] for c in chats]
-        # gr.Radio uses choices for values and may display them as-is; set label implicitly via choices
-        return gr.update(choices=choices, value=state_chat_key.value)
+        choices = []
+        current_label = None
+        current_key = state_chat_key.value
+        for c in chats:
+            title = c.get("title") or "Untitled"
+            key = c.get("key")
+            # Build a display label that includes a short portion of the key so it
+            # remains unique even if multiple chats share the same title.
+            label = f"{title} ({key[:8]})"
+            choices.append(label)
+            if key == current_key:
+                current_label = label
+        # If the current chat isn't found (e.g. after deletion), select the first chat
+        if not current_label and choices:
+            current_label = choices[0]
+        return gr.update(choices=choices, value=current_label)
 
     # Load the selected chat into memory and return display pairs/history
-    def load_session(selected_key):
-        if not selected_key:
+    def load_session(selected_value):
+        """Load a chat session when selected in the sidebar.
+
+        The session_picker returns a formatted label that includes the chat key in
+        parentheses. Extract the chat key from this label and then load the
+        conversation messages from the database. The chat history and pair lists
+        are reconstructed for display in the Chatbot, while sources, plan and
+        resource_cache are reset.
+
+        Args:
+            selected_value: The radio selection value from session_picker, which is
+                a string like ``"Title (abcdef12)"``. If None, returns empty state.
+
+        Returns:
+            Tuple of (chatbot_pairs, chat_history, chat_pairs, sources, plan, resource_cache).
+        """
+        if not selected_value:
             return [], [], [], [], "", {}
-        msgs = load_chat(selected_key)
+        # Extract the chat key from the label. It is enclosed in parentheses at the end.
+        key = selected_value
+        if "(" in selected_value and selected_value.endswith(")"):
+            key = selected_value[selected_value.rfind("(") + 1 : -1]
+        # Update the global state for the current chat key
+        state_chat_key.value = key
+        msgs = load_chat(key)
         history = []
         pairs = []
         buffer = []