save and load buttons

Author: Tara-Lakshmipathy

js/widget.jsx

@@ -286,6 +286,26 @@ const render = createRender(() => {
     }
   }
 
+  const saveFunction = (dateTime) => {
+    console.log('save_workflow at ', dateTime);
+    if (model) {
+      model.set("commands", `save_workflow at ${dateTime}`);
+      model.save_changes();
+    } else {
+      console.error('model is undefined');
+    }
+  }
+
+  const loadFunction = (dateTime) => {
+    console.log('load_workflow at ', dateTime);
+    if (model) {
+      model.set("commands", `load_workflow at ${dateTime}`);
+      model.save_changes();
+    } else {
+      console.error('model is undefined');
+    }
+  }
+
   return (    
     <div style={{ position: "relative", height: "800px", width: "100%" }}>
       <UpdateDataContext.Provider value={updateData}> 
@@ -325,13 +345,13 @@ const render = createRender(() => {
           </button>
           <button
             style={{position: "absolute", left: "120px", top: "10px", zIndex: "4"}}
-            /*onClick={() => macroFunction(macroName)}*/
+            onClick={() => saveFunction(currentDateTime)}
           >
             Save Workflow
           </button>
           <button
             style={{position: "absolute", left: "230px", top: "10px", zIndex: "4"}}
-            /*onClick={() => macroFunction(macroName)}*/
+            onClick={() => loadFunction(currentDateTime)}
           >
             Load Workflow
           </button>

pyironflow/reactflow.py

@@ -267,9 +267,23 @@ def on_value_change(self, change):
                 elif global_command == 'run_workflow':
                     if self.accordion_widget is not None:
                         self.accordion_widget.selected_index = 1
+                    self.out_widget.clear_output()
                     out = self.wf.run()
                     display(out)
 
+                elif global_command == 'save_workflow':
+                    if self.accordion_widget is not None:
+                        self.accordion_widget.selected_index = 1
+                    # self.wf.save()
+                    print("Functionality not yet implemented")
+
+                elif global_command == 'load_workflow':
+                    if self.accordion_widget is not None:
+                        self.accordion_widget.selected_index = 1
+                    # self.wf.load()
+                    print("Functionality not yet implemented")
+                    
+
     def update(self):
         nodes = get_nodes(self.wf)
         edges = get_edges(self.wf)