@macrostrat/feedback-components 1.0.1 → 1.1.1

package/dist/esm/{extractions.65bb73cc.js → feedback-components.5df2a926.js}

@@ -1,9 +1,10 @@
-import "./main.module.2f2972c8.css";
-import $9y0hV$mainmodulecd706d67js from "./main.module.cd706d67.js";
+import "./feedback-components.e273ed5b.css";
+import $9y0hV$feedbackcomponents6d32ee91js from "./feedback-components.6d32ee91.js";
 import $9y0hV$classnames from "classnames";
 import {Tag as $9y0hV$Tag} from "@blueprintjs/core";
 import {asChromaColor as $9y0hV$asChromaColor} from "@macrostrat/color-utils";
 import $9y0hV$macrostrathyper from "@macrostrat/hyper";
+import {useDarkMode as $9y0hV$useDarkMode} from "@macrostrat/ui-components";
 
 
 function $parcel$interopDefault(a) {
@@ -14,7 +15,8 @@ function $parcel$interopDefault(a) {
 
 
 
-const $03d8811e9c9b360d$var$h = (0, $9y0hV$macrostrathyper).styled((0, ($parcel$interopDefault($9y0hV$mainmodulecd706d67js))));
+
+const $03d8811e9c9b360d$var$h = (0, $9y0hV$macrostrathyper).styled((0, ($parcel$interopDefault($9y0hV$feedbackcomponents6d32ee91js))));
 function $03d8811e9c9b360d$export$c4b91360064ad200(entities, parent) {
     let highlights = [];
     let parents = [];
@@ -27,8 +29,8 @@ function $03d8811e9c9b360d$export$c4b91360064ad200(entities, parent) {
             start: entity.indices[0],
             end: entity.indices[1],
             text: entity.name,
-            backgroundColor: entity.type.color ?? "#ddd",
-            tag: entity.type.name,
+            backgroundColor: entity.type?.color,
+            tag: entity.type?.name ?? "lith",
             id: entity.id,
             parents: parents
         });
@@ -44,25 +46,32 @@ function $03d8811e9c9b360d$export$d60b563f571177d(extractionData, models, entity
     };
 }
 function $03d8811e9c9b360d$export$35baa338324d8550(baseColor, options) {
-    const _baseColor = (0, $9y0hV$asChromaColor)(baseColor ?? "#ddd");
-    const { highlighted: highlighted = true, inDarkMode: inDarkMode = false, active: active = false } = options;
+    const _baseColor = (0, $9y0hV$asChromaColor)(baseColor ?? "#fff");
+    const { highlighted: highlighted = true, inDarkMode: inDarkMode = (0, $9y0hV$useDarkMode)().isEnabled, active: active = false } = options;
     let mixAmount = highlighted ? 0.8 : 0.5;
     let backgroundAlpha = highlighted ? 0.8 : 0.2;
     if (active) {
         mixAmount = 1;
         backgroundAlpha = 1;
     }
-    const mixTarget = inDarkMode ? "white" : "black";
-    const color = _baseColor.mix(mixTarget, mixAmount).css();
-    const borderColor = highlighted ? _baseColor.mix(mixTarget, mixAmount / 2).css() : "transparent";
+    const mixTarget = "black";
+    const color = active ? "#000" : _baseColor.mix(mixTarget, mixAmount).hex();
+    const borderColor = highlighted ? _baseColor.mix(mixTarget, mixAmount / 1.1).hex() : "transparent";
+    let backgroundColor = active ? _baseColor.alpha(backgroundAlpha).hex() : $03d8811e9c9b360d$var$normalizeColor(_baseColor.alpha(backgroundAlpha).hex());
+    // handle white backgrounds in light mode
+    if (!inDarkMode && backgroundColor === "#ffffff") {
+        console.log("Adjusting background color for light mode:", backgroundColor);
+        backgroundColor = "#f0f0f0";
+    }
     return {
         color: color,
-        backgroundColor: _baseColor.alpha(backgroundAlpha).css(),
+        backgroundColor: backgroundColor,
         boxSizing: "border-box",
         borderStyle: "solid",
         borderColor: borderColor,
-        borderWidth: "1px",
-        fontWeight: active ? "bold" : "normal"
+        borderWidth: "1.5px",
+        fontWeight: active ? "bold" : "normal",
+        fontSize: "0.9em"
     };
 }
 function $03d8811e9c9b360d$var$enhanceEntity(entity, entityTypes) {
@@ -73,7 +82,7 @@ function $03d8811e9c9b360d$var$enhanceEntity(entity, entityTypes) {
     };
 }
 function $03d8811e9c9b360d$var$addColor(entityType, match = false) {
-    const color = (0, $9y0hV$asChromaColor)(entityType.color ?? "#ddd").brighten(match ? 1 : 2);
+    const color = (0, $9y0hV$asChromaColor)(entityType.color ?? "#fff").brighten(match ? 1 : 2);
     return {
         ...entityType,
         color: color.css()
@@ -105,9 +114,9 @@ function $03d8811e9c9b360d$export$117e56c71b172cde({ data: data, highlighted: hi
     const { name: name, type: type, match: match } = data;
     const className = (0, $9y0hV$classnames)({
         matched: match != null,
-        type: data.type.name
+        type: data.type?.name ?? "lith"
     }, "entity");
-    const style = $03d8811e9c9b360d$export$35baa338324d8550(type.color ?? "#aaaaaa", {
+    const style = $03d8811e9c9b360d$export$35baa338324d8550(type?.color, {
         highlighted: highlighted,
         active: active
     });
@@ -129,7 +138,7 @@ function $03d8811e9c9b360d$export$117e56c71b172cde({ data: data, highlighted: hi
                 }
             }
         }, [
-            type.name,
+            type?.name,
             _matchLink
         ])
     ]);
@@ -171,7 +180,26 @@ function $03d8811e9c9b360d$var$HighlightedText(props) {
     parts.push(text.slice(start));
     return $03d8811e9c9b360d$var$h("span", parts);
 }
+function $03d8811e9c9b360d$var$normalizeColor(hex8) {
+    const background = (0, $9y0hV$useDarkMode)().isEnabled ? "#000000" : "#ffffff";
+    const r = parseInt(hex8.slice(1, 3), 16);
+    const g = parseInt(hex8.slice(3, 5), 16);
+    const b = parseInt(hex8.slice(5, 7), 16);
+    const a = parseInt(hex8.slice(7, 9), 16) / 255;
+    const bgR = parseInt(background.slice(1, 3), 16);
+    const bgG = parseInt(background.slice(3, 5), 16);
+    const bgB = parseInt(background.slice(5, 7), 16);
+    const blend = (fg, bg)=>Math.round((1 - a) * bg + a * fg);
+    const blendedR = blend(r, bgR);
+    const blendedG = blend(g, bgG);
+    const blendedB = blend(b, bgB);
+    return "#" + blendedR.toString(16).padStart(2, "0") + blendedG.toString(16).padStart(2, "0") + blendedB.toString(16).padStart(2, "0");
+}
+function $03d8811e9c9b360d$var$isHighlighted(id, selectedNodes, nodes) {
+    if (selectedNodes?.length === 0) return true;
+    return selectedNodes?.includes(id) || nodes?.some((node)=>selectedNodes?.includes(node.id) && node.children.some((child)=>child.id === id));
+}
 
 
 export {$03d8811e9c9b360d$export$c4b91360064ad200 as buildHighlights, $03d8811e9c9b360d$export$d60b563f571177d as enhanceData, $03d8811e9c9b360d$export$35baa338324d8550 as getTagStyle, $03d8811e9c9b360d$export$2b07158757c249e0 as ExtractionContext, $03d8811e9c9b360d$export$4eb2a0ce903ce967 as ModelInfo, $03d8811e9c9b360d$export$117e56c71b172cde as EntityTag};
-//# sourceMappingURL=extractions.65bb73cc.js.map
+//# sourceMappingURL=feedback-components.5df2a926.js.map

package/dist/esm/feedback-components.5df2a926.js.map

@@ -0,0 +1 @@
+{"mappings":";;;;;;;;;;;;;;;;;;AAWA,MAAM,0BAAI,CAAA,GAAA,sBAAI,EAAE,MAAM,CAAC,CAAA,GAAA,8EAAK;AAErB,SAAS,0CACd,QAAqB,EACrB,MAAwB;IAExB,IAAI,aAAa,EAAE;IACnB,IAAI,UAAU,EAAE;IAChB,IAAI,UAAU,MACZ,UAAU;QAAC,OAAO,EAAE;WAAM,OAAO,OAAO,IAAI,EAAE;KAAE;IAGlD,KAAK,MAAM,UAAU,SAAU;QAC7B,WAAW,IAAI,CAAC;YACd,OAAO,OAAO,OAAO,CAAC,EAAE;YACxB,KAAK,OAAO,OAAO,CAAC,EAAE;YACtB,MAAM,OAAO,IAAI;YACjB,iBAAiB,OAAO,IAAI,EAAE;YAC9B,KAAK,OAAO,IAAI,EAAE,QAAQ;YAC1B,IAAI,OAAO,EAAE;qBACb;QACF;QACA,WAAW,IAAI,IAAI,0CAAgB,OAAO,QAAQ,IAAI,EAAE,EAAE;IAC5D;IACA,OAAO;AACT;AAEO,SAAS,yCAAY,cAAc,EAAE,MAAM,EAAE,WAAW;IAC7D,OAAO;QACL,GAAG,cAAc;QACjB,OAAO,OAAO,GAAG,CAAC,eAAe,QAAQ;QACzC,UAAU,eAAe,QAAQ,EAAE,IAAI,CAAC,IACtC,oCAAc,GAAG;IAErB;AACF;AAEO,SAAS,0CACd,SAAiB,EACjB,OAA0E;IAE1E,MAAM,aAAa,CAAA,GAAA,oBAAY,EAAE,aAAa;IAC9C,MAAM,eACJ,cAAc,kBACd,aAAa,CAAA,GAAA,kBAAU,IAAI,SAAS,UACpC,SAAS,OACV,GAAG;IAEJ,IAAI,YAAY,cAAc,MAAM;IACpC,IAAI,kBAAkB,cAAc,MAAM;IAE1C,IAAI,QAAQ;QACV,YAAY;QACZ,kBAAkB;IACpB;IAEA,MAAM,YAAY;IAElB,MAAM,QAAQ,SAAS,SAAS,WAAW,GAAG,CAAC,WAAW,WAAW,GAAG;IACxE,MAAM,cAAc,cAChB,WAAW,GAAG,CAAC,WAAW,YAAY,KAAK,GAAG,KAC9C;IAEJ,IAAI,kBAAkB,SAClB,WAAW,KAAK,CAAC,iBAAiB,GAAG,KACrC,qCAAe,WAAW,KAAK,CAAC,iBAAiB,GAAG;IAExD,yCAAyC;IACzC,IAAI,CAAC,cAAc,oBAAoB,WAAW;QAChD,QAAQ,GAAG,CAAC,8CAA8C;QAC1D,kBAAkB;IACpB;IAEA,OAAO;eACL;yBACA;QACA,WAAW;QACX,aAAa;qBACb;QACA,aAAa;QACb,YAAY,SAAS,SAAS;QAC9B,UAAU;IACZ;AACF;AAEA,SAAS,oCACP,MAAc,EACd,WAAoC;IAEpC,OAAO;QACL,GAAG,MAAM;QACT,MAAM,+BAAS,YAAY,GAAG,CAAC,OAAO,IAAI,GAAG,OAAO,KAAK,IAAI;QAC7D,UAAU,OAAO,QAAQ,EAAE,IAAI,CAAC,IAAM,oCAAc,GAAG;IACzD;AACF;AAEA,SAAS,+BAAS,UAAsB,EAAE,QAAQ,KAAK;IACrD,MAAM,QAAQ,CAAA,GAAA,oBAAY,EAAE,WAAW,KAAK,IAAI,QAAQ,QAAQ,CAC9D,QAAQ,IAAI;IAGd,OAAO;QAAE,GAAG,UAAU;QAAE,OAAO,MAAM,GAAG;IAAG;AAC7C;AAEO,SAAS,0CAAkB,QAChC,IAAI,eACJ,WAAW,kBACX,cAAc,EAKf;IACC,MAAM,aAAa,0CAAgB,KAAK,QAAQ,EAAE;IAElD,OAAO,wBAAE,OAAO;QACd,wBAAE,KAAK,wBAAE,uCAAiB;YAAE,MAAM,KAAK,cAAc;wBAAE;QAAW;QAClE,wBAAE,2CAAW;YAAE,MAAM,KAAK,KAAK;QAAC;QAChC,wBACE,eACA,KAAK,QAAQ,CAAC,GAAG,CAAC,CAAC,IAAM,wBAAE,sCAAgB;gBAAE,MAAM;gCAAG;YAAe;KAExE;AACH;AAEO,SAAS,0CAAU,QAAE,IAAI,EAAE;IAChC,OAAO,wBAAE,gBAAgB;QAAC;QAAW,wBAAE,iBAAiB,KAAK,IAAI;KAAE;AACrE;AAYO,SAAS,0CAAU,QACxB,IAAI,eACJ,cAAc,cACd,SAAS,oBACT,WAAW,kBACX,iBAAiB,MACF;IACf,MAAM,QAAE,IAAI,QAAE,IAAI,SAAE,KAAK,EAAE,GAAG;IAE9B,MAAM,YAAY,CAAA,GAAA,iBAAS,EACzB;QACE,SAAS,SAAS;QAClB,MAAM,KAAK,IAAI,EAAE,QAAQ;IAC3B,GACA;IAGF,MAAM,QAAQ,0CAAY,MAAM,OAAO;qBAAE;gBAAa;IAAO;IAE7D,IAAI,aAAa;IACjB,IAAI,SAAS,QAAQ,kBAAkB,MACrC,aAAa,wBAAE,gBAAgB;QAAE,MAAM;IAAM;IAG/C,OAAO,wBAAE,CAAA,GAAA,UAAE,GAAG;eAAE;mBAAO;IAAU,GAAG;QAClC,wBAAE,oBAAoB;QACtB;QACA,wBACE,6BACA;YACE,SAAQ,GAAG;gBACT,IAAI,UAAU,eAAe,MAAM;oBACjC,YAAY;oBACZ,IAAI,eAAe;gBACrB;YACF;QACF,GACA;YAAC,MAAM;YAAM;SAAW;KAE3B;AACH;AAEA,SAAS,qCAAe,QACtB,IAAI,kBACJ,iBAAiB,MAIlB;IACC,MAAM,WAAW,KAAK,QAAQ,IAAI,EAAE;IAEpC,OAAO,wBAAE,iBAAiB;QACxB,wBAAE,2CAAW;kBAAE;4BAAM;QAAe;QACpC,wBAAE,EAAE,CAAC,SAAS,MAAM,GAAG,GAAG;YACxB,wBACE,eACA,SAAS,GAAG,CAAC,CAAC,IAAM,wBAAE,sCAAgB;oBAAE,MAAM;oCAAG;gBAAe;SAEnE;KACF;AACH;AAEA,SAAS,sCAAgB,KAAgD;IACvE,MAAM,QAAE,IAAI,cAAE,aAAa,EAAE,EAAE,GAAG;IAClC,MAAM,QAAQ,EAAE;IAChB,IAAI,QAAQ;IAEZ,MAAM,mBAAmB,WAAW,IAAI,CAAC,CAAC,GAAG,IAAM,EAAE,KAAK,GAAG,EAAE,KAAK;IACpE,MAAM,yBAAyB,iBAAiB,GAAG,CAAC,CAAC,WAAW;QAC9D,IAAI,MAAM,GAAG,OAAO;QACpB,MAAM,OAAO,gBAAgB,CAAC,IAAI,EAAE;QACpC,IAAI,UAAU,KAAK,GAAG,KAAK,GAAG,EAC5B,UAAU,KAAK,GAAG,KAAK,GAAG;QAE5B,OAAO;IACT;IAEA,KAAK,MAAM,aAAa,uBAAwB;QAC9C,MAAM,EAAE,OAAO,CAAC,OAAE,GAAG,EAAE,GAAG,MAAM,GAAG;QACnC,MAAM,IAAI,CAAC,KAAK,KAAK,CAAC,OAAO;QAC7B,MAAM,IAAI,CAAC,wBAAE,kBAAkB;YAAE,OAAO;QAAK,GAAG,KAAK,KAAK,CAAC,GAAG;QAC9D,QAAQ;IACV;IACA,MAAM,IAAI,CAAC,KAAK,KAAK,CAAC;IACtB,OAAO,wBAAE,QAAQ;AACnB;AAEA,SAAS,qCAAe,IAAI;IAC1B,MAAM,aAAa,CAAA,GAAA,kBAAU,IAAI,SAAS,GAAG,YAAY;IAEzD,MAAM,IAAI,SAAS,KAAK,KAAK,CAAC,GAAG,IAAI;IACrC,MAAM,IAAI,SAAS,KAAK,KAAK,CAAC,GAAG,IAAI;IACrC,MAAM,IAAI,SAAS,KAAK,KAAK,CAAC,GAAG,IAAI;IACrC,MAAM,IAAI,SAAS,KAAK,KAAK,CAAC,GAAG,IAAI,MAAM;IAE3C,MAAM,MAAM,SAAS,WAAW,KAAK,CAAC,GAAG,IAAI;IAC7C,MAAM,MAAM,SAAS,WAAW,KAAK,CAAC,GAAG,IAAI;IAC7C,MAAM,MAAM,SAAS,WAAW,KAAK,CAAC,GAAG,IAAI;IAE7C,MAAM,QAAQ,CAAC,IAAI,KAAO,KAAK,KAAK,CAAC,AAAC,CAAA,IAAI,CAAA,IAAK,KAAK,IAAI;IAExD,MAAM,WAAW,MAAM,GAAG;IAC1B,MAAM,WAAW,MAAM,GAAG;IAC1B,MAAM,WAAW,MAAM,GAAG;IAE1B,OACE,MACA,SAAS,QAAQ,CAAC,IAAI,QAAQ,CAAC,GAAG,OAClC,SAAS,QAAQ,CAAC,IAAI,QAAQ,CAAC,GAAG,OAClC,SAAS,QAAQ,CAAC,IAAI,QAAQ,CAAC,GAAG;AAEtC;AAEA,SAAS,oCAAc,EAAU,EAAE,aAAuB,EAAE,KAAY;IACtE,IAAI,eAAe,WAAW,GAAG,OAAO;IACxC,OACE,eAAe,SAAS,OACxB,OAAO,KACL,CAAC,OACC,eAAe,SAAS,KAAK,EAAE,KAC/B,KAAK,QAAQ,CAAC,IAAI,CAAC,CAAC,QAAU,MAAM,EAAE,KAAK;AAGnD","sources":["packages/feedback-components/src/extractions/index.ts"],"sourcesContent":["import styles from \"./main.module.sass\";\nimport classNames from \"classnames\";\nimport { Tag } from \"@blueprintjs/core\";\nimport type { Entity, EntityExt, Highlight, EntityType } from \"./types\";\nimport { CSSProperties } from \"react\";\nimport { asChromaColor } from \"@macrostrat/color-utils\";\nimport hyper from \"@macrostrat/hyper\";\nimport { useDarkMode } from \"@macrostrat/ui-components\";\n\nexport type { Entity, EntityExt };\n\nconst h = hyper.styled(styles);\n\nexport function buildHighlights(\n  entities: EntityExt[],\n  parent: EntityExt | null,\n): Highlight[] {\n  let highlights = [];\n  let parents = [];\n  if (parent != null) {\n    parents = [parent.id, ...(parent.parents ?? [])];\n  }\n\n  for (const entity of entities) {\n    highlights.push({\n      start: entity.indices[0],\n      end: entity.indices[1],\n      text: entity.name,\n      backgroundColor: entity.type?.color,\n      tag: entity.type?.name ?? \"lith\",\n      id: entity.id,\n      parents,\n    });\n    highlights.push(...buildHighlights(entity.children ?? [], entity));\n  }\n  return highlights;\n}\n\nexport function enhanceData(extractionData, models, entityTypes) {\n  return {\n    ...extractionData,\n    model: models.get(extractionData.model_id),\n    entities: extractionData.entities?.map((d) =>\n      enhanceEntity(d, entityTypes),\n    ),\n  };\n}\n\nexport function getTagStyle(\n  baseColor: string,\n  options: { highlighted?: boolean; inDarkMode?: boolean; active?: boolean },\n): CSSProperties {\n  const _baseColor = asChromaColor(baseColor ?? \"#fff\");\n  const {\n    highlighted = true,\n    inDarkMode = useDarkMode().isEnabled,\n    active = false,\n  } = options;\n\n  let mixAmount = highlighted ? 0.8 : 0.5;\n  let backgroundAlpha = highlighted ? 0.8 : 0.2;\n\n  if (active) {\n    mixAmount = 1;\n    backgroundAlpha = 1;\n  }\n\n  const mixTarget = \"black\";\n\n  const color = active ? \"#000\" : _baseColor.mix(mixTarget, mixAmount).hex();\n  const borderColor = highlighted\n    ? _baseColor.mix(mixTarget, mixAmount / 1.1).hex()\n    : \"transparent\";\n\n  let backgroundColor = active\n    ? _baseColor.alpha(backgroundAlpha).hex()\n    : normalizeColor(_baseColor.alpha(backgroundAlpha).hex());\n\n  // handle white backgrounds in light mode\n  if (!inDarkMode && backgroundColor === \"#ffffff\") {\n    console.log(\"Adjusting background color for light mode:\", backgroundColor);\n    backgroundColor = \"#f0f0f0\";\n  }\n\n  return {\n    color,\n    backgroundColor,\n    boxSizing: \"border-box\",\n    borderStyle: \"solid\",\n    borderColor,\n    borderWidth: \"1.5px\",\n    fontWeight: active ? \"bold\" : \"normal\",\n    fontSize: \"0.9em\",\n  };\n}\n\nfunction enhanceEntity(\n  entity: Entity,\n  entityTypes: Map<number, EntityType>,\n): EntityExt {\n  return {\n    ...entity,\n    type: addColor(entityTypes.get(entity.type), entity.match != null),\n    children: entity.children?.map((d) => enhanceEntity(d, entityTypes)),\n  };\n}\n\nfunction addColor(entityType: EntityType, match = false) {\n  const color = asChromaColor(entityType.color ?? \"#fff\").brighten(\n    match ? 1 : 2,\n  );\n\n  return { ...entityType, color: color.css() };\n}\n\nexport function ExtractionContext({\n  data,\n  entityTypes,\n  matchComponent,\n}: {\n  data: any;\n  entityTypes: Map<number, EntityType>;\n  matchComponent: MatchComponent;\n}) {\n  const highlights = buildHighlights(data.entities, null);\n\n  return h(\"div\", [\n    h(\"p\", h(HighlightedText, { text: data.paragraph_text, highlights })),\n    h(ModelInfo, { data: data.model }),\n    h(\n      \"ul.entities\",\n      data.entities.map((d) => h(ExtractionInfo, { data: d, matchComponent })),\n    ),\n  ]);\n}\n\nexport function ModelInfo({ data }) {\n  return h(\"p.model-name\", [\"Model: \", h(\"code.bp5-code\", data.name)]);\n}\n\nexport type MatchComponent = (props: { data: any }) => any;\n\ntype EntityTagProps = {\n  data: EntityExt;\n  highlighted?: boolean;\n  active?: boolean;\n  onClickType?: (type: EntityType) => void;\n  matchComponent?: MatchComponent;\n};\n\nexport function EntityTag({\n  data,\n  highlighted = true,\n  active = false,\n  onClickType,\n  matchComponent = null,\n}: EntityTagProps) {\n  const { name, type, match } = data;\n\n  const className = classNames(\n    {\n      matched: match != null,\n      type: data.type?.name ?? \"lith\",\n    },\n    \"entity\",\n  );\n\n  const style = getTagStyle(type?.color, { highlighted, active });\n\n  let _matchLink = null;\n  if (match != null && matchComponent != null) {\n    _matchLink = h(matchComponent, { data: match });\n  }\n\n  return h(Tag, { style, className }, [\n    h(\"span.entity-name\", name),\n    \" \",\n    h(\n      \"code.entity-type.bp5-code\",\n      {\n        onClick(evt) {\n          if (active && onClickType != null) {\n            onClickType(type);\n            evt.stopPropagation();\n          }\n        },\n      },\n      [type?.name, _matchLink],\n    ),\n  ]);\n}\n\nfunction ExtractionInfo({\n  data,\n  matchComponent = null,\n}: {\n  data: EntityExt;\n  matchComponent: MatchComponent;\n}) {\n  const children = data.children ?? [];\n\n  return h(\"li.entity-row\", [\n    h(EntityTag, { data, matchComponent }),\n    h.if(children.length > 0)([\n      h(\n        \"ul.children\",\n        children.map((d) => h(ExtractionInfo, { data: d, matchComponent })),\n      ),\n    ]),\n  ]);\n}\n\nfunction HighlightedText(props: { text: string; highlights: Highlight[] }) {\n  const { text, highlights = [] } = props;\n  const parts = [];\n  let start = 0;\n\n  const sortedHighlights = highlights.sort((a, b) => a.start - b.start);\n  const deconflictedHighlights = sortedHighlights.map((highlight, i) => {\n    if (i === 0) return highlight;\n    const prev = sortedHighlights[i - 1];\n    if (highlight.start < prev.end) {\n      highlight.start = prev.end;\n    }\n    return highlight;\n  });\n\n  for (const highlight of deconflictedHighlights) {\n    const { start: s, end, ...rest } = highlight;\n    parts.push(text.slice(start, s));\n    parts.push(h(\"span.highlight\", { style: rest }, text.slice(s, end)));\n    start = end;\n  }\n  parts.push(text.slice(start));\n  return h(\"span\", parts);\n}\n\nfunction normalizeColor(hex8) {\n  const background = useDarkMode().isEnabled ? \"#000000\" : \"#ffffff\";\n\n  const r = parseInt(hex8.slice(1, 3), 16);\n  const g = parseInt(hex8.slice(3, 5), 16);\n  const b = parseInt(hex8.slice(5, 7), 16);\n  const a = parseInt(hex8.slice(7, 9), 16) / 255;\n\n  const bgR = parseInt(background.slice(1, 3), 16);\n  const bgG = parseInt(background.slice(3, 5), 16);\n  const bgB = parseInt(background.slice(5, 7), 16);\n\n  const blend = (fg, bg) => Math.round((1 - a) * bg + a * fg);\n\n  const blendedR = blend(r, bgR);\n  const blendedG = blend(g, bgG);\n  const blendedB = blend(b, bgB);\n\n  return (\n    \"#\" +\n    blendedR.toString(16).padStart(2, \"0\") +\n    blendedG.toString(16).padStart(2, \"0\") +\n    blendedB.toString(16).padStart(2, \"0\")\n  );\n}\n\nfunction isHighlighted(id: number, selectedNodes: number[], nodes: any[]) {\n  if (selectedNodes?.length === 0) return true;\n  return (\n    selectedNodes?.includes(id) ||\n    nodes?.some(\n      (node) =>\n        selectedNodes?.includes(node.id) &&\n        node.children.some((child) => child.id === id),\n    )\n  );\n}\n"],"names":[],"version":3,"file":"feedback-components.5df2a926.js.map"}

package/dist/esm/{main.module.cd706d67.js → feedback-components.6d32ee91.js}

@@ -13,4 +13,4 @@ $83cb1ce4a2b71eef$export$4b5127edceda427d = `UZr1Jq_entity`;
 
 
 export {$83cb1ce4a2b71eef$exports as default};
-//# sourceMappingURL=main.module.cd706d67.js.map
+//# sourceMappingURL=feedback-components.6d32ee91.js.map

package/dist/esm/{main.module.cd706d67.js.map → feedback-components.6d32ee91.js.map}

@@ -1 +1 @@
-{"mappings":";;;;;;;;AAAA,IAAA;AACA,IAAA;AADA,4CAA6B,CAAC,eAAe,CAAC;AAC9C,4CAA2B,CAAC,aAAa,CAAC","sources":["packages/feedback-components/src/extractions/main.module.sass"],"sourcesContent":[".entities\n  list-style: none\n  padding-left: 0\n\n  .entities ul\n    list-style: none\n\n.entity\n  margin: 0.2em 0 0.5em\n  padding-right: 3px\n"],"names":[],"version":3,"file":"main.module.cd706d67.js.map"}
+{"mappings":";;;;;;;;AAAA,IAAA;AACA,IAAA;AADA,4CAA6B,CAAC,eAAe,CAAC;AAC9C,4CAA2B,CAAC,aAAa,CAAC","sources":["packages/feedback-components/src/extractions/main.module.sass"],"sourcesContent":[".entities\n  list-style: none\n  padding-left: 0\n\n  .entities ul\n    list-style: none\n\n.entity\n  margin: 0.2em 0 0.5em\n  padding-right: 3px\n"],"names":[],"version":3,"file":"feedback-components.6d32ee91.js.map"}

package/dist/esm/feedback-components.95dbe7d7.js

@@ -0,0 +1,82 @@
+
+function $parcel$export(e, n, v, s) {
+  Object.defineProperty(e, n, {get: v, set: s, enumerable: true, configurable: true});
+}
+var $49c87ee97f392268$exports = {};
+
+$parcel$export($49c87ee97f392268$exports, "add-type", () => $49c87ee97f392268$export$cb889aaddfa2926c, (v) => $49c87ee97f392268$export$cb889aaddfa2926c = v);
+$parcel$export($49c87ee97f392268$exports, "add-type-overlay", () => $49c87ee97f392268$export$5f8f750fd71deeed, (v) => $49c87ee97f392268$export$5f8f750fd71deeed = v);
+$parcel$export($49c87ee97f392268$exports, "control-content", () => $49c87ee97f392268$export$42e8b5293e69631e, (v) => $49c87ee97f392268$export$42e8b5293e69631e = v);
+$parcel$export($49c87ee97f392268$exports, "control-panel", () => $49c87ee97f392268$export$a0a23f31d5562853, (v) => $49c87ee97f392268$export$a0a23f31d5562853 = v);
+$parcel$export($49c87ee97f392268$exports, "description", () => $49c87ee97f392268$export$6d59db4903f20f7d, (v) => $49c87ee97f392268$export$6d59db4903f20f7d = v);
+$parcel$export($49c87ee97f392268$exports, "entity-panel", () => $49c87ee97f392268$export$38f2cd12e45697e, (v) => $49c87ee97f392268$export$38f2cd12e45697e = v);
+$parcel$export($49c87ee97f392268$exports, "feedback-component", () => $49c87ee97f392268$export$579bd45c78bf5510, (v) => $49c87ee97f392268$export$579bd45c78bf5510 = v);
+$parcel$export($49c87ee97f392268$exports, "feedback-container", () => $49c87ee97f392268$export$23889d205b405f6e, (v) => $49c87ee97f392268$export$23889d205b405f6e = v);
+$parcel$export($49c87ee97f392268$exports, "feedback-text", () => $49c87ee97f392268$export$890489be1870bed5, (v) => $49c87ee97f392268$export$890489be1870bed5 = v);
+$parcel$export($49c87ee97f392268$exports, "feedback-text-wrapper", () => $49c87ee97f392268$export$6bd0fd0a746c88f, (v) => $49c87ee97f392268$export$6bd0fd0a746c88f = v);
+$parcel$export($49c87ee97f392268$exports, "form-group", () => $49c87ee97f392268$export$55a1f87a143fb302, (v) => $49c87ee97f392268$export$55a1f87a143fb302 = v);
+$parcel$export($49c87ee97f392268$exports, "highlight", () => $49c87ee97f392268$export$e4043f67d36cbb26, (v) => $49c87ee97f392268$export$e4043f67d36cbb26 = v);
+$parcel$export($49c87ee97f392268$exports, "icons", () => $49c87ee97f392268$export$df03f54e09e486fa, (v) => $49c87ee97f392268$export$df03f54e09e486fa = v);
+$parcel$export($49c87ee97f392268$exports, "node", () => $49c87ee97f392268$export$35059013cd4a06db, (v) => $49c87ee97f392268$export$35059013cd4a06db = v);
+$parcel$export($49c87ee97f392268$exports, "node-label", () => $49c87ee97f392268$export$75d14899fb692bb6, (v) => $49c87ee97f392268$export$75d14899fb692bb6 = v);
+$parcel$export($49c87ee97f392268$exports, "overlay-container", () => $49c87ee97f392268$export$6a1093e41947f68e, (v) => $49c87ee97f392268$export$6a1093e41947f68e = v);
+$parcel$export($49c87ee97f392268$exports, "page-wrapper", () => $49c87ee97f392268$export$7c414e944018fb15, (v) => $49c87ee97f392268$export$7c414e944018fb15 = v);
+$parcel$export($49c87ee97f392268$exports, "selected", () => $49c87ee97f392268$export$d90250155de6d7e7, (v) => $49c87ee97f392268$export$d90250155de6d7e7 = v);
+$parcel$export($49c87ee97f392268$exports, "selection-tree", () => $49c87ee97f392268$export$3ae1b51693c80a9c, (v) => $49c87ee97f392268$export$3ae1b51693c80a9c = v);
+$parcel$export($49c87ee97f392268$exports, "text-inputs", () => $49c87ee97f392268$export$bc7581b048bdc9d, (v) => $49c87ee97f392268$export$bc7581b048bdc9d = v);
+$parcel$export($49c87ee97f392268$exports, "title", () => $49c87ee97f392268$export$fb184b623420d9be, (v) => $49c87ee97f392268$export$fb184b623420d9be = v);
+$parcel$export($49c87ee97f392268$exports, "type-container", () => $49c87ee97f392268$export$b2d17fca990bb66c, (v) => $49c87ee97f392268$export$b2d17fca990bb66c = v);
+$parcel$export($49c87ee97f392268$exports, "type-list", () => $49c87ee97f392268$export$915fe2ca68846e4c, (v) => $49c87ee97f392268$export$915fe2ca68846e4c = v);
+$parcel$export($49c87ee97f392268$exports, "type-tag", () => $49c87ee97f392268$export$5e5acbc770fb0e4, (v) => $49c87ee97f392268$export$5e5acbc770fb0e4 = v);
+var $49c87ee97f392268$export$cb889aaddfa2926c;
+var $49c87ee97f392268$export$5f8f750fd71deeed;
+var $49c87ee97f392268$export$42e8b5293e69631e;
+var $49c87ee97f392268$export$a0a23f31d5562853;
+var $49c87ee97f392268$export$6d59db4903f20f7d;
+var $49c87ee97f392268$export$38f2cd12e45697e;
+var $49c87ee97f392268$export$579bd45c78bf5510;
+var $49c87ee97f392268$export$23889d205b405f6e;
+var $49c87ee97f392268$export$890489be1870bed5;
+var $49c87ee97f392268$export$6bd0fd0a746c88f;
+var $49c87ee97f392268$export$55a1f87a143fb302;
+var $49c87ee97f392268$export$e4043f67d36cbb26;
+var $49c87ee97f392268$export$df03f54e09e486fa;
+var $49c87ee97f392268$export$35059013cd4a06db;
+var $49c87ee97f392268$export$75d14899fb692bb6;
+var $49c87ee97f392268$export$6a1093e41947f68e;
+var $49c87ee97f392268$export$7c414e944018fb15;
+var $49c87ee97f392268$export$d90250155de6d7e7;
+var $49c87ee97f392268$export$3ae1b51693c80a9c;
+var $49c87ee97f392268$export$bc7581b048bdc9d;
+var $49c87ee97f392268$export$fb184b623420d9be;
+var $49c87ee97f392268$export$b2d17fca990bb66c;
+var $49c87ee97f392268$export$915fe2ca68846e4c;
+var $49c87ee97f392268$export$5e5acbc770fb0e4;
+$49c87ee97f392268$export$cb889aaddfa2926c = `QoX8-W_add-type`;
+$49c87ee97f392268$export$5f8f750fd71deeed = `QoX8-W_add-type-overlay`;
+$49c87ee97f392268$export$42e8b5293e69631e = `QoX8-W_control-content`;
+$49c87ee97f392268$export$a0a23f31d5562853 = `QoX8-W_control-panel`;
+$49c87ee97f392268$export$6d59db4903f20f7d = `QoX8-W_description`;
+$49c87ee97f392268$export$38f2cd12e45697e = `QoX8-W_entity-panel`;
+$49c87ee97f392268$export$579bd45c78bf5510 = `QoX8-W_feedback-component`;
+$49c87ee97f392268$export$23889d205b405f6e = `QoX8-W_feedback-container`;
+$49c87ee97f392268$export$890489be1870bed5 = `QoX8-W_feedback-text`;
+$49c87ee97f392268$export$6bd0fd0a746c88f = `QoX8-W_feedback-text-wrapper`;
+$49c87ee97f392268$export$55a1f87a143fb302 = `QoX8-W_form-group`;
+$49c87ee97f392268$export$e4043f67d36cbb26 = `QoX8-W_highlight`;
+$49c87ee97f392268$export$df03f54e09e486fa = `QoX8-W_icons`;
+$49c87ee97f392268$export$35059013cd4a06db = `QoX8-W_node`;
+$49c87ee97f392268$export$75d14899fb692bb6 = `QoX8-W_node-label`;
+$49c87ee97f392268$export$6a1093e41947f68e = `QoX8-W_overlay-container`;
+$49c87ee97f392268$export$7c414e944018fb15 = `QoX8-W_page-wrapper`;
+$49c87ee97f392268$export$d90250155de6d7e7 = `QoX8-W_selected`;
+$49c87ee97f392268$export$3ae1b51693c80a9c = `QoX8-W_selection-tree`;
+$49c87ee97f392268$export$bc7581b048bdc9d = `QoX8-W_text-inputs`;
+$49c87ee97f392268$export$fb184b623420d9be = `QoX8-W_title`;
+$49c87ee97f392268$export$b2d17fca990bb66c = `QoX8-W_type-container`;
+$49c87ee97f392268$export$915fe2ca68846e4c = `QoX8-W_type-list`;
+$49c87ee97f392268$export$5e5acbc770fb0e4 = `QoX8-W_type-tag`;
+
+
+export {$49c87ee97f392268$exports as default};
+//# sourceMappingURL=feedback-components.95dbe7d7.js.map

package/dist/esm/feedback-components.95dbe7d7.js.map

@@ -0,0 +1 @@
+{"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AAvBA,4CAA6B,CAAC,eAAe,CAAC;AAC9C,4CAAqC,CAAC,uBAAuB,CAAC;AAC9D,4CAAoC,CAAC,sBAAsB,CAAC;AAC5D,4CAAkC,CAAC,oBAAoB,CAAC;AACxD,4CAAgC,CAAC,kBAAkB,CAAC;AACpD,2CAAiC,CAAC,mBAAmB,CAAC;AACtD,4CAAuC,CAAC,yBAAyB,CAAC;AAClE,4CAAuC,CAAC,yBAAyB,CAAC;AAClE,4CAAkC,CAAC,oBAAoB,CAAC;AACxD,2CAA0C,CAAC,4BAA4B,CAAC;AACxE,4CAA+B,CAAC,iBAAiB,CAAC;AAClD,4CAA8B,CAAC,gBAAgB,CAAC;AAChD,4CAA0B,CAAC,YAAY,CAAC;AACxC,4CAAyB,CAAC,WAAW,CAAC;AACtC,4CAA+B,CAAC,iBAAiB,CAAC;AAClD,4CAAsC,CAAC,wBAAwB,CAAC;AAChE,4CAAiC,CAAC,mBAAmB,CAAC;AACtD,4CAA6B,CAAC,eAAe,CAAC;AAC9C,4CAAmC,CAAC,qBAAqB,CAAC;AAC1D,2CAAgC,CAAC,kBAAkB,CAAC;AACpD,4CAA0B,CAAC,YAAY,CAAC;AACxC,4CAAmC,CAAC,qBAAqB,CAAC;AAC1D,4CAA8B,CAAC,gBAAgB,CAAC;AAChD,2CAA6B,CAAC,eAAe,CAAC","sources":["packages/feedback-components/src/feedback/feedback.module.sass"],"sourcesContent":[":root\n  --text-line-height: 3em\n  --main-extra-width: 200px\n\n.page-wrapper\n  display: flex\n  flex-direction: row\n  position: relative\n  gap: 2em\n  align-items: flex-start // makes control-content lose stick\n\n.feedback-container\n  flex: 4\n\n.control-panel\n  flex: 1\n  height: auto\n\n.control-content\n  position: sticky\n  top: 0\n\n.feedback-component\n  position: relative\n  width: 800px\n\n  & > svg\n    width: 800px\n\n.node\n  cursor: pointer\n\ncircle\n  cursor: pointer\n  border: 1px solid black\n\n.selected\n  border: 1px solid white\n  \n.feedback-text\n  margin-bottom: 2em\n\n.entity-panel\n  position: relative\n  max-height: 600px\n  width: calc(100% - 2em)\n  flex: 1\n  min-height: 100px\n  padding: 1em\n  background: var(--panel-secondary-background-color)\n  border-radius: 4px\n  // Inset box shadow\n  box-shadow: 0 0 0 1px var(--panel-border-color) inset\n\n.selection-tree\n  margin: -1em 0\n  padding: 1em 0\n  \n.type-list\n  display: grid\n  grid-auto-flow: column\n  grid-template-rows: repeat(10, auto)\n  gap: 0.2em\n\n  .type-tag\n    padding: .2em .5em\n    border-radius: .2em\n\n.description\n  max-width: 300px\n  padding: .5em\n\nmark\n  border-radius: .2em\n  cursor: pointer\n  color: black !important\n\n[role=\"treeitem\"]\n  width: auto !important\n\n.highlight\n  cursor: pointer\n  padding: .2em 0\n  border-radius: .2em\n  position: relative\n  zIndex: 10\n\n.feedback-text-wrapper \n  position: relative\n  z-index: 0\n  overflow: visible\n  line-height: var(--text-line-height)\n\n.type-container\n  display: flex\n  justify-content: space-between\n  align-items: center\n  column-gap: 1em\n\n.add-type\n  cursor: pointer\n  display: flex\n  justify-content: space-between\n  padding: 0 .5em\n  align-items: center\n\n  p\n    margin: 0\n\n.overlay-container\n  height: 80vh\n  width: 100vw\n  display: flex\n  justify-content: center\n  align-items: center\n\n  .add-type-overlay\n    background-color: var(--secondary-color)\n    padding: .5em 1em\n    display: flex\n    flex-direction: column\n    gap: 1em\n    border-radius: .2em\n\n    .title\n      display: flex\n      justify-content: space-between\n      align-items: center\n\n    h2\n      margin: 0\n\n    .form-group\n      display: flex\n      gap: 3em\n\n      .text-inputs\n        display: flex\n        flex-direction: column\n        gap: 1em\n\n.icons\n  display: flex\n  gap: .25em\n\n.node-label\n  fill: var(--text-emphasized-color)\n  fontSize: 10px\n  pointerEvents: none"],"names":[],"version":3,"file":"feedback-components.95dbe7d7.js.map"}

package/dist/esm/{type-selector.6e8952d6.js → feedback-components.ad9f284e.js}

@@ -1,5 +1,5 @@
-import "./main.module.8d366b6e.css";
-import $N2zU5$mainmoduled2fbdf09js from "./main.module.d2fbdf09.js";
+import "./feedback-components.bf93773c.css";
+import $N2zU5$feedbackcomponentsf9850d85js from "./feedback-components.f9850d85.js";
 import $N2zU5$macrostrathyper from "@macrostrat/hyper";
 import $N2zU5$classnames from "classnames";
 import {Omnibar as $N2zU5$Omnibar} from "@blueprintjs/select";
@@ -16,7 +16,7 @@ function $parcel$interopDefault(a) {
 
 
 
-const $fda9ef5406c1cfb4$var$h = (0, $N2zU5$macrostrathyper).styled((0, ($parcel$interopDefault($N2zU5$mainmoduled2fbdf09js))));
+const $fda9ef5406c1cfb4$var$h = (0, $N2zU5$macrostrathyper).styled((0, ($parcel$interopDefault($N2zU5$feedbackcomponentsf9850d85js))));
 const $fda9ef5406c1cfb4$var$TagListItem = (props)=>{
     /** Render a tag for the omnibox list */ let { active: active, selected: selected, className: className, onSelect: onSelect, item: item, children: children } = props;
     className = (0, $N2zU5$classnames)({
@@ -38,10 +38,11 @@ const $fda9ef5406c1cfb4$var$TagListItem = (props)=>{
     ]);
 };
 function $fda9ef5406c1cfb4$export$d8660660a589068c(props) {
-    /** A general omnibox for annotation types */ const { onSelectItem: onSelectItem, items: items, isOpen: isOpen, onClose: onClose } = props;
+    /** A general omnibox for annotation types */ const { onSelectItem: onSelectItem, items: items, isOpen: isOpen, onClose: onClose, onQueryChange: onQueryChange } = props;
     return $fda9ef5406c1cfb4$var$h((0, $N2zU5$Omnibar), {
         onItemSelect: onSelectItem,
         items: items,
+        onQueryChange: onQueryChange,
         resetOnSelect: false,
         isOpen: isOpen,
         onClose: onClose,
@@ -59,4 +60,4 @@ function $fda9ef5406c1cfb4$export$d8660660a589068c(props) {
 
 
 export {$fda9ef5406c1cfb4$export$d8660660a589068c as OmniboxSelector};
-//# sourceMappingURL=type-selector.6e8952d6.js.map
+//# sourceMappingURL=feedback-components.ad9f284e.js.map

package/dist/esm/feedback-components.ad9f284e.js.map

@@ -0,0 +1 @@
+{"mappings":";;;;;;;;;;;AAAA;;CAEC;;;;;AAUD,MAAM,0BAAI,CAAA,GAAA,sBAAI,EAAE,MAAM,CAAC,CAAA,GAAA,8EAAK;AAa5B,MAAM,oCAAoD,CAAC;IACzD,sCAAsC,GACtC,IAAI,UAAE,MAAM,YAAE,QAAQ,aAAE,SAAS,YAAE,QAAQ,QAAE,IAAI,YAAE,QAAQ,EAAE,GAAG;IAChE,YAAY,CAAA,GAAA,iBAAS,EAAE;gBAAE;kBAAQ;IAAS,GAAG;IAC7C,MAAM,UAAU,IAAM,SAAS;IAE/B,OAAO,wBACL,sBACA;QACE,KAAK,KAAK,EAAE;mBACZ;iBACA;IACF,GACA;QACE,wBAAE,cAAc;YAAE,OAAO;gBAAE,iBAAiB,KAAK,KAAK;YAAC;QAAE;QACzD,wBAAE,YAAY,CAAC,GAAG,KAAK,IAAI;KAC5B;AAEL;AAYO,SAAS,0CAAmB,KAAsB;IACvD,2CAA2C,GAC3C,MAAM,gBAAE,YAAY,SAAE,KAAK,UAAE,MAAM,WAAE,OAAO,iBAAE,aAAa,EAAE,GAAG;IAEhE,OAAO,wBAAE,CAAA,GAAA,cAAM,GAAG;QAChB,cAAc;eACd;uBACA;QACA,eAAe;gBACf;iBACA;QACA,cAAa,IAAO,EAAE,eAAE,WAAW,aAAE,SAAS,EAAE;YAC9C,OAAO,wBAAE,mCAAa;gBACpB,KAAK,KAAK,EAAE;sBACZ;gBACA,UAAU;gBACV,QAAQ,UAAU,MAAM;gBACxB,UAAU,UAAU,MAAM;YAC5B;QACF;IACF;AACF","sources":["packages/feedback-components/src/feedback/type-selector/index.ts"],"sourcesContent":["/**\n * Entity type selector\n */\n\nimport styles from \"./main.module.sass\";\nimport hyper from \"@macrostrat/hyper\";\n\nimport classNames from \"classnames\";\nimport React from \"react\";\nimport { Omnibar, OmnibarProps } from \"@blueprintjs/select\";\nimport \"@blueprintjs/select/lib/css/blueprint-select.css\";\n\nconst h = hyper.styled(styles);\n\ninterface TagItemProps<T> {\n  selected: boolean;\n  active: boolean;\n  className?: string;\n  item: T;\n\n  onSelect(t: T): void;\n\n  children?: React.ReactElement;\n}\n\nconst TagListItem: React.ComponentType<TagItemProps<T>> = (props) => {\n  /** Render a tag for the omnibox list */\n  let { active, selected, className, onSelect, item, children } = props;\n  className = classNames({ active, selected }, className);\n  const onClick = () => onSelect(item);\n\n  return h(\n    \"div.item-container\",\n    {\n      key: item.id,\n      className,\n      onClick,\n    },\n    [\n      h(\"div.swatch\", { style: { backgroundColor: item.color } }),\n      h(\"div.item\", {}, item.name),\n    ],\n  );\n};\n\ntype BoxLifecycleProps<T> = Pick<OmnibarProps<T>, \"onClose\" | \"isOpen\">;\n\ninterface OmniboxProps<T> extends BoxLifecycleProps<T> {\n  items: T[];\n  selectedItem: T;\n  onSelectItem: (t: T) => void;\n  onQueryChange: (query: string) => void;\n  listItemComponent?: React.ComponentType<TagItemProps<T>>;\n}\n\nexport function OmniboxSelector<T>(props: OmniboxProps<T>) {\n  /** A general omnibox for annotation types */\n  const { onSelectItem, items, isOpen, onClose, onQueryChange } = props;\n\n  return h(Omnibar, {\n    onItemSelect: onSelectItem,\n    items,\n    onQueryChange,\n    resetOnSelect: false,\n    isOpen,\n    onClose,\n    itemRenderer(item: T, { handleClick, modifiers }) {\n      return h(TagListItem, {\n        key: item.id,\n        item,\n        onSelect: handleClick,\n        active: modifiers.active,\n        selected: modifiers.active,\n      });\n    },\n  });\n}\n"],"names":[],"version":3,"file":"feedback-components.ad9f284e.js.map"}

package/dist/esm/{main.module.8d366b6e.css → feedback-components.bf93773c.css}

@@ -14,4 +14,4 @@
   width: 30px;
   height: 30px;
 }
-/*# sourceMappingURL=main.module.8d366b6e.css.map */
+/*# sourceMappingURL=feedback-components.bf93773c.css.map */

package/dist/esm/{main.module.8d366b6e.css.map → feedback-components.bf93773c.css.map}

@@ -1 +1 @@
-{"mappings":"AAAA;;;;;;;AAME;;;;AAGF","sources":["packages/feedback-components/src/feedback/type-selector/main.module.sass"],"sourcesContent":[".item-container\n  min-height: 40px\n  display: flex\n  align-items: center\n  gap: 1em\n\n  &.selected\n    background-color: var(--panel-background-color)\n\n.swatch\n  width: 30px\n  height: 30px\n  border-radius: 4px\n"],"names":[],"version":3,"file":"main.module.8d366b6e.css.map"}
+{"mappings":"AAAA;;;;;;;AAME;;;;AAGF","sources":["packages/feedback-components/src/feedback/type-selector/main.module.sass"],"sourcesContent":[".item-container\n  min-height: 40px\n  display: flex\n  align-items: center\n  gap: 1em\n\n  &.selected\n    background-color: var(--panel-background-color)\n\n.swatch\n  width: 30px\n  height: 30px\n  border-radius: 4px\n"],"names":[],"version":3,"file":"feedback-components.bf93773c.css.map"}

package/dist/esm/{main.module.2f2972c8.css → feedback-components.e273ed5b.css}

@@ -11,4 +11,4 @@
   margin: .2em 0 .5em;
   padding-right: 3px;
 }
-/*# sourceMappingURL=main.module.2f2972c8.css.map */
+/*# sourceMappingURL=feedback-components.e273ed5b.css.map */

package/dist/esm/{main.module.2f2972c8.css.map → feedback-components.e273ed5b.css.map}

@@ -1 +1 @@
-{"mappings":"AAAA;;;;;AAIE;;;;AAGF","sources":["packages/feedback-components/src/extractions/main.module.sass"],"sourcesContent":[".entities\n  list-style: none\n  padding-left: 0\n\n  .entities ul\n    list-style: none\n\n.entity\n  margin: 0.2em 0 0.5em\n  padding-right: 3px\n"],"names":[],"version":3,"file":"main.module.2f2972c8.css.map"}
+{"mappings":"AAAA;;;;;AAIE;;;;AAGF","sources":["packages/feedback-components/src/extractions/main.module.sass"],"sourcesContent":[".entities\n  list-style: none\n  padding-left: 0\n\n  .entities ul\n    list-style: none\n\n.entity\n  margin: 0.2em 0 0.5em\n  padding-right: 3px\n"],"names":[],"version":3,"file":"feedback-components.e273ed5b.css.map"}

package/dist/esm/{main.module.d2fbdf09.js → feedback-components.f9850d85.js}

@@ -16,4 +16,4 @@ $f8bcd91cbf17e53f$export$c672d9950c40fe0a = `WofFTa_swatch`;
 
 
 export {$f8bcd91cbf17e53f$exports as default};
-//# sourceMappingURL=main.module.d2fbdf09.js.map
+//# sourceMappingURL=feedback-components.f9850d85.js.map

package/dist/esm/{main.module.d2fbdf09.js.map → feedback-components.f9850d85.js.map}

@@ -1 +1 @@
-{"mappings":";;;;;;;;;AAAA,IAAA;AACA,IAAA;AACA,IAAA;AAFA,4CAAmC,CAAC,qBAAqB,CAAC;AAC1D,4CAA6B,CAAC,eAAe,CAAC;AAC9C,4CAA2B,CAAC,aAAa,CAAC","sources":["packages/feedback-components/src/feedback/type-selector/main.module.sass"],"sourcesContent":[".item-container\n  min-height: 40px\n  display: flex\n  align-items: center\n  gap: 1em\n\n  &.selected\n    background-color: var(--panel-background-color)\n\n.swatch\n  width: 30px\n  height: 30px\n  border-radius: 4px\n"],"names":[],"version":3,"file":"main.module.d2fbdf09.js.map"}
+{"mappings":";;;;;;;;;AAAA,IAAA;AACA,IAAA;AACA,IAAA;AAFA,4CAAmC,CAAC,qBAAqB,CAAC;AAC1D,4CAA6B,CAAC,eAAe,CAAC;AAC9C,4CAA2B,CAAC,aAAa,CAAC","sources":["packages/feedback-components/src/feedback/type-selector/main.module.sass"],"sourcesContent":[".item-container\n  min-height: 40px\n  display: flex\n  align-items: center\n  gap: 1em\n\n  &.selected\n    background-color: var(--panel-background-color)\n\n.swatch\n  width: 30px\n  height: 30px\n  border-radius: 4px\n"],"names":[],"version":3,"file":"feedback-components.f9850d85.js.map"}

package/dist/esm/{edit-state.c39d8466.js → feedback-components.fa1d3641.js}

@@ -10,6 +10,7 @@ var $b79bf29960412ca7$export$53ec42062883b2d4 = /*#__PURE__*/ function(ViewMode)
 }({});
 function $b79bf29960412ca7$export$911bb4b9c8065d3d(initialTree, entityTypes) {
     // Get the first entity type
+    // issue: grabs second entity instead of selected one
     const type = entityTypes.values().next().value;
     return (0, $abOVY$useReducer)($b79bf29960412ca7$var$treeReducer, {
         initialTree: initialTree,
@@ -29,8 +30,69 @@ function $b79bf29960412ca7$export$e1068f2d1c68f87e() {
     return dispatch;
 }
 function $b79bf29960412ca7$var$treeReducer(state, action) {
-    console.log(action);
     switch(action.type){
+        case "add-entity-type":
+            {
+                // Add a new entity type to the map
+                const { name: name, description: description, color: color } = action.payload;
+                const newId = state.lastInternalId - 1;
+                const newType = {
+                    id: newId,
+                    name: name,
+                    description: description === "" ? null : description,
+                    color: color
+                };
+                const newEntityTypesMap = new Map(state.entityTypesMap);
+                newEntityTypesMap.set(newId, newType);
+                return {
+                    ...state,
+                    entityTypesMap: newEntityTypesMap,
+                    selectedEntityType: newType,
+                    lastInternalId: newId
+                };
+            }
+        case "update-entity-type":
+            {
+                // Update an existing entity type in the map
+                const { id: id, name: name, description: description, color: 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: name,
+                    description: description === "" ? null : description,
+                    color: color
+                };
+                newEntityTypesMap.set(id, updatedType);
+                // Update the tree to reflect the new type
+                const newTree = $b79bf29960412ca7$var$updateTreeTypes(state.tree, oldType, updatedType);
+                return {
+                    ...state,
+                    tree: newTree,
+                    entityTypesMap: newEntityTypesMap,
+                    selectedEntityType: updatedType
+                };
+            }
+        case "select-range":
+            // Select a range of nodes by their IDs
+            const payloadIds = action.payload.ids;
+            const node1 = payloadIds[0];
+            const node2 = payloadIds[1];
+            // make list of nodes in order
+            const allNodes = $b79bf29960412ca7$var$flattenAndSort(state.tree);
+            // select all nodes between node1 and node2
+            const startIndex = allNodes.findIndex((node)=>node.id === node1);
+            const endIndex = allNodes.findIndex((node)=>node.id === node2);
+            const selectedNodes = allNodes.slice(startIndex, endIndex + 1);
+            console.log("Selecting range:", selectedNodes);
+            return {
+                ...state,
+                selectedNodes: selectedNodes.map((node)=>node.id)
+            };
         case "move-node":
             // For each node in the tree, if the node is in the dragIds, remove it from the tree and collect it
             const [newTree, removedNodes] = $b79bf29960412ca7$var$removeNodes(state.tree, action.payload.dragIds);
@@ -70,20 +132,24 @@ function $b79bf29960412ca7$var$treeReducer(state, action) {
             };
         case "select-node":
             const { ids: ids } = action.payload;
+            const type = action.payload.ids.length > 0 ? $b79bf29960412ca7$var$findNodeById(state.tree, ids[0])?.type : null;
             return {
                 ...state,
-                selectedNodes: ids
+                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 ? $b79bf29960412ca7$var$findNodeById(state.tree, action.payload.ids[0])?.type : null;
             return {
                 ...state,
                 selectedNodes: [
                     ...nodesToKeep,
                     ...nodesToAdd
-                ]
+                ],
+                selectedEntityType: newType
             };
         case "create-node":
             const newId = state.lastInternalId - 1;
@@ -109,6 +175,23 @@ function $b79bf29960412ca7$var$treeReducer(state, action) {
                 ],
                 lastInternalId: newId
             };
+        case "delete-entity-type":
+            {
+                // Remove the entity type from the map
+                console.log("Deleting entity type:", action.payload.id);
+                const { id: id } = action.payload;
+                const newEntityTypesMap = new Map(state.entityTypesMap);
+                const oldType = newEntityTypesMap.get(id);
+                newEntityTypesMap.delete(id);
+                const defaultType = newEntityTypesMap.values().next().value;
+                const newTree = $b79bf29960412ca7$var$updateTreeTypes(state.tree, oldType, defaultType);
+                return {
+                    ...state,
+                    tree: newTree,
+                    entityTypesMap: newEntityTypesMap,
+                    selectedNodes: []
+                };
+            }
         /** Entity type selection */ case "toggle-entity-type-selector":
             return {
                 ...state,
@@ -216,16 +299,18 @@ function $b79bf29960412ca7$export$8d9dbb7a64bf2a5e(tree) {
         // If we've already found an instance of this node, we don't need to record
         // it again
         if (nodeMap.has(node.id)) continue;
-        const { indices: indices, id: id, name: name } = node;
+        const { indices: indices, id: id, name: name, type: type, children: children } = node;
         const nodeData = {
             id: id,
-            type: node.type.id,
+            type: type.id,
+            color: type.color,
             name: name,
             txt_range: [
                 indices
             ],
             reasoning: null,
-            match: node.match
+            match: node.match,
+            children: children
         };
         nodeMap.set(node.id, node);
         nodes.push(nodeData);
@@ -245,7 +330,40 @@ function $b79bf29960412ca7$export$8d9dbb7a64bf2a5e(tree) {
         edges: edges
     };
 }
+function $b79bf29960412ca7$var$findNodeById(tree, id) {
+    for (const node of tree){
+        if (node.id === id) return node;
+        if (node.children) {
+            const found = $b79bf29960412ca7$var$findNodeById(node.children, id);
+            if (found) return found;
+        }
+    }
+    return null;
+}
+function $b79bf29960412ca7$var$updateTreeTypes(tree, oldType, defaultType) {
+    return tree.map((node)=>$b79bf29960412ca7$var$updateNodeType(node, oldType, defaultType));
+}
+function $b79bf29960412ca7$var$updateNodeType(node, oldType, defaultType) {
+    const type = node.type.id === oldType.id ? defaultType : node.type;
+    return {
+        ...node,
+        type: type,
+        children: node.children ? $b79bf29960412ca7$var$updateTreeTypes(node.children, oldType, defaultType) : []
+    };
+}
+function $b79bf29960412ca7$var$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);
+    // sort by start
+    return result.sort((a, b)=>a.indices[0] - b.indices[0]);
+}
 
 
 export {$b79bf29960412ca7$export$53ec42062883b2d4 as ViewMode, $b79bf29960412ca7$export$911bb4b9c8065d3d as useUpdatableTree, $b79bf29960412ca7$export$e5ce04c5b3f58533 as TreeDispatchContext, $b79bf29960412ca7$export$e1068f2d1c68f87e as useTreeDispatch, $b79bf29960412ca7$export$8d9dbb7a64bf2a5e as treeToGraph};
-//# sourceMappingURL=edit-state.c39d8466.js.map
+//# sourceMappingURL=feedback-components.fa1d3641.js.map

package/dist/esm/feedback-components.fa1d3641.js.map

@@ -0,0 +1 @@
+{"mappings":";;;;;AAKO,IAAA,AAAK,mEAAA;;;WAAA;;AAiDL,SAAS,0CACd,WAAuB,EACvB,WAAoC;IAEpC,4BAA4B;IAC5B,qDAAqD;IACrD,MAAM,OAAO,YAAY,MAAM,GAAG,IAAI,GAAG,KAAK;IAE9C,OAAO,CAAA,GAAA,iBAAS,EAAE,mCAAa;qBAC7B;QACA,MAAM;QACN,eAAe,EAAE;QACjB,gBAAgB;QAChB,oBAAoB;QACpB,gBAAgB;QAChB,uBAAuB;QACvB,QAAQ;IACV;AACF;AAEO,MAAM,4CAAsB,CAAA,GAAA,oBAAY,EAAuB;AAE/D,SAAS;IACd,MAAM,WAAW,CAAA,GAAA,iBAAS,EAAE;IAC5B,IAAI,YAAY,MACd,MAAM,IAAI,MAAM;IAElB,OAAO;AACT;AAEA,SAAS,kCAAY,KAAgB,EAAE,MAAkB;IACvD,OAAQ,OAAO,IAAI;QACjB,KAAK;YAAmB;gBACtB,mCAAmC;gBACnC,MAAM,QAAE,IAAI,eAAE,WAAW,SAAE,KAAK,EAAE,GAAG,OAAO,OAAO;gBACnD,MAAM,QAAQ,MAAM,cAAc,GAAG;gBACrC,MAAM,UAAsB;oBAC1B,IAAI;0BACJ;oBACA,aAAa,gBAAgB,KAAK,OAAO;2BACzC;gBACF;gBAEA,MAAM,oBAAoB,IAAI,IAAI,MAAM,cAAc;gBACtD,kBAAkB,GAAG,CAAC,OAAO;gBAE7B,OAAO;oBACL,GAAG,KAAK;oBACR,gBAAgB;oBAChB,oBAAoB;oBACpB,gBAAgB;gBAClB;YACF;QACA,KAAK;YAAsB;gBACzB,4CAA4C;gBAC5C,MAAM,MAAE,EAAE,QAAE,IAAI,eAAE,WAAW,SAAE,KAAK,EAAE,GAAG,OAAO,OAAO;gBACvD,MAAM,oBAAoB,IAAI,IAAI,MAAM,cAAc;gBACtD,MAAM,UAAU,kBAAkB,GAAG,CAAC;gBAEtC,IAAI,CAAC,SAAS;oBACZ,QAAQ,IAAI,CAAC,CAAC,oBAAoB,EAAE,GAAG,UAAU,CAAC;oBAClD,OAAO;gBACT;gBAEA,MAAM,cAA0B;oBAC9B,GAAG,OAAO;0BACV;oBACA,aAAa,gBAAgB,KAAK,OAAO;2BACzC;gBACF;gBAEA,kBAAkB,GAAG,CAAC,IAAI;gBAE1B,0CAA0C;gBAC1C,MAAM,UAAU,sCAAgB,MAAM,IAAI,EAAE,SAAS;gBAErD,OAAO;oBACL,GAAG,KAAK;oBACR,MAAM;oBACN,gBAAgB;oBAChB,oBAAoB;gBACtB;YACF;QACA,KAAK;YACH,uCAAuC;YACvC,MAAM,aAAa,OAAO,OAAO,CAAC,GAAG;YACrC,MAAM,QAAQ,UAAU,CAAC,EAAE;YAC3B,MAAM,QAAQ,UAAU,CAAC,EAAE;YAE3B,8BAA8B;YAC9B,MAAM,WAAW,qCAAe,MAAM,IAAI;YAE1C,2CAA2C;YAC3C,MAAM,aAAa,SAAS,SAAS,CAAC,CAAC,OAAS,KAAK,EAAE,KAAK;YAC5D,MAAM,WAAW,SAAS,SAAS,CAAC,CAAC,OAAS,KAAK,EAAE,KAAK;YAE1D,MAAM,gBAAgB,SAAS,KAAK,CAAC,YAAY,WAAW;YAE5D,QAAQ,GAAG,CAAC,oBAAoB;YAChC,OAAO;gBACL,GAAG,KAAK;gBACR,eAAe,cAAc,GAAG,CAAC,CAAC,OAAS,KAAK,EAAE;YACpD;QAEF,KAAK;YACH,mGAAmG;YACnG,MAAM,CAAC,SAAS,aAAa,GAAG,kCAC9B,MAAM,IAAI,EACV,OAAO,OAAO,CAAC,OAAO;YAGxB,IAAI,UAAmC,EAAE;YACzC,IAAI,OAAO,OAAO,CAAC,QAAQ,EAAE;gBAC3B,UAAU,+BAAS,SAAS,OAAO,OAAO,CAAC,QAAQ;gBACnD,QAAQ,IAAI,CAAC;YACf;YAEA,4DAA4D;YAC5D,IAAI,aAAa,sCAAgB,SAAS;gBACxC,SAAS;oBAAC;wBAAC,OAAO,OAAO,CAAC,KAAK;wBAAE;2BAAM;qBAAa;iBAAC;YACvD;YAEA,OAAO;gBAAE,GAAG,KAAK;gBAAE,MAAM,CAAA,GAAA,yBAAK,EAAE,SAAS;YAAY;QACvD,KAAK;YACH,gFAAgF;YAChF,MAAM,CAAC,UAAU,cAAc,GAAG,kCAChC,MAAM,IAAI,EACV,OAAO,OAAO,CAAC,GAAG;YAEpB,oCAAoC;YACpC,iEAAiE;YAEjE,MAAM,WAAW,cACd,OAAO,CAAC,CAAC,OAAS,KAAK,QAAQ,IAAI,EAAE,EACrC,MAAM,CAAC,CAAC,QAAU,CAAC,mCAAa,UAAU,MAAM,EAAE;YAErD,sBAAsB;YAEtB,OAAO;gBACL,GAAG,KAAK;gBACR,MAAM;uBAAI;uBAAa;iBAAS;gBAChC,eAAe,MAAM,aAAa,CAAC,MAAM,CACvC,CAAC,KAAO,CAAC,OAAO,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC;YAEzC;QACF,KAAK;YACH,MAAM,OAAE,GAAG,EAAE,GAAG,OAAO,OAAO;YAE9B,MAAM,OACJ,OAAO,OAAO,CAAC,GAAG,CAAC,MAAM,GAAG,IACxB,mCAAa,MAAM,IAAI,EAAE,GAAG,CAAC,EAAE,GAAG,OAClC;YAEN,OAAO;gBAAE,GAAG,KAAK;gBAAE,eAAe;gBAAK,oBAAoB;YAAK;QAClE,iEAAiE;QACjE,KAAK;YACH,MAAM,aAAa,OAAO,OAAO,CAAC,GAAG,CAAC,MAAM,CAC1C,CAAC,KAAO,CAAC,MAAM,aAAa,CAAC,QAAQ,CAAC;YAExC,MAAM,cAAc,MAAM,aAAa,CAAC,MAAM,CAC5C,CAAC,KAAO,CAAC,OAAO,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC;YAGvC,MAAM,UACJ,OAAO,OAAO,CAAC,GAAG,CAAC,MAAM,GAAG,IACxB,mCAAa,MAAM,IAAI,EAAE,OAAO,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,OACjD;YAEN,OAAO;gBACL,GAAG,KAAK;gBACR,eAAe;uBAAI;uBAAgB;iBAAW;gBAC9C,oBAAoB;YACtB;QAEF,KAAK;YACH,MAAM,QAAQ,MAAM,cAAc,GAAG;YACrC,MAAM,QAAE,IAAI,SAAE,KAAK,OAAE,GAAG,EAAE,GAAG,OAAO,OAAO;YAC3C,MAAM,OAAiB;gBACrB,IAAI;gBACJ,MAAM;gBACN,UAAU,EAAE;gBACZ,SAAS;oBAAC;oBAAO;iBAAI;gBACrB,MAAM,MAAM,kBAAkB;YAChC;YAEA,OAAO;gBACL,GAAG,KAAK;gBACR,MAAM;uBAAI,MAAM,IAAI;oBAAE;iBAAK;gBAC3B,eAAe;oBAAC;iBAAM;gBACtB,gBAAgB;YAClB;QAEF,KAAK;YAAsB;gBACzB,sCAAsC;gBACtC,QAAQ,GAAG,CAAC,yBAAyB,OAAO,OAAO,CAAC,EAAE;gBACtD,MAAM,MAAE,EAAE,EAAE,GAAG,OAAO,OAAO;gBAC7B,MAAM,oBAAoB,IAAI,IAAI,MAAM,cAAc;gBACtD,MAAM,UAAU,kBAAkB,GAAG,CAAC;gBACtC,kBAAkB,MAAM,CAAC;gBAEzB,MAAM,cAAc,kBAAkB,MAAM,GAAG,IAAI,GAAG,KAAK;gBAC3D,MAAM,UAAU,sCAAgB,MAAM,IAAI,EAAE,SAAS;gBAErD,OAAO;oBACL,GAAG,KAAK;oBACR,MAAM;oBACN,gBAAgB;oBAChB,eAAe,EAAE;gBACnB;YACF;QAEA,0BAA0B,GAC1B,KAAK;YACH,OAAO;gBACL,GAAG,KAAK;gBACR,uBAAuB,OAAO,OAAO,IAAI,CAAC,MAAM,qBAAqB;YACvE;QACF,KAAK;YAAsB;gBACzB,0CAA0C;gBAC1C,IAAI,WAAW,MAAM,IAAI;gBACzB,KAAK,IAAI,MAAM,MAAM,aAAa,CAAE;oBAClC,MAAM,UAAU,+BAAS,MAAM,IAAI,EAAE;oBACrC,MAAM,aAAa,sCAAgB,SAAS;wBAC1C,MAAM;4BAAE,MAAM,OAAO,OAAO;wBAAC;oBAC/B;oBACA,WAAW,CAAA,GAAA,yBAAK,EAAE,UAAU;gBAC9B;gBAEA,OAAO;oBACL,GAAG,KAAK;oBACR,MAAM;oBACN,oBAAoB,OAAO,OAAO;gBACpC;YACF;QACA,KAAK;YACH,OAAO;gBAAE,GAAG,KAAK;gBAAE,eAAe,EAAE;YAAC;QACvC,KAAK;YACH,OAAO;gBACL,GAAG,KAAK;gBACR,MAAM,MAAM,WAAW;gBACvB,eAAe,EAAE;YACnB;QACF,KAAK;YACH,OAAO;gBAAE,GAAG,KAAK;gBAAE,UAAU,OAAO,OAAO;YAAC;IAChD;AACF;AAEA,SAAS,mCAAa,IAAgB,EAAE,EAAU;IAChD,KAAK,IAAI,QAAQ,KAAM;QACrB,IAAI,KAAK,EAAE,IAAI,IACb,OAAO;aACF,IAAI,KAAK,QAAQ,EAAE;YACxB,IAAI,mCAAa,KAAK,QAAQ,EAAE,KAC9B,OAAO;QAEX;IACF;IACA,OAAO;AACT;AAEA,SAAS,sCACP,OAAgC,EAChC,SAAoB;IAEpB,wCAAwC;IAExC,IAAI,OAAO;IACX,IAAK,IAAI,IAAI,QAAQ,MAAM,GAAG,GAAG,KAAK,GAAG,IACvC,OAAO;QAAE,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE;IAAK;IAE9B,OAAO;AACP,wFAAwF;AAC1F;AAEA,SAAS,+BACP,IAAgB,EAChB,EAAU;IAEV,mFAAmF;IACnF,IAAK,IAAI,IAAI,GAAG,IAAI,KAAK,MAAM,EAAE,IAAK;QACpC,IAAI,IAAI,CAAC,EAAE,CAAC,EAAE,IAAI,IAChB,OAAO;YAAC;SAAE;aACL,IAAI,IAAI,CAAC,EAAE,CAAC,QAAQ,EAAE;YAC3B,IAAI,OAAO,+BAAS,IAAI,CAAC,EAAE,CAAC,QAAQ,EAAE;YACtC,IAAI,QAAQ,MACV,OAAO;gBAAC;gBAAG;mBAAe;aAAK;QAEnC;IACF;IACA,OAAO;AACT;AAEA,SAAS,kCACP,IAAgB,EAChB,GAAa;IAEb,gGAAgG,GAChG,IAAI,UAAsB,EAAE;IAC5B,IAAI,eAA2B,EAAE;IAEjC,KAAK,IAAI,QAAQ,KACf,IAAI,IAAI,QAAQ,CAAC,KAAK,EAAE,GACtB,aAAa,IAAI,CAAC;SACb;QACL,wBAAwB;QACxB,IAAI,KAAK,QAAQ,EAAE;YACjB,IAAI,CAAC,aAAa,gBAAgB,GAAG,kCAAY,KAAK,QAAQ,EAAE;YAChE,OAAO;gBAAE,GAAG,IAAI;gBAAE,UAAU;YAAY;YACxC,aAAa,IAAI,IAAI;QACvB;QACA,QAAQ,IAAI,CAAC;IACf;IAGF,OAAO;QAAC;QAAS;KAAa;AAChC;AAkBO,SAAS,0CAAY,IAAgB;IAC1C,8BAA8B;IAC9B,IAAI,QAAwB,EAAE;IAC9B,IAAI,QAA4C,EAAE;IAClD,MAAM,UAAU,IAAI;IAEpB,KAAK,IAAI,QAAQ,KAAM;QACrB,2EAA2E;QAC3E,WAAW;QACX,IAAI,QAAQ,GAAG,CAAC,KAAK,EAAE,GACrB;QAGF,MAAM,WAAE,OAAO,MAAE,EAAE,QAAE,IAAI,QAAE,IAAI,YAAE,QAAQ,EAAE,GAAG;QAE9C,MAAM,WAAyB;gBAC7B;YACA,MAAM,KAAK,EAAE;YACb,OAAO,KAAK,KAAK;kBACjB;YACA,WAAW;gBAAC;aAAQ;YACpB,WAAW;YACX,OAAO,KAAK,KAAK;sBACjB;QACF;QAEA,QAAQ,GAAG,CAAC,KAAK,EAAE,EAAE;QACrB,MAAM,IAAI,CAAC;QAEX,IAAI,KAAK,QAAQ,EAAE;YACjB,KAAK,IAAI,SAAS,KAAK,QAAQ,CAC7B,MAAM,IAAI,CAAC;gBAAE,QAAQ,KAAK,EAAE;gBAAE,MAAM,MAAM,EAAE;YAAC;YAG/C,2BAA2B;YAC3B,MAAM,EAAE,OAAO,UAAU,EAAE,OAAO,UAAU,EAAE,GAAG,0CAC/C,KAAK,QAAQ;YAEf,MAAM,IAAI,IAAI;YACd,MAAM,IAAI,IAAI;QAChB;IACF;IAEA,OAAO;eAAE;eAAO;IAAM;AACxB;AAEA,SAAS,mCAAa,IAAI,EAAE,EAAE;IAC5B,KAAK,MAAM,QAAQ,KAAM;QACvB,IAAI,KAAK,EAAE,KAAK,IACd,OAAO;QAET,IAAI,KAAK,QAAQ,EAAE;YACjB,MAAM,QAAQ,mCAAa,KAAK,QAAQ,EAAE;YAC1C,IAAI,OAAO,OAAO;QACpB;IACF;IACA,OAAO;AACT;AAEA,SAAS,sCAAgB,IAAI,EAAE,OAAO,EAAE,WAAW;IACjD,OAAO,KAAK,GAAG,CAAC,CAAC,OAAS,qCAAe,MAAM,SAAS;AAC1D;AAEA,SAAS,qCAAe,IAAI,EAAE,OAAO,EAAE,WAAW;IAChD,MAAM,OAAO,KAAK,IAAI,CAAC,EAAE,KAAK,QAAQ,EAAE,GAAG,cAAc,KAAK,IAAI;IAElE,OAAO;QACL,GAAG,IAAI;cACP;QACA,UAAU,KAAK,QAAQ,GACnB,sCAAgB,KAAK,QAAQ,EAAE,SAAS,eACxC,EAAE;IACR;AACF;AAEA,SAAS,qCAAe,KAAK;IAC3B,MAAM,SAAS,EAAE;IAEjB,SAAS,SAAS,QAAQ;QACxB,KAAK,MAAM,QAAQ,SAAU;YAC3B,OAAO,IAAI,CAAC;YACZ,IAAI,MAAM,OAAO,CAAC,KAAK,QAAQ,KAAK,KAAK,QAAQ,CAAC,MAAM,GAAG,GACzD,SAAS,KAAK,QAAQ;QAE1B;IACF;IAEA,SAAS;IAET,gBAAgB;IAChB,OAAO,OAAO,IAAI,CAAC,CAAC,GAAG,IAAM,EAAE,OAAO,CAAC,EAAE,GAAG,EAAE,OAAO,CAAC,EAAE;AAC1D","sources":["packages/feedback-components/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}\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\nexport type TreeDispatch = Dispatch<TreeAction>;\n\nexport function useUpdatableTree(\n  initialTree: TreeData[],\n  entityTypes: Map<number, EntityType>,\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  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  });\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  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      console.log(\"Deleting entity type:\", action.payload.id);\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\nfunction 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"],"names":[],"version":3,"file":"feedback-components.fa1d3641.js.map"}