Add button to save csv for differential echem (#1498)

Author: ml-evs

pydatalab/src/pydatalab/bokeh_plots.py

@@ -893,14 +893,28 @@ def double_axes_echem_plot(
         p.js_on_event(DoubleTap, CustomJS(args=dict(p=p), code="p.reset.emit()"))
 
     if mode == "dQ/dV":
-        grid = [[p1, p2], [xaxis_select]]
+        save_data = Button(label="Download .csv", button_type="primary", width_policy="min")
+        save_data_callback = CustomJS(
+            args=dict(source=ColumnDataSource(dfs[0])),
+            code=GENERATE_CSV_CALLBACK,
+        )
+        save_data.js_on_click(save_data_callback)
+        grid = [[save_data], [p1, p2], [xaxis_select]]
+
     elif mode == "dV/dQ":
-        grid = [[p1], [p2]]
+        save_data = Button(label="Download .csv", button_type="primary", width_policy="min")
+        save_data_callback = CustomJS(
+            args=dict(source=ColumnDataSource(dfs[0])),
+            code=GENERATE_CSV_CALLBACK,
+        )
+        save_data.js_on_click(save_data_callback)
+        grid = [[save_data], [p1], [p2]]
+
     elif mode == "final capacity":
-        if cycle_summary is not None:
+        if cycle_summary_dfs:
             save_data = Button(label="Download .csv", button_type="primary", width_policy="min")
             save_data_callback = CustomJS(
-                args=dict(source=ColumnDataSource(cycle_summary)),
+                args=dict(source=ColumnDataSource(cycle_summary_dfs[0])),
                 code=GENERATE_CSV_CALLBACK,
             )
             save_data.js_on_click(save_data_callback)