@macrostrat/feedback-components 1.1.10 → 2.0.0

package/dist/feedback/edit-state.cjs

@@ -0,0 +1,382 @@
+"use strict";
+Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
+const react = require("react");
+const update = require("immutability-helper");
+function useUpdatableTree(initialTree, entityTypes, viewOnly, matchMode2, autoSelect = []) {
+  const type = entityTypes.values().next().value;
+  let selectedNodes = [];
+  autoSelect = autoSelect.map((name) => name.toLowerCase());
+  if (autoSelect.length > 0) {
+    selectedNodes = initialTree.flatMap((node) => node.children ?? []).filter((node) => autoSelect.includes(node.name.toLowerCase())).map((node) => node.id);
+  }
+  return react.useReducer(treeReducer, {
+    initialTree,
+    tree: initialTree,
+    selectedNodes,
+    entityTypesMap: entityTypes,
+    selectedEntityType: type,
+    lastInternalId: 0,
+    isSelectingEntityType: false,
+    viewMode: "tree",
+    viewOnly,
+    matchMode: matchMode2
+  });
+}
+const TreeDispatchContext = react.createContext(null);
+function useTreeDispatch() {
+  const dispatch = react.useContext(TreeDispatchContext);
+  if (dispatch == null) {
+    throw new Error("No dispatch context available");
+  }
+  return dispatch;
+}
+function treeReducer(state, action) {
+  if (action.type === "toggle-match-mode") {
+    return { ...state, matchMode: !state.matchMode };
+  }
+  if (action.type === "toggle-view-only") {
+    return { ...state, viewOnly: !state.viewOnly, selectedNodes: [] };
+  }
+  if (state.viewOnly) return viewMode(state, action);
+  if (state.matchMode) return matchMode(state, action);
+  switch (action.type) {
+    case "add-entity-type": {
+      const { name, description, color } = action.payload;
+      const newId2 = state.lastInternalId - 1;
+      const newType2 = {
+        id: newId2,
+        name,
+        description: description === "" ? null : description,
+        color
+      };
+      const newEntityTypesMap = new Map(state.entityTypesMap);
+      newEntityTypesMap.set(newId2, newType2);
+      return {
+        ...state,
+        entityTypesMap: newEntityTypesMap,
+        selectedEntityType: newType2,
+        lastInternalId: newId2
+      };
+    }
+    case "update-entity-type": {
+      const { id, name, description, color } = action.payload;
+      const newEntityTypesMap = new Map(state.entityTypesMap);
+      const oldType = newEntityTypesMap.get(id);
+      if (!oldType) {
+        console.warn(`Entity type with id ${id} not found`);
+        return state;
+      }
+      const updatedType = {
+        ...oldType,
+        name,
+        description: description === "" ? null : description,
+        color
+      };
+      newEntityTypesMap.set(id, updatedType);
+      const newTree3 = updateTreeTypes(state.tree, oldType, updatedType);
+      return {
+        ...state,
+        tree: newTree3,
+        entityTypesMap: newEntityTypesMap,
+        selectedEntityType: updatedType
+      };
+    }
+    case "select-range":
+      const payloadIds = action.payload.ids;
+      const node1 = payloadIds[0];
+      const node2 = payloadIds[1];
+      const allNodes = flattenAndSort(state.tree);
+      const startIndex = allNodes.findIndex((node3) => node3.id === node1);
+      const endIndex = allNodes.findIndex((node3) => node3.id === node2);
+      const selectedNodes = allNodes.slice(startIndex, endIndex + 1);
+      console.log("Selecting range:", selectedNodes);
+      return {
+        ...state,
+        selectedNodes: selectedNodes.map((node3) => node3.id)
+      };
+    case "move-node":
+      const [newTree, removedNodes] = removeNodes(
+        state.tree,
+        action.payload.dragIds
+      );
+      let keyPath = [];
+      if (action.payload.parentId) {
+        keyPath = findNode(newTree, action.payload.parentId);
+        keyPath.push("children");
+      }
+      let updateSpec = buildNestedSpec(keyPath, {
+        $splice: [[action.payload.index, 0, ...removedNodes]]
+      });
+      return { ...state, tree: update(newTree, updateSpec) };
+    case "delete-node":
+      const [newTree2, _removedNodes] = removeNodes(
+        state.tree,
+        action.payload.ids
+      );
+      const children = _removedNodes.flatMap((node3) => node3.children ?? []).filter((child) => !nodeIsInTree(newTree2, child.id));
+      return {
+        ...state,
+        tree: [...newTree2, ...children],
+        selectedNodes: state.selectedNodes.filter(
+          (id) => !action.payload.ids.includes(id)
+        )
+      };
+    case "select-node":
+      const { ids } = action.payload;
+      const type = action.payload.ids.length > 0 ? findNodeById(state.tree, ids[0])?.type : null;
+      return { ...state, selectedNodes: ids, selectedEntityType: type };
+    // otherwise fall through to toggle-node-selected for a single ID
+    case "toggle-node-selected":
+      const nodesToAdd = action.payload.ids.filter(
+        (id) => !state.selectedNodes.includes(id)
+      );
+      const nodesToKeep = state.selectedNodes.filter(
+        (id) => !action.payload.ids.includes(id)
+      );
+      const newType = action.payload.ids.length > 0 ? findNodeById(state.tree, action.payload.ids[0])?.type : null;
+      return {
+        ...state,
+        selectedNodes: [...nodesToKeep, ...nodesToAdd],
+        selectedEntityType: newType
+      };
+    case "create-node":
+      const newId = state.lastInternalId - 1;
+      const { text, start, end } = action.payload;
+      const node = {
+        id: newId,
+        name: text,
+        children: [],
+        indices: [start, end],
+        type: state.selectedEntityType
+      };
+      return {
+        ...state,
+        tree: [...state.tree, node],
+        selectedNodes: [newId],
+        lastInternalId: newId
+      };
+    case "delete-entity-type": {
+      const { id } = action.payload;
+      const newEntityTypesMap = new Map(state.entityTypesMap);
+      const oldType = newEntityTypesMap.get(id);
+      newEntityTypesMap.delete(id);
+      const defaultType = newEntityTypesMap.values().next().value;
+      const newTree3 = updateTreeTypes(state.tree, oldType, defaultType);
+      return {
+        ...state,
+        tree: newTree3,
+        entityTypesMap: newEntityTypesMap,
+        selectedNodes: []
+      };
+    }
+    /** Entity type selection */
+    case "toggle-entity-type-selector":
+      return {
+        ...state,
+        isSelectingEntityType: action.payload ?? !state.isSelectingEntityType
+      };
+    case "select-entity-type": {
+      let newTree22 = state.tree;
+      for (let id of state.selectedNodes) {
+        const keyPath2 = findNode(state.tree, id);
+        const nestedSpec = buildNestedSpec(keyPath2, {
+          type: { $set: action.payload }
+        });
+        newTree22 = update(newTree22, nestedSpec);
+      }
+      return {
+        ...state,
+        tree: newTree22,
+        selectedEntityType: action.payload
+      };
+    }
+    case "deselect":
+      return { ...state, selectedNodes: [] };
+    case "reset":
+      return {
+        ...state,
+        tree: state.initialTree,
+        selectedNodes: []
+      };
+    case "set-view-mode":
+      return { ...state, viewMode: action.payload };
+  }
+}
+function nodeIsInTree(tree, id) {
+  for (let node of tree) {
+    if (node.id == id) {
+      return true;
+    } else if (node.children) {
+      if (nodeIsInTree(node.children, id)) {
+        return true;
+      }
+    }
+  }
+  return false;
+}
+function buildNestedSpec(keyPath, innerSpec) {
+  let spec = innerSpec;
+  for (let i = keyPath.length - 1; i >= 0; i--) {
+    spec = { [keyPath[i]]: spec };
+  }
+  return spec;
+}
+function findNode(tree, id) {
+  for (let i = 0; i < tree.length; i++) {
+    if (tree[i].id == id) {
+      return [i];
+    } else if (tree[i].children) {
+      let path = findNode(tree[i].children, id);
+      if (path != null) {
+        return [i, "children", ...path];
+      }
+    }
+  }
+  return null;
+}
+function removeNodes(tree, ids) {
+  let newTree = [];
+  let removedNodes = [];
+  for (let node of tree) {
+    if (ids.includes(node.id)) {
+      removedNodes.push(node);
+    } else {
+      if (node.children) {
+        let [newChildren, removedChildren] = removeNodes(node.children, ids);
+        node = { ...node, children: newChildren };
+        removedNodes.push(...removedChildren);
+      }
+      newTree.push(node);
+    }
+  }
+  return [newTree, removedNodes];
+}
+function treeToGraph(tree) {
+  let nodes = [];
+  let edges = [];
+  const nodeMap = /* @__PURE__ */ new Map();
+  for (let node of tree) {
+    if (nodeMap.has(node.id)) {
+      continue;
+    }
+    const { indices, id, name, type, children } = node;
+    const nodeData = {
+      id,
+      type: type.id,
+      color: type.color,
+      name,
+      txt_range: [indices],
+      reasoning: null,
+      match: node.match,
+      children
+    };
+    nodeMap.set(node.id, node);
+    nodes.push(nodeData);
+    if (node.children) {
+      for (let child of node.children) {
+        edges.push({ source: node.id, dest: child.id });
+      }
+      const { nodes: childNodes, edges: childEdges } = treeToGraph(
+        node.children
+      );
+      nodes.push(...childNodes);
+      edges.push(...childEdges);
+    }
+  }
+  return { nodes, edges };
+}
+function findNodeById(tree, id) {
+  for (const node of tree) {
+    if (node.id === id) {
+      return node;
+    }
+    if (node.children) {
+      const found = findNodeById(node.children, id);
+      if (found) return found;
+    }
+  }
+  return null;
+}
+function updateTreeTypes(tree, oldType, defaultType) {
+  return tree.map((node) => updateNodeType(node, oldType, defaultType));
+}
+function updateNodeType(node, oldType, defaultType) {
+  const type = node.type.id === oldType.id ? defaultType : node.type;
+  return {
+    ...node,
+    type,
+    children: node.children ? updateTreeTypes(node.children, oldType, defaultType) : []
+  };
+}
+function flattenAndSort(nodes) {
+  const result = [];
+  function traverse(nodeList) {
+    for (const node of nodeList) {
+      result.push(node);
+      if (Array.isArray(node.children) && node.children.length > 0) {
+        traverse(node.children);
+      }
+    }
+  }
+  traverse(nodes);
+  return result.sort((a, b) => a.indices[0] - b.indices[0]);
+}
+function matchMode(state, action) {
+  if (action.type === "select-node" || action.type === "toggle-node-selected") {
+    const { ids } = action.payload;
+    if (ids.length != 1) return state;
+    if (state.selectedNodes.length === 1) {
+      if (ids[0] === state.selectedNodes[0]) {
+        return { ...state, selectedNodes: [] };
+      }
+    }
+    const type = action.payload.ids.length > 0 ? findNodeById(state.tree, ids[0])?.type : null;
+    return { ...state, selectedNodes: ids, selectedEntityType: type };
+  }
+  if (action.type === "add-match") {
+    const { id } = action.payload;
+    const keyPath = findNode(state.tree, id);
+    if (!keyPath) {
+      console.warn(`Node with id ${id} not found`);
+      return state;
+    }
+    const matchUpdateSpec = buildNestedSpec(keyPath, {
+      match: { $set: action.payload.payload }
+    });
+    const updatedTree = update(state.tree, matchUpdateSpec);
+    return {
+      ...state,
+      tree: updatedTree
+    };
+  }
+  if (action.type === "remove-match") {
+    const { id } = action.payload;
+    console.log("Removing match for node with id:", id);
+    const keyPath = findNode(state.tree, id);
+    if (!keyPath) {
+      console.warn(`Node with id ${id} not found`);
+      return state;
+    }
+    const matchUpdateSpec = buildNestedSpec(keyPath, {
+      match: { $set: null }
+    });
+    const updatedTree = update(state.tree, matchUpdateSpec);
+    return {
+      ...state,
+      tree: updatedTree
+    };
+  }
+  return state;
+}
+function viewMode(state, action) {
+  if (action.type === "set-view-mode") {
+    return { ...state, viewMode: action.payload };
+  }
+  return state;
+}
+exports.TreeDispatchContext = TreeDispatchContext;
+exports.findNodeById = findNodeById;
+exports.treeToGraph = treeToGraph;
+exports.useTreeDispatch = useTreeDispatch;
+exports.useUpdatableTree = useUpdatableTree;
+//# sourceMappingURL=edit-state.cjs.map

