added slice part

app.py

@@ -108,11 +108,11 @@ with gr.Blocks(title='Cabasus') as cabasus_sensor:
         csv_file_box.change(process_data, inputs=[csv_file_box, slice_size_slider, sample_rate, window_size_slider], 
                             outputs=[processed_file_box, json_file_box, slices_per_leg, plot_box_leg, plot_box_overlay, slice_slider, plot_slice_leg, get_all_slice, slice_json_box])
         leg_dropdown.change(plot_sensor_data_from_json, inputs=[json_file_box, leg_dropdown, slice_slider], 
-                            outputs=[plot_box_leg, plot_slice_leg, get_all_slice, slice_json_box])
+                            outputs=[plot_box_leg, plot_slice_leg, get_all_slice, slice_json_box, plot_box_overlay])
         repeat_process.click(process_data, inputs=[csv_file_box, slice_size_slider, sample_rate, window_size_slider], 
                              outputs=[processed_file_box, json_file_box, slices_per_leg, plot_box_leg, plot_box_overlay, slice_slider, plot_slice_leg, get_all_slice, slice_json_box])
         slice_slider.change(plot_sensor_data_from_json, inputs=[json_file_box, leg_dropdown, slice_slider], 
-                            outputs=[plot_box_leg, plot_slice_leg, get_all_slice, slice_json_box])
+                            outputs=[plot_box_leg, plot_slice_leg, get_all_slice, slice_json_box, plot_box_overlay])
         
         som_create.click(get_som_mp4, inputs=[json_file_box, slice_slider], outputs=[som_figures])
         button_label_Add.click(attach_label_to_json, inputs=[slice_json_box, label_name], outputs=[slice_json_label_box])

funcs/plot_func.py

@@ -62,17 +62,15 @@ def plot_sensor_data_from_json(json_file, sensor, slice_select=1):
     plt.tight_layout()
 
     fig2, file = plot_other_sensor_with_same_timestamp(json_file, sensor, slice_select)
+    fig3 = plot_overlay_data_from_json(json_file, slice_select=slice_select)
 
-    return fig, fig1, fig2, file
+    return fig, fig1, fig2, file, fig3
 
-def plot_overlay_data_from_json(json_file, sensors, use_precise_timestamp=False):
+def plot_overlay_data_from_json(json_file, slice_select, sensors=['GZ1', 'GZ2', 'GZ3', 'GZ4']):
     # Read the JSON file
     with open(json_file, "r") as f:
         slices = json.load(f)
 
-    # Set up the colormap
-    cmap = plt.get_cmap('viridis')
-
     # Create subplots for each sensor
     fig, axs = plt.subplots(len(sensors), 1, figsize=(12, 2 * len(sensors)), sharex=True)
 
@@ -86,9 +84,11 @@ def plot_overlay_data_from_json(json_file, sensors, use_precise_timestamp=False)
             sensor_data = slice_dict[sensor]
 
             data = pd.DataFrame({sensor: sensor_data}, index=slice_timestamps)
-            color = cmap(slice_idx / len(slices))
 
-            axs[idx].plot(data[sensor], color=color, label=f'Slice {slice_idx + 1}')
+            if slice_idx+1 == slice_select:
+                axs[idx].plot(data[sensor], '-r', label=f'Slice {slice_idx + 1}')
+            else:
+                axs[idx].plot(data[sensor], '-c')
 
         axs[idx].set_ylabel(sensor)
 
@@ -174,13 +174,6 @@ def plot_other_sensor_with_same_timestamp(json_file, sensor, slice_select):
     json_dict = {}
     for tkeys in total_keys:
         json_dict[tkeys] = slice_dict[tkeys]
-    # json_dict = {'data': sensor_data, 'sensor': slice_item, 'time': timestamps,
-    #              f"{sensor}_precise_time_diff": slice_dict[f"{sensor}_precise_time_diff"],
-    #              "precise_timestamp": slice_dict["precise_timestamp"],
-    #              "timestamp": slice_dict["timestamp"],
-    #              "time_diff": slice_dict["time_diff"],
-    #              "precise_time_diff": slice_dict["precise_time_diff"],
-    #              "label": 'unknown'}
     
     with open(f'slice_{slice_select}.json', "w") as f:
         json.dump(numpy_to_native(json_dict), f, indent=2)

funcs/processor.py

@@ -78,7 +78,7 @@ def process_data(input_file, slice_size=64, min_slice_size=16, sample_rate=20, w
     file, len_ = slice_csv_to_json('output.csv', slice_size, min_slice_size, sample_rate, window_size=window_size)
 
     # get the plot automatically
-    sensor_fig, slice_fig, get_all_slice, slice_json = plot_sensor_data_from_json(file, "GZ1")
-    overlay_fig = plot_overlay_data_from_json(file, ["GZ1", "GZ2", "GZ3", "GZ4"], use_precise_timestamp=True)
+    sensor_fig, slice_fig, get_all_slice, slice_json, overlay_fig = plot_sensor_data_from_json(file, "GZ1")
+    # overlay_fig = plot_overlay_data_from_json(file, ["GZ1", "GZ2", "GZ3", "GZ4"])
 
     return 'output.csv', file, f'num of slices found: {len_}', sensor_fig, overlay_fig, gr.Slider.update(interactive=True, maximum=len_, minimum=1, value=1), slice_fig, get_all_slice, slice_json