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],