package/dist/feedback/edit-state.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"edit-state.cjs","sources":["../../src/feedback/edit-state.ts"],"sourcesContent":["import { TreeData } from \"./types\";\nimport { createContext, Dispatch, useContext, useReducer } from \"react\";\nimport update, { Spec } from \"immutability-helper\";\nimport { EntityType } from \"../extractions/types\";\n\nexport enum ViewMode {\n  Tree = \"tree\",\n  Graph = \"graph\",\n}\n\ninterface TreeState {\n  initialTree: TreeData[];\n  tree: TreeData[];\n  selectedNodes: number[];\n  entityTypesMap: Map<number, EntityType>;\n  selectedEntityType: EntityType;\n  lastInternalId: number;\n  isSelectingEntityType: boolean;\n  viewMode: ViewMode;\n  viewOnly: boolean;\n  matchMode: boolean;\n}\n\ntype TextRange = {\n  start: number;\n  end: number;\n  text: string;\n};\n\ntype TreeAction =\n  | {\n      type: \"move-node\";\n      payload: { dragIds: number[]; parentId: number; index: number };\n    }\n  | { type: \"delete-node\"; payload: { ids: number[] } }\n  | { type: \"select-node\"; payload: { ids: number[] } }\n  | { type: \"toggle-node-selected\"; payload: { ids: number[] } }\n  | { type: \"set-view-mode\"; payload: ViewMode }\n  | { type: \"create-node\"; payload: TextRange }\n  | { type: \"select-entity-type\"; payload: EntityType }\n  | { type: \"toggle-entity-type-selector\"; payload?: boolean | null }\n  | { type: \"deselect\" }\n  | { type: \"reset\" }\n  | { type: \"delete-entity-type\"; payload: { id: number } }\n  | {\n      type: \"add-entity-type\";\n      payload: { name: string; description: string; color: string };\n    }\n  | {\n      type: \"update-entity-type\";\n      payload: { id: number; name: string; description: string; color: string };\n    }\n  | { type: \"select-range\"; payload: { ids: number[] } }\n  | { type: \"add-match\"; payload: { id: number; payload: any } }\n  | { type: \"remove-match\"; payload: { id: number } }\n  | { type: \"toggle-match-mode\" }\n  | { type: \"toggle-view-only\" };\n\nexport type TreeDispatch = Dispatch<TreeAction>;\n\nexport function useUpdatableTree(\n  initialTree: TreeData[],\n  entityTypes: Map<number, EntityType>,\n  viewOnly: boolean,\n  matchMode: boolean,\n  autoSelect: string[] = [],\n): [TreeState, TreeDispatch] {\n  // Get the first entity type\n  // issue: grabs second entity instead of selected one\n  const type = entityTypes.values().next().value;\n\n  let selectedNodes = [];\n  autoSelect = autoSelect.map((name) => name.toLowerCase());\n\n  if (autoSelect.length > 0) {\n    // If autoSelect is provided, find the nodes with the matching names\n    selectedNodes = initialTree\n      .flatMap((node) => node.children ?? [])\n      .filter((node) => autoSelect.includes(node.name.toLowerCase()))\n      .map((node) => node.id);\n  }\n\n  return useReducer(treeReducer, {\n    initialTree,\n    tree: initialTree,\n    selectedNodes,\n    entityTypesMap: entityTypes,\n    selectedEntityType: type,\n    lastInternalId: 0,\n    isSelectingEntityType: false,\n    viewMode: ViewMode.Tree,\n    viewOnly,\n    matchMode,\n  });\n}\n\nexport const TreeDispatchContext = createContext<TreeDispatch | null>(null);\n\nexport function useTreeDispatch() {\n  const dispatch = useContext(TreeDispatchContext);\n  if (dispatch == null) {\n    throw new Error(\"No dispatch context available\");\n  }\n  return dispatch;\n}\n\nfunction treeReducer(state: TreeState, action: TreeAction) {\n  if (action.type === \"toggle-match-mode\") {\n    return { ...state, matchMode: !state.matchMode };\n  }\n\n  if (action.type === \"toggle-view-only\") {\n    return { ...state, viewOnly: !state.viewOnly, selectedNodes: [] };\n  }\n\n  if (state.viewOnly) return viewMode(state, action);\n\n  if (state.matchMode) return matchMode(state, action);\n\n  switch (action.type) {\n    case \"add-entity-type\": {\n      // Add a new entity type to the map\n      const { name, description, color } = action.payload;\n      const newId = state.lastInternalId - 1;\n      const newType: EntityType = {\n        id: newId,\n        name,\n        description: description === \"\" ? null : description,\n        color,\n      };\n\n      const newEntityTypesMap = new Map(state.entityTypesMap);\n      newEntityTypesMap.set(newId, newType);\n\n      return {\n        ...state,\n        entityTypesMap: newEntityTypesMap,\n        selectedEntityType: newType,\n        lastInternalId: newId,\n      };\n    }\n    case \"update-entity-type\": {\n      // Update an existing entity type in the map\n      const { id, name, description, color } = action.payload;\n      const newEntityTypesMap = new Map(state.entityTypesMap);\n      const oldType = newEntityTypesMap.get(id);\n\n      if (!oldType) {\n        console.warn(`Entity type with id ${id} not found`);\n        return state;\n      }\n\n      const updatedType: EntityType = {\n        ...oldType,\n        name,\n        description: description === \"\" ? null : description,\n        color,\n      };\n\n      newEntityTypesMap.set(id, updatedType);\n\n      // Update the tree to reflect the new type\n      const newTree = updateTreeTypes(state.tree, oldType, updatedType);\n\n      return {\n        ...state,\n        tree: newTree,\n        entityTypesMap: newEntityTypesMap,\n        selectedEntityType: updatedType,\n      };\n    }\n    case \"select-range\":\n      // Select a range of nodes by their IDs\n      const payloadIds = action.payload.ids;\n      const node1 = payloadIds[0];\n      const node2 = payloadIds[1];\n\n      // make list of nodes in order\n      const allNodes = flattenAndSort(state.tree);\n\n      // select all nodes between node1 and node2\n      const startIndex = allNodes.findIndex((node) => node.id === node1);\n      const endIndex = allNodes.findIndex((node) => node.id === node2);\n\n      const selectedNodes = allNodes.slice(startIndex, endIndex + 1);\n\n      console.log(\"Selecting range:\", selectedNodes);\n      return {\n        ...state,\n        selectedNodes: selectedNodes.map((node) => node.id),\n      };\n\n    case \"move-node\":\n      // For each node in the tree, if the node is in the dragIds, remove it from the tree and collect it\n      const [newTree, removedNodes] = removeNodes(\n        state.tree,\n        action.payload.dragIds,\n      );\n\n      let keyPath: (number | \"children\")[] = [];\n      if (action.payload.parentId) {\n        keyPath = findNode(newTree, action.payload.parentId);\n        keyPath.push(\"children\");\n      }\n\n      // Add removed nodes to the new tree at the correct location\n      let updateSpec = buildNestedSpec(keyPath, {\n        $splice: [[action.payload.index, 0, ...removedNodes]],\n      });\n\n      return { ...state, tree: update(newTree, updateSpec) };\n    case \"delete-node\":\n      // For each node in the tree, if the node is in the ids, remove it from the tree\n      const [newTree2, _removedNodes] = removeNodes(\n        state.tree,\n        action.payload.ids,\n      );\n      // Get children of the removed nodes\n      // If children are not present elsewhere in the tree, insert them\n\n      const children = _removedNodes\n        .flatMap((node) => node.children ?? [])\n        .filter((child) => !nodeIsInTree(newTree2, child.id));\n\n      // Reset the selection\n\n      return {\n        ...state,\n        tree: [...newTree2, ...children],\n        selectedNodes: state.selectedNodes.filter(\n          (id) => !action.payload.ids.includes(id),\n        ),\n      };\n    case \"select-node\":\n      const { ids } = action.payload;\n\n      const type =\n        action.payload.ids.length > 0\n          ? findNodeById(state.tree, ids[0])?.type\n          : null;\n\n      return { ...state, selectedNodes: ids, selectedEntityType: type };\n    // otherwise fall through to toggle-node-selected for a single ID\n    case \"toggle-node-selected\":\n      const nodesToAdd = action.payload.ids.filter(\n        (id) => !state.selectedNodes.includes(id),\n      );\n      const nodesToKeep = state.selectedNodes.filter(\n        (id) => !action.payload.ids.includes(id),\n      );\n\n      const newType =\n        action.payload.ids.length > 0\n          ? findNodeById(state.tree, action.payload.ids[0])?.type\n          : null;\n\n      return {\n        ...state,\n        selectedNodes: [...nodesToKeep, ...nodesToAdd],\n        selectedEntityType: newType,\n      };\n\n    case \"create-node\":\n      const newId = state.lastInternalId - 1;\n      const { text, start, end } = action.payload;\n      const node: TreeData = {\n        id: newId,\n        name: text,\n        children: [],\n        indices: [start, end],\n        type: state.selectedEntityType,\n      };\n\n      return {\n        ...state,\n        tree: [...state.tree, node],\n        selectedNodes: [newId],\n        lastInternalId: newId,\n      };\n\n    case \"delete-entity-type\": {\n      // Remove the entity type from the map\n      const { id } = action.payload;\n      const newEntityTypesMap = new Map(state.entityTypesMap);\n      const oldType = newEntityTypesMap.get(id);\n      newEntityTypesMap.delete(id);\n\n      const defaultType = newEntityTypesMap.values().next().value;\n      const newTree = updateTreeTypes(state.tree, oldType, defaultType);\n\n      return {\n        ...state,\n        tree: newTree,\n        entityTypesMap: newEntityTypesMap,\n        selectedNodes: [],\n      };\n    }\n\n    /** Entity type selection */\n    case \"toggle-entity-type-selector\":\n      return {\n        ...state,\n        isSelectingEntityType: action.payload ?? !state.isSelectingEntityType,\n      };\n    case \"select-entity-type\": {\n      // For each selected node, update the type\n      let newTree2 = state.tree;\n      for (let id of state.selectedNodes) {\n        const keyPath = findNode(state.tree, id);\n        const nestedSpec = buildNestedSpec(keyPath, {\n          type: { $set: action.payload },\n        });\n        newTree2 = update(newTree2, nestedSpec);\n      }\n\n      return {\n        ...state,\n        tree: newTree2,\n        selectedEntityType: action.payload,\n      };\n    }\n    case \"deselect\":\n      return { ...state, selectedNodes: [] };\n    case \"reset\":\n      return {\n        ...state,\n        tree: state.initialTree,\n        selectedNodes: [],\n      };\n    case \"set-view-mode\":\n      return { ...state, viewMode: action.payload };\n  }\n}\n\nfunction nodeIsInTree(tree: TreeData[], id: number): boolean {\n  for (let node of tree) {\n    if (node.id == id) {\n      return true;\n    } else if (node.children) {\n      if (nodeIsInTree(node.children, id)) {\n        return true;\n      }\n    }\n  }\n  return false;\n}\n\nfunction buildNestedSpec(\n  keyPath: (number | \"children\")[],\n  innerSpec: Spec<any>,\n): Spec<TreeData[]> {\n  // Build a nested object from a key path\n\n  let spec = innerSpec;\n  for (let i = keyPath.length - 1; i >= 0; i--) {\n    spec = { [keyPath[i]]: spec };\n  }\n  return spec as any;\n  // Since we don't have a \"children\" key at the root, we make the top-level spec an array\n}\n\nfunction findNode(\n  tree: TreeData[],\n  id: number,\n): (number | \"children\")[] | null {\n  // Find the index of the node with the given id in the tree, returning the key path\n  for (let i = 0; i < tree.length; i++) {\n    if (tree[i].id == id) {\n      return [i];\n    } else if (tree[i].children) {\n      let path = findNode(tree[i].children, id);\n      if (path != null) {\n        return [i, \"children\", ...path];\n      }\n    }\n  }\n  return null;\n}\n\nfunction removeNodes(\n  tree: TreeData[],\n  ids: number[],\n): [TreeData[], TreeData[]] {\n  /** Remove nodes with the given ids from the tree and return the new tree and the removed nodes */\n  let newTree: TreeData[] = [];\n  let removedNodes: TreeData[] = [];\n\n  for (let node of tree) {\n    if (ids.includes(node.id)) {\n      removedNodes.push(node);\n    } else {\n      // Recurse into children\n      if (node.children) {\n        let [newChildren, removedChildren] = removeNodes(node.children, ids);\n        node = { ...node, children: newChildren };\n        removedNodes.push(...removedChildren);\n      }\n      newTree.push(node);\n    }\n  }\n\n  return [newTree, removedNodes];\n}\n\nexport interface EntityOutput {\n  id: number;\n  type: number | null;\n  txt_range: number[][];\n  name: string;\n  match: any | null;\n  reasoning: string | null;\n  color: string | null;\n  children: any[] | null;\n}\n\nexport interface GraphData {\n  nodes: EntityOutput[];\n  edges: { source: number; dest: number }[];\n}\n\nexport function treeToGraph(tree: TreeData[]): GraphData {\n  // Convert the tree to a graph\n  let nodes: EntityOutput[] = [];\n  let edges: { source: number; dest: number }[] = [];\n  const nodeMap = new Map<number, TreeData>();\n\n  for (let node of tree) {\n    // If we've already found an instance of this node, we don't need to record\n    // it again\n    if (nodeMap.has(node.id)) {\n      continue;\n    }\n\n    const { indices, id, name, type, children } = node;\n\n    const nodeData: EntityOutput = {\n      id,\n      type: type.id,\n      color: type.color,\n      name,\n      txt_range: [indices],\n      reasoning: null,\n      match: node.match,\n      children,\n    };\n\n    nodeMap.set(node.id, node);\n    nodes.push(nodeData);\n\n    if (node.children) {\n      for (let child of node.children) {\n        edges.push({ source: node.id, dest: child.id });\n      }\n\n      // Now process the children\n      const { nodes: childNodes, edges: childEdges } = treeToGraph(\n        node.children,\n      );\n      nodes.push(...childNodes);\n      edges.push(...childEdges);\n    }\n  }\n\n  return { nodes, edges };\n}\n\nexport function findNodeById(tree, id) {\n  for (const node of tree) {\n    if (node.id === id) {\n      return node;\n    }\n    if (node.children) {\n      const found = findNodeById(node.children, id);\n      if (found) return found;\n    }\n  }\n  return null;\n}\n\nfunction updateTreeTypes(tree, oldType, defaultType) {\n  return tree.map((node) => updateNodeType(node, oldType, defaultType));\n}\n\nfunction updateNodeType(node, oldType, defaultType) {\n  const type = node.type.id === oldType.id ? defaultType : node.type;\n\n  return {\n    ...node,\n    type,\n    children: node.children\n      ? updateTreeTypes(node.children, oldType, defaultType)\n      : [],\n  };\n}\n\nfunction flattenAndSort(nodes) {\n  const result = [];\n\n  function traverse(nodeList) {\n    for (const node of nodeList) {\n      result.push(node);\n      if (Array.isArray(node.children) && node.children.length > 0) {\n        traverse(node.children);\n      }\n    }\n  }\n\n  traverse(nodes);\n\n  // sort by start\n  return result.sort((a, b) => a.indices[0] - b.indices[0]);\n}\n\nfunction matchMode(state, action) {\n  if (action.type === \"select-node\" || action.type === \"toggle-node-selected\") {\n    const { ids } = action.payload;\n\n    if (ids.length != 1) return state;\n\n    if (state.selectedNodes.length === 1) {\n      if (ids[0] === state.selectedNodes[0]) {\n        // If the selected node is the same as the current selection, deselect it\n        return { ...state, selectedNodes: [] };\n      }\n    }\n\n    const type =\n      action.payload.ids.length > 0\n        ? findNodeById(state.tree, ids[0])?.type\n        : null;\n\n    return { ...state, selectedNodes: ids, selectedEntityType: type };\n  }\n\n  if (action.type === \"add-match\") {\n    const { id } = action.payload;\n\n    // Find the node path\n    const keyPath = findNode(state.tree, id);\n    if (!keyPath) {\n      console.warn(`Node with id ${id} not found`);\n      return state;\n    }\n\n    // Build update spec to set the `match` property\n    const matchUpdateSpec = buildNestedSpec(keyPath, {\n      match: { $set: action.payload.payload },\n    });\n\n    const updatedTree = update(state.tree, matchUpdateSpec);\n\n    return {\n      ...state,\n      tree: updatedTree,\n    };\n  }\n\n  if (action.type === \"remove-match\") {\n    const { id } = action.payload;\n\n    console.log(\"Removing match for node with id:\", id);\n\n    // Find the node path\n    const keyPath = findNode(state.tree, id);\n    if (!keyPath) {\n      console.warn(`Node with id ${id} not found`);\n      return state;\n    }\n\n    // Build update spec to unset the `match` property\n    const matchUpdateSpec = buildNestedSpec(keyPath, {\n      match: { $set: null },\n    });\n\n    const updatedTree = update(state.tree, matchUpdateSpec);\n\n    return {\n      ...state,\n      tree: updatedTree,\n    };\n  }\n\n  return state;\n}\n\nfunction viewMode(state, action) {\n  if (action.type === \"set-view-mode\") {\n    return { ...state, viewMode: action.payload };\n  }\n\n  return state;\n}\n"],"names":["matchMode","useReducer","createContext","useContext","newId","newType","newTree","node","newTree2","keyPath"],"mappings":";;;;AA4DO,SAAS,iBACd,aACA,aACA,UACAA,YACA,aAAuB,IACI;AAG3B,QAAM,OAAO,YAAY,OAAA,EAAS,OAAO;AAEzC,MAAI,gBAAgB,CAAA;AACpB,eAAa,WAAW,IAAI,CAAC,SAAS,KAAK,aAAa;AAExD,MAAI,WAAW,SAAS,GAAG;AAEzB,oBAAgB,YACb,QAAQ,CAAC,SAAS,KAAK,YAAY,CAAA,CAAE,EACrC,OAAO,CAAC,SAAS,WAAW,SAAS,KAAK,KAAK,YAAA,CAAa,CAAC,EAC7D,IAAI,CAAC,SAAS,KAAK,EAAE;AAAA,EAC1B;AAEA,SAAOC,MAAAA,WAAW,aAAa;AAAA,IAC7B;AAAA,IACA,MAAM;AAAA,IACN;AAAA,IACA,gBAAgB;AAAA,IAChB,oBAAoB;AAAA,IACpB,gBAAgB;AAAA,IAChB,uBAAuB;AAAA,IACvB,UAAU;AAAA,IACV;AAAA,IACA,WAAAD;AAAAA,EAAA,CACD;AACH;AAEO,MAAM,sBAAsBE,MAAAA,cAAmC,IAAI;AAEnE,SAAS,kBAAkB;AAChC,QAAM,WAAWC,MAAAA,WAAW,mBAAmB;AAC/C,MAAI,YAAY,MAAM;AACpB,UAAM,IAAI,MAAM,+BAA+B;AAAA,EACjD;AACA,SAAO;AACT;AAEA,SAAS,YAAY,OAAkB,QAAoB;AACzD,MAAI,OAAO,SAAS,qBAAqB;AACvC,WAAO,EAAE,GAAG,OAAO,WAAW,CAAC,MAAM,UAAA;AAAA,EACvC;AAEA,MAAI,OAAO,SAAS,oBAAoB;AACtC,WAAO,EAAE,GAAG,OAAO,UAAU,CAAC,MAAM,UAAU,eAAe,GAAC;AAAA,EAChE;AAEA,MAAI,MAAM,SAAU,QAAO,SAAS,OAAO,MAAM;AAEjD,MAAI,MAAM,UAAW,QAAO,UAAU,OAAO,MAAM;AAEnD,UAAQ,OAAO,MAAA;AAAA,IACb,KAAK,mBAAmB;AAEtB,YAAM,EAAE,MAAM,aAAa,MAAA,IAAU,OAAO;AAC5C,YAAMC,SAAQ,MAAM,iBAAiB;AACrC,YAAMC,WAAsB;AAAA,QAC1B,IAAID;AAAAA,QACJ;AAAA,QACA,aAAa,gBAAgB,KAAK,OAAO;AAAA,QACzC;AAAA,MAAA;AAGF,YAAM,oBAAoB,IAAI,IAAI,MAAM,cAAc;AACtD,wBAAkB,IAAIA,QAAOC,QAAO;AAEpC,aAAO;AAAA,QACL,GAAG;AAAA,QACH,gBAAgB;AAAA,QAChB,oBAAoBA;AAAAA,QACpB,gBAAgBD;AAAAA,MAAA;AAAA,IAEpB;AAAA,IACA,KAAK,sBAAsB;AAEzB,YAAM,EAAE,IAAI,MAAM,aAAa,MAAA,IAAU,OAAO;AAChD,YAAM,oBAAoB,IAAI,IAAI,MAAM,cAAc;AACtD,YAAM,UAAU,kBAAkB,IAAI,EAAE;AAExC,UAAI,CAAC,SAAS;AACZ,gBAAQ,KAAK,uBAAuB,EAAE,YAAY;AAClD,eAAO;AAAA,MACT;AAEA,YAAM,cAA0B;AAAA,QAC9B,GAAG;AAAA,QACH;AAAA,QACA,aAAa,gBAAgB,KAAK,OAAO;AAAA,QACzC;AAAA,MAAA;AAGF,wBAAkB,IAAI,IAAI,WAAW;AAGrC,YAAME,WAAU,gBAAgB,MAAM,MAAM,SAAS,WAAW;AAEhE,aAAO;AAAA,QACL,GAAG;AAAA,QACH,MAAMA;AAAAA,QACN,gBAAgB;AAAA,QAChB,oBAAoB;AAAA,MAAA;AAAA,IAExB;AAAA,IACA,KAAK;AAEH,YAAM,aAAa,OAAO,QAAQ;AAClC,YAAM,QAAQ,WAAW,CAAC;AAC1B,YAAM,QAAQ,WAAW,CAAC;AAG1B,YAAM,WAAW,eAAe,MAAM,IAAI;AAG1C,YAAM,aAAa,SAAS,UAAU,CAACC,UAASA,MAAK,OAAO,KAAK;AACjE,YAAM,WAAW,SAAS,UAAU,CAACA,UAASA,MAAK,OAAO,KAAK;AAE/D,YAAM,gBAAgB,SAAS,MAAM,YAAY,WAAW,CAAC;AAE7D,cAAQ,IAAI,oBAAoB,aAAa;AAC7C,aAAO;AAAA,QACL,GAAG;AAAA,QACH,eAAe,cAAc,IAAI,CAACA,UAASA,MAAK,EAAE;AAAA,MAAA;AAAA,IAGtD,KAAK;AAEH,YAAM,CAAC,SAAS,YAAY,IAAI;AAAA,QAC9B,MAAM;AAAA,QACN,OAAO,QAAQ;AAAA,MAAA;AAGjB,UAAI,UAAmC,CAAA;AACvC,UAAI,OAAO,QAAQ,UAAU;AAC3B,kBAAU,SAAS,SAAS,OAAO,QAAQ,QAAQ;AACnD,gBAAQ,KAAK,UAAU;AAAA,MACzB;AAGA,UAAI,aAAa,gBAAgB,SAAS;AAAA,QACxC,SAAS,CAAC,CAAC,OAAO,QAAQ,OAAO,GAAG,GAAG,YAAY,CAAC;AAAA,MAAA,CACrD;AAED,aAAO,EAAE,GAAG,OAAO,MAAM,OAAO,SAAS,UAAU,EAAA;AAAA,IACrD,KAAK;AAEH,YAAM,CAAC,UAAU,aAAa,IAAI;AAAA,QAChC,MAAM;AAAA,QACN,OAAO,QAAQ;AAAA,MAAA;AAKjB,YAAM,WAAW,cACd,QAAQ,CAACA,UAASA,MAAK,YAAY,CAAA,CAAE,EACrC,OAAO,CAAC,UAAU,CAAC,aAAa,UAAU,MAAM,EAAE,CAAC;AAItD,aAAO;AAAA,QACL,GAAG;AAAA,QACH,MAAM,CAAC,GAAG,UAAU,GAAG,QAAQ;AAAA,QAC/B,eAAe,MAAM,cAAc;AAAA,UACjC,CAAC,OAAO,CAAC,OAAO,QAAQ,IAAI,SAAS,EAAE;AAAA,QAAA;AAAA,MACzC;AAAA,IAEJ,KAAK;AACH,YAAM,EAAE,QAAQ,OAAO;AAEvB,YAAM,OACJ,OAAO,QAAQ,IAAI,SAAS,IACxB,aAAa,MAAM,MAAM,IAAI,CAAC,CAAC,GAAG,OAClC;AAEN,aAAO,EAAE,GAAG,OAAO,eAAe,KAAK,oBAAoB,KAAA;AAAA;AAAA,IAE7D,KAAK;AACH,YAAM,aAAa,OAAO,QAAQ,IAAI;AAAA,QACpC,CAAC,OAAO,CAAC,MAAM,cAAc,SAAS,EAAE;AAAA,MAAA;AAE1C,YAAM,cAAc,MAAM,cAAc;AAAA,QACtC,CAAC,OAAO,CAAC,OAAO,QAAQ,IAAI,SAAS,EAAE;AAAA,MAAA;AAGzC,YAAM,UACJ,OAAO,QAAQ,IAAI,SAAS,IACxB,aAAa,MAAM,MAAM,OAAO,QAAQ,IAAI,CAAC,CAAC,GAAG,OACjD;AAEN,aAAO;AAAA,QACL,GAAG;AAAA,QACH,eAAe,CAAC,GAAG,aAAa,GAAG,UAAU;AAAA,QAC7C,oBAAoB;AAAA,MAAA;AAAA,IAGxB,KAAK;AACH,YAAM,QAAQ,MAAM,iBAAiB;AACrC,YAAM,EAAE,MAAM,OAAO,IAAA,IAAQ,OAAO;AACpC,YAAM,OAAiB;AAAA,QACrB,IAAI;AAAA,QACJ,MAAM;AAAA,QACN,UAAU,CAAA;AAAA,QACV,SAAS,CAAC,OAAO,GAAG;AAAA,QACpB,MAAM,MAAM;AAAA,MAAA;AAGd,aAAO;AAAA,QACL,GAAG;AAAA,QACH,MAAM,CAAC,GAAG,MAAM,MAAM,IAAI;AAAA,QAC1B,eAAe,CAAC,KAAK;AAAA,QACrB,gBAAgB;AAAA,MAAA;AAAA,IAGpB,KAAK,sBAAsB;AAEzB,YAAM,EAAE,OAAO,OAAO;AACtB,YAAM,oBAAoB,IAAI,IAAI,MAAM,cAAc;AACtD,YAAM,UAAU,kBAAkB,IAAI,EAAE;AACxC,wBAAkB,OAAO,EAAE;AAE3B,YAAM,cAAc,kBAAkB,OAAA,EAAS,OAAO;AACtD,YAAMD,WAAU,gBAAgB,MAAM,MAAM,SAAS,WAAW;AAEhE,aAAO;AAAA,QACL,GAAG;AAAA,QACH,MAAMA;AAAAA,QACN,gBAAgB;AAAA,QAChB,eAAe,CAAA;AAAA,MAAC;AAAA,IAEpB;AAAA;AAAA,IAGA,KAAK;AACH,aAAO;AAAA,QACL,GAAG;AAAA,QACH,uBAAuB,OAAO,WAAW,CAAC,MAAM;AAAA,MAAA;AAAA,IAEpD,KAAK,sBAAsB;AAEzB,UAAIE,YAAW,MAAM;AACrB,eAAS,MAAM,MAAM,eAAe;AAClC,cAAMC,WAAU,SAAS,MAAM,MAAM,EAAE;AACvC,cAAM,aAAa,gBAAgBA,UAAS;AAAA,UAC1C,MAAM,EAAE,MAAM,OAAO,QAAA;AAAA,QAAQ,CAC9B;AACDD,oBAAW,OAAOA,WAAU,UAAU;AAAA,MACxC;AAEA,aAAO;AAAA,QACL,GAAG;AAAA,QACH,MAAMA;AAAAA,QACN,oBAAoB,OAAO;AAAA,MAAA;AAAA,IAE/B;AAAA,IACA,KAAK;AACH,aAAO,EAAE,GAAG,OAAO,eAAe,GAAC;AAAA,IACrC,KAAK;AACH,aAAO;AAAA,QACL,GAAG;AAAA,QACH,MAAM,MAAM;AAAA,QACZ,eAAe,CAAA;AAAA,MAAC;AAAA,IAEpB,KAAK;AACH,aAAO,EAAE,GAAG,OAAO,UAAU,OAAO,QAAA;AAAA,EAAQ;AAElD;AAEA,SAAS,aAAa,MAAkB,IAAqB;AAC3D,WAAS,QAAQ,MAAM;AACrB,QAAI,KAAK,MAAM,IAAI;AACjB,aAAO;AAAA,IACT,WAAW,KAAK,UAAU;AACxB,UAAI,aAAa,KAAK,UAAU,EAAE,GAAG;AACnC,eAAO;AAAA,MACT;AAAA,IACF;AAAA,EACF;AACA,SAAO;AACT;AAEA,SAAS,gBACP,SACA,WACkB;AAGlB,MAAI,OAAO;AACX,WAAS,IAAI,QAAQ,SAAS,GAAG,KAAK,GAAG,KAAK;AAC5C,WAAO,EAAE,CAAC,QAAQ,CAAC,CAAC,GAAG,KAAA;AAAA,EACzB;AACA,SAAO;AAET;AAEA,SAAS,SACP,MACA,IACgC;AAEhC,WAAS,IAAI,GAAG,IAAI,KAAK,QAAQ,KAAK;AACpC,QAAI,KAAK,CAAC,EAAE,MAAM,IAAI;AACpB,aAAO,CAAC,CAAC;AAAA,IACX,WAAW,KAAK,CAAC,EAAE,UAAU;AAC3B,UAAI,OAAO,SAAS,KAAK,CAAC,EAAE,UAAU,EAAE;AACxC,UAAI,QAAQ,MAAM;AAChB,eAAO,CAAC,GAAG,YAAY,GAAG,IAAI;AAAA,MAChC;AAAA,IACF;AAAA,EACF;AACA,SAAO;AACT;AAEA,SAAS,YACP,MACA,KAC0B;AAE1B,MAAI,UAAsB,CAAA;AAC1B,MAAI,eAA2B,CAAA;AAE/B,WAAS,QAAQ,MAAM;AACrB,QAAI,IAAI,SAAS,KAAK,EAAE,GAAG;AACzB,mBAAa,KAAK,IAAI;AAAA,IACxB,OAAO;AAEL,UAAI,KAAK,UAAU;AACjB,YAAI,CAAC,aAAa,eAAe,IAAI,YAAY,KAAK,UAAU,GAAG;AACnE,eAAO,EAAE,GAAG,MAAM,UAAU,YAAA;AAC5B,qBAAa,KAAK,GAAG,eAAe;AAAA,MACtC;AACA,cAAQ,KAAK,IAAI;AAAA,IACnB;AAAA,EACF;AAEA,SAAO,CAAC,SAAS,YAAY;AAC/B;AAkBO,SAAS,YAAY,MAA6B;AAEvD,MAAI,QAAwB,CAAA;AAC5B,MAAI,QAA4C,CAAA;AAChD,QAAM,8BAAc,IAAA;AAEpB,WAAS,QAAQ,MAAM;AAGrB,QAAI,QAAQ,IAAI,KAAK,EAAE,GAAG;AACxB;AAAA,IACF;AAEA,UAAM,EAAE,SAAS,IAAI,MAAM,MAAM,aAAa;AAE9C,UAAM,WAAyB;AAAA,MAC7B;AAAA,MACA,MAAM,KAAK;AAAA,MACX,OAAO,KAAK;AAAA,MACZ;AAAA,MACA,WAAW,CAAC,OAAO;AAAA,MACnB,WAAW;AAAA,MACX,OAAO,KAAK;AAAA,MACZ;AAAA,IAAA;AAGF,YAAQ,IAAI,KAAK,IAAI,IAAI;AACzB,UAAM,KAAK,QAAQ;AAEnB,QAAI,KAAK,UAAU;AACjB,eAAS,SAAS,KAAK,UAAU;AAC/B,cAAM,KAAK,EAAE,QAAQ,KAAK,IAAI,MAAM,MAAM,IAAI;AAAA,MAChD;AAGA,YAAM,EAAE,OAAO,YAAY,OAAO,eAAe;AAAA,QAC/C,KAAK;AAAA,MAAA;AAEP,YAAM,KAAK,GAAG,UAAU;AACxB,YAAM,KAAK,GAAG,UAAU;AAAA,IAC1B;AAAA,EACF;AAEA,SAAO,EAAE,OAAO,MAAA;AAClB;AAEO,SAAS,aAAa,MAAM,IAAI;AACrC,aAAW,QAAQ,MAAM;AACvB,QAAI,KAAK,OAAO,IAAI;AAClB,aAAO;AAAA,IACT;AACA,QAAI,KAAK,UAAU;AACjB,YAAM,QAAQ,aAAa,KAAK,UAAU,EAAE;AAC5C,UAAI,MAAO,QAAO;AAAA,IACpB;AAAA,EACF;AACA,SAAO;AACT;AAEA,SAAS,gBAAgB,MAAM,SAAS,aAAa;AACnD,SAAO,KAAK,IAAI,CAAC,SAAS,eAAe,MAAM,SAAS,WAAW,CAAC;AACtE;AAEA,SAAS,eAAe,MAAM,SAAS,aAAa;AAClD,QAAM,OAAO,KAAK,KAAK,OAAO,QAAQ,KAAK,cAAc,KAAK;AAE9D,SAAO;AAAA,IACL,GAAG;AAAA,IACH;AAAA,IACA,UAAU,KAAK,WACX,gBAAgB,KAAK,UAAU,SAAS,WAAW,IACnD,CAAA;AAAA,EAAC;AAET;AAEA,SAAS,eAAe,OAAO;AAC7B,QAAM,SAAS,CAAA;AAEf,WAAS,SAAS,UAAU;AAC1B,eAAW,QAAQ,UAAU;AAC3B,aAAO,KAAK,IAAI;AAChB,UAAI,MAAM,QAAQ,KAAK,QAAQ,KAAK,KAAK,SAAS,SAAS,GAAG;AAC5D,iBAAS,KAAK,QAAQ;AAAA,MACxB;AAAA,IACF;AAAA,EACF;AAEA,WAAS,KAAK;AAGd,SAAO,OAAO,KAAK,CAAC,GAAG,MAAM,EAAE,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;AAC1D;AAEA,SAAS,UAAU,OAAO,QAAQ;AAChC,MAAI,OAAO,SAAS,iBAAiB,OAAO,SAAS,wBAAwB;AAC3E,UAAM,EAAE,QAAQ,OAAO;AAEvB,QAAI,IAAI,UAAU,EAAG,QAAO;AAE5B,QAAI,MAAM,cAAc,WAAW,GAAG;AACpC,UAAI,IAAI,CAAC,MAAM,MAAM,cAAc,CAAC,GAAG;AAErC,eAAO,EAAE,GAAG,OAAO,eAAe,GAAC;AAAA,MACrC;AAAA,IACF;AAEA,UAAM,OACJ,OAAO,QAAQ,IAAI,SAAS,IACxB,aAAa,MAAM,MAAM,IAAI,CAAC,CAAC,GAAG,OAClC;AAEN,WAAO,EAAE,GAAG,OAAO,eAAe,KAAK,oBAAoB,KAAA;AAAA,EAC7D;AAEA,MAAI,OAAO,SAAS,aAAa;AAC/B,UAAM,EAAE,OAAO,OAAO;AAGtB,UAAM,UAAU,SAAS,MAAM,MAAM,EAAE;AACvC,QAAI,CAAC,SAAS;AACZ,cAAQ,KAAK,gBAAgB,EAAE,YAAY;AAC3C,aAAO;AAAA,IACT;AAGA,UAAM,kBAAkB,gBAAgB,SAAS;AAAA,MAC/C,OAAO,EAAE,MAAM,OAAO,QAAQ,QAAA;AAAA,IAAQ,CACvC;AAED,UAAM,cAAc,OAAO,MAAM,MAAM,eAAe;AAEtD,WAAO;AAAA,MACL,GAAG;AAAA,MACH,MAAM;AAAA,IAAA;AAAA,EAEV;AAEA,MAAI,OAAO,SAAS,gBAAgB;AAClC,UAAM,EAAE,OAAO,OAAO;AAEtB,YAAQ,IAAI,oCAAoC,EAAE;AAGlD,UAAM,UAAU,SAAS,MAAM,MAAM,EAAE;AACvC,QAAI,CAAC,SAAS;AACZ,cAAQ,KAAK,gBAAgB,EAAE,YAAY;AAC3C,aAAO;AAAA,IACT;AAGA,UAAM,kBAAkB,gBAAgB,SAAS;AAAA,MAC/C,OAAO,EAAE,MAAM,KAAA;AAAA,IAAK,CACrB;AAED,UAAM,cAAc,OAAO,MAAM,MAAM,eAAe;AAEtD,WAAO;AAAA,MACL,GAAG;AAAA,MACH,MAAM;AAAA,IAAA;AAAA,EAEV;AAEA,SAAO;AACT;AAEA,SAAS,SAAS,OAAO,QAAQ;AAC/B,MAAI,OAAO,SAAS,iBAAiB;AACnC,WAAO,EAAE,GAAG,OAAO,UAAU,OAAO,QAAA;AAAA,EACtC;AAEA,SAAO;AACT;;;;;;"}

package/dist/feedback/edit-state.d.ts

@@ -0,0 +1,127 @@
+import { TreeData } from './types';
+import { Dispatch } from 'react';
+import { EntityType } from '../extractions/types';
+export declare enum ViewMode {
+    Tree = "tree",
+    Graph = "graph"
+}
+interface TreeState {
+    initialTree: TreeData[];
+    tree: TreeData[];
+    selectedNodes: number[];
+    entityTypesMap: Map<number, EntityType>;
+    selectedEntityType: EntityType;
+    lastInternalId: number;
+    isSelectingEntityType: boolean;
+    viewMode: ViewMode;
+    viewOnly: boolean;
+    matchMode: boolean;
+}
+type TextRange = {
+    start: number;
+    end: number;
+    text: string;
+};
+type TreeAction = {
+    type: "move-node";
+    payload: {
+        dragIds: number[];
+        parentId: number;
+        index: number;
+    };
+} | {
+    type: "delete-node";
+    payload: {
+        ids: number[];
+    };
+} | {
+    type: "select-node";
+    payload: {
+        ids: number[];
+    };
+} | {
+    type: "toggle-node-selected";
+    payload: {
+        ids: number[];
+    };
+} | {
+    type: "set-view-mode";
+    payload: ViewMode;
+} | {
+    type: "create-node";
+    payload: TextRange;
+} | {
+    type: "select-entity-type";
+    payload: EntityType;
+} | {
+    type: "toggle-entity-type-selector";
+    payload?: boolean | null;
+} | {
+    type: "deselect";
+} | {
+    type: "reset";
+} | {
+    type: "delete-entity-type";
+    payload: {
+        id: number;
+    };
+} | {
+    type: "add-entity-type";
+    payload: {
+        name: string;
+        description: string;
+        color: string;
+    };
+} | {
+    type: "update-entity-type";
+    payload: {
+        id: number;
+        name: string;
+        description: string;
+        color: string;
+    };
+} | {
+    type: "select-range";
+    payload: {
+        ids: number[];
+    };
+} | {
+    type: "add-match";
+    payload: {
+        id: number;
+        payload: any;
+    };
+} | {
+    type: "remove-match";
+    payload: {
+        id: number;
+    };
+} | {
+    type: "toggle-match-mode";
+} | {
+    type: "toggle-view-only";
+};
+export type TreeDispatch = Dispatch<TreeAction>;
+export declare function useUpdatableTree(initialTree: TreeData[], entityTypes: Map<number, EntityType>, viewOnly: boolean, matchMode: boolean, autoSelect?: string[]): [TreeState, TreeDispatch];
+export declare const TreeDispatchContext: import('react').Context<TreeDispatch>;
+export declare function useTreeDispatch(): TreeDispatch;
+export interface EntityOutput {
+    id: number;
+    type: number | null;
+    txt_range: number[][];
+    name: string;
+    match: any | null;
+    reasoning: string | null;
+    color: string | null;
+    children: any[] | null;
+}
+export interface GraphData {
+    nodes: EntityOutput[];
+    edges: {
+        source: number;
+        dest: number;
+    }[];
+}
+export declare function treeToGraph(tree: TreeData[]): GraphData;
+export declare function findNodeById(tree: any, id: any): any;
+export {};