Update state.py

state.py

@@ -504,6 +504,7 @@ def load_demo_single(demo_path):
         _remap_demo_labels(dname)
     with st.spinner("Analyzing Demo Data"):
         analyze(dname)
+    _setup_demo_roles_population(dname)
     st.success(f"Took {time.time() - start_time:.1f}s to analyze the demo file!")
     st.session_state.select_currFile = dname
     return dname
@@ -614,6 +615,10 @@ def run_analysis_loop(file_names, file_paths_dict):
 
         with st.spinner(f"Analyzing {i+1}/{totalFiles}"):
             analyze(fname)
+        # Auto-assign roles and population for demo files
+        plain = fname.rsplit(".", 1)[0].lower()
+        if plain.startswith("class") or fname.startswith("audioSamples/class"):
+            _setup_demo_roles_population(fname)
 
     # Return to cpu
     pipeline.toCPU()