.

Author: HernandoMV

behavior_data_visualizer/main.py

@@ -145,7 +145,7 @@ def update_session_dropdown(selected_value):
         if selected_value is None:
             return []
         df = mouse_data_dict[selected_value]
-        session_data_dict = utils.get_dicctionary_of_sessions(df)
+        session_data_dict = utils.get_dicctionary_of_dates(df)
         return [{'label': key, 'value': session_data_dict[key]} for key in session_data_dict.keys()]
 
     # Figure for the session summary
@@ -158,7 +158,7 @@ def update_session_summary(mouse, session):
         if mouse is None or session is None:
             return ''
         df = mouse_data_dict[mouse]
-        sdf = df[df["session"] == session]
+        sdf = df[df["year_month_day"] == session]
         fig = fm.session_summary_figure(sdf, mouse, perf_window=100)
         return utils.fig_to_uri(fig)
 

behavior_data_visualizer/utils.py

@@ -25,6 +25,13 @@ def get_dicctionary_of_sessions(df):
         sessions_dict[key] = session
     return sessions_dict
 
+def get_diccionary_of_dates(df):
+    dates_dict = {}
+    for date in df['year_month_day'].unique():
+        key = str(date)
+        dates_dict[key] = date
+    return dates_dict
+
 def display_click_data(clickData, mouse_name):
     try:
         date = clickData['points'][0]['customdata'][0]