Fixed WorkflowDesignerState

src/components/workflows/components/WorkflowDesigner.py

@@ -93,7 +93,10 @@ class WorkflowDesigner(BaseComponent):
     return self._key
   
   def refresh_designer(self, oob=False):
+    if oob:
       return self._mk_canvas(oob)
+    else:
+      return self._mk_elements()
   
   def refresh_properties(self, oob=False):
     return self._mk_properties(oob)
@@ -137,7 +140,7 @@ class WorkflowDesigner(BaseComponent):
       undo_redo_attrs = UndoRedoAttrs(f"Move Component '{component.title}'", on_undo=self.refresh_state)
       self._db.save_state(self._key, self._state, undo_redo_attrs)  # update db
     
-    return self.refresh_designer(), self.refresh_properties(True)
+    return self.refresh_designer(), self.refresh_properties(True), self._undo_redo.refresh()
   
   def delete_component(self, component_id):
     # Remove component
@@ -153,7 +156,7 @@ class WorkflowDesigner(BaseComponent):
       undo_redo_attrs = UndoRedoAttrs(f"Remove Component '{component.title}'", on_undo=self.refresh_state)
       self._db.save_state(self._key, self._state, undo_redo_attrs)
     
-    return self.refresh_designer()
+    return self.refresh_designer(), self._undo_redo.refresh()
   
   def add_connection(self, from_id, to_id):
     # Check if connection already exists
@@ -169,7 +172,7 @@ class WorkflowDesigner(BaseComponent):
     undo_redo_attrs = UndoRedoAttrs(f"Add Connection", on_undo=self.refresh_state)
     self._db.save_state(self._key, self._state, undo_redo_attrs)
     
-    return self.refresh_designer()
+    return self.refresh_designer(), self._undo_redo.refresh()
   
   def delete_connection(self, from_id, to_id):
     for connection in self._state.connections:
@@ -180,14 +183,14 @@ class WorkflowDesigner(BaseComponent):
     undo_redo_attrs = UndoRedoAttrs(f"Delete Connection", on_undo=self.refresh_state)
     self._db.save_state(self._key, self._state, undo_redo_attrs)
     
-    return self.refresh_designer()
+    return self.refresh_designer(), self._undo_redo.refresh()
   
   def set_designer_height(self, height):
     self._state.designer_height = height
     
-    undo_redo_attrs = UndoRedoAttrs(f"Resize Designer", on_undo=self.refresh_state)
+    undo_redo_attrs = UndoRedoAttrs(f"Resize Designer", on_undo=lambda: self.refresh_state())
     self._db.save_state(self._key, self._state, undo_redo_attrs)
-    return self.__ft__()  # refresh the whole component
+    return self.__ft__(), self._undo_redo.refresh()  # refresh the whole component
   
   def select_component(self, component_id):
     if component_id in self._state.components:
@@ -197,7 +200,7 @@ class WorkflowDesigner(BaseComponent):
       undo_redo_attrs = UndoRedoAttrs(f"Select Component {component.title}", on_undo=self.refresh_state)
       self._db.save_state(self._key, self._state, undo_redo_attrs)
     
-    return self.refresh_properties()
+    return self.refresh_properties(), self._undo_redo.refresh()
   
   def save_properties(self, component_id: str, details: dict):
     if component_id in self._state.components:
@@ -208,7 +211,7 @@ class WorkflowDesigner(BaseComponent):
       self._db.save_state(self._key, self._state, undo_redo_attrs)
       logger.debug(f"Saved properties for component {component_id}: {details}")
     
-    return self.refresh_properties()
+    return self.refresh_properties(), self._undo_redo.refresh()
   
   def cancel_properties(self, component_id: str):
     if component_id in self._state.components:
@@ -223,7 +226,7 @@ class WorkflowDesigner(BaseComponent):
       
       undo_redo_attrs = UndoRedoAttrs(f"Set Processor for {component.title}", on_undo=self.refresh_state)
       self._db.save_state(self._key, self._state, undo_redo_attrs)
-    return self.refresh_properties()
+    return self.refresh_properties(), self._undo_redo.refresh()
   
   def play_workflow(self, boundaries: dict):
     self._error_message = None

src/components/workflows/db_management.py

@@ -59,9 +59,9 @@ class WorkflowsDesignerSettings:
 class WorkflowsDesignerState:
   components: dict[str, WorkflowComponent] = field(default_factory=dict)
   connections: list[Connection] = field(default_factory=list)
-  component_counter = 0
-  designer_height = 230
-  selected_component_id = None
+  component_counter: int = 0
+  designer_height: int = 230
+  selected_component_id: str | None = None
 
 
 @dataclass