catch a error

Author: hanhou

code/util/plot_autotrain_manager.py

@@ -158,6 +158,9 @@ def plot_manager_all_progress_bokeh_source(
         df_subject["y"] = len(subject_ids) - n
         df_subjects.append(df_subject)
 
+    if not df_subjects:
+        return None, None
+    
     data_df = pd.concat(df_subjects, ignore_index=True)
 
     return data_df, subject_ids
@@ -188,6 +191,10 @@ def plot_manager_all_progress_bokeh(
         if_use_filtered_data=if_use_filtered_data,
         filtered_session_ids=filtered_session_ids,
     )
+    
+    if data_df is None:
+        return None, None
+    
     source = ColumnDataSource(data_df)
 
     # Add hover tool

code/util/streamlit.py

@@ -839,7 +839,6 @@ def add_auto_train_manager():
         highlight_subjects = []
 
     # --- Bokeh ---
-
     fig_auto_train, data_df = plot_manager_all_progress_bokeh(
         x_axis=x_axis,
         recent_days=recent_months * 30.437,  # Turn months into days
@@ -856,6 +855,10 @@ def add_auto_train_manager():
             else None
         ),
     )
+    
+    if fig_auto_train is None:
+        st.markdown("### In the filtered sessions, no AutoTrain history to show!")
+        return
 
     event_result = streamlit_bokeh3_events(
         events="TestSelectEvent",