Update state.py

state.py

@@ -565,6 +565,36 @@ def load_demo_multi(demo_paths):
     st.session_state.analyzeAllToggle = True
 
 
+def _setup_demo_roles_population(fname):
+    """Assign SPEAKER_001 to Instructor, all others to Students, and set
+    a random population in [24, 28] for the given demo file."""
+    if fname not in st.session_state.results:
+        return
+    annotation, _ = st.session_state.results[fname]
+    labels = list(annotation.labels())
+
+    instructor_token = f"{fname}: SPEAKER_001"
+    student_tokens   = [f"{fname}: {sp}" for sp in labels if sp != "SPEAKER_001"]
+
+    while len(st.session_state.categorySelect) < 2:
+        st.session_state.categorySelect.append([])
+
+    existing_instructor = set(st.session_state.categorySelect[0])
+    existing_students   = set(st.session_state.categorySelect[1])
+
+    if instructor_token not in existing_instructor:
+        st.session_state.categorySelect[0].append(instructor_token)
+    for token in student_tokens:
+        if token not in existing_students and token not in existing_instructor:
+            st.session_state.categorySelect[1].append(token)
+
+    all_assigned = {t for tokens in st.session_state.categorySelect for t in tokens}
+    st.session_state.unusedSpeakers[fname] = [
+        sp for sp in labels if f"{fname}: {sp}" not in all_assigned
+    ]
+    st.session_state.studentPopulations[fname] = random.randint(24, 28)
+
+
 def run_analysis_loop(file_names, file_paths_dict):
     """Process only new (not yet analyzed) files and populate session state."""
     import time