import React from "react";
import { storiesOf } from "@storybook/react";
import { action } from "@storybook/addon-actions";
import TextClassification from "./";
storiesOf("TextClassification", module).add("Basic", function () {
  return /*#__PURE__*/React.createElement(TextClassification, {
    containerProps: {
      requireCompleteToPressNext: true
    },
    onModifySample: action("onModifySample"),
    interface: {
      type: "text_classification",
      description: "Classify the person name into the correct house.",
      labels: [{
        id: "gryffindor",
        displayName: "Gryffindor",
        description: "Daring, strong nerve and chivalry."
      }, {
        id: "slytherin",
        displayName: "Slytherin",
        description: "Cunning and ambitious. Possibly dark wizard."
      }]
    },
    sample: {
      document: "Harry",
      annotation: "gryffindor"
    }
  });
}).add("Multi", function () {
  return /*#__PURE__*/React.createElement(TextClassification, {
    containerProps: {
      requireCompleteToPressNext: true
    },
    onModifySample: action("onModifySample"),
    interface: {
      type: "text_classification",
      description: "Classify the person name into the correct house.",
      multiple: true,
      labels: [{
        id: "gryffindor",
        displayName: "Gryffindor",
        description: "Daring, strong nerve and chivalry."
      }, {
        id: "slytherin",
        displayName: "Slytherin",
        description: "Cunning and ambitious. Possibly dark wizard."
      }]
    },
    sample: {
      document: "Harry",
      annotation: ["slytherin", "gryffindor"]
    }
  });
});