initial relationships working

Author: seveibar

src/components/RelationshipAnnotator/index.js

@@ -29,20 +29,27 @@ const LabelSelectorContainer = styled("div")({ display: "flex" })
 export default function RelationshipAnnotator(
   props: RelationshipAnnotatorProps
 ) {
-  console.log({ props })
   const [highlightedItems, changeHighlightedItems] = useState([])
-  const [relationships, setRelationships] = useState(props.relationships || [])
+  const [relationships, setRelationships] = useState(
+    props.initialRelationships || []
+  )
   const [activePair, setActivePair] = useState(null)
   const [creatingRelationships, setCreatingRelationships] = useState(true)
-  const [sequence, changeSequence] = useState(() =>
-    props.initialSequence
+  const [sequence, changeSequence] = useState(() => {
+    const textIdsInRelationship = new Set(
+      relationships.flatMap(({ to, from }) => [to, from])
+    )
+    return props.initialSequence
       ? props.initialSequence.flatMap(entity =>
-          entity.label
+          entity.label ||
+          (entity.textId && textIdsInRelationship.has(entity.textId))
             ? [withId(entity)]
             : stringToSequence(entity.text, props.separatorRegex).map(withId)
         )
       : stringToSequence(props.document).map(withId)
-  )
+  })
+
+  console.log(sequence, relationships)
 
   const labels = creatingRelationships
     ? props.relationshipLabels

src/components/RelationshipAnnotator/index.story.js

@@ -45,11 +45,12 @@ storiesOf("RelationshipAnnotator", module)
       type="label-relationships"
       document={`Lorem ipsum dolor.`}
       initialSequence={[
-        { label: "noun", text: "Lorem" },
-        { text: " ipsum  " },
-        { label: "proper-noun", text: "dolor." },
-        { text: " this should be broken up into words." }
+        { label: "noun", text: "Lorem", textId: "lorem" },
+        { text: " ipsum  ", textId: "ipsum" },
+        { label: "proper-noun", text: "dolor.", textId: "dolor" },
+        { text: " this should be broken up into words.", textId: "therest" }
       ]}
+      initialRelationships={[{ from: "lorem", to: "ipsum", label: "eaten-by" }]}
       entityLabels={[
         {
           color: colors[0],