@searpent/react-image-annotate 2.0.2 → 2.0.5

package/src/DemoSite/Editor.js

@@ -0,0 +1,235 @@
+// @flow
+
+import React, { useState } from "react"
+import Button from "@mui/material/Button"
+import { makeStyles } from "@mui/styles"
+import { createTheme, ThemeProvider } from "@mui/material/styles"
+import Select from "react-select"
+import Code from "react-syntax-highlighter"
+import Dialog from "@mui/material/Dialog"
+import DialogTitle from "@mui/material/DialogTitle"
+import DialogContent from "@mui/material/DialogContent"
+import DialogActions from "@mui/material/DialogActions"
+import MonacoEditor from "react-monaco-editor"
+
+const theme = createTheme()
+const useStyles = makeStyles((theme) => ({
+  editBar: {
+    padding: 10,
+    borderBottom: "1px solid #ccc",
+    backgroundColor: "#f8f8f8",
+    display: "flex",
+    alignItems: "center",
+    "& .button": { margin: 5 },
+  },
+  select: { width: 240, fontSize: 14 },
+  contentArea: {
+    padding: 10,
+  },
+  specificationArea: {
+    padding: 10,
+  },
+}))
+
+const loadSavedInput = () => {
+  try {
+    return JSON.parse(window.localStorage.getItem("customInput") || "{}")
+  } catch (e) {
+    return {}
+  }
+}
+
+export const examples = {
+  "Simple Bounding Box": () => ({
+    taskDescription:
+      "Annotate each image according to this _markdown_ specification.",
+    // regionTagList: [],
+    // regionClsList: ["hotdog"],
+    regionTagList: ["has-bun"],
+    regionClsList: ["hotdog", "not-hotdog"],
+    enabledTools: ["select", "create-box"],
+    // showTags: true,
+    images: [
+      {
+        src: "https://images.unsplash.com/photo-1496905583330-eb54c7e5915a?ixlib=rb-1.2.1&ixid=eyJhcHBfaWQiOjEyMDd9&auto=format&fit=crop&w=634&q=80",
+        name: "hot-dogs-1",
+      },
+      {
+        src: "https://www.bianchi.com/wp-content/uploads/2019/07/YPB17I555K.jpg",
+        name: "bianchi-oltre-xr4",
+      },
+    ],
+    allowComments: true,
+  }),
+  "Simple Segmentation": () => ({
+    taskDescription:
+      "Annotate each image according to this _markdown_ specification.",
+    regionClsList: ["car", "truck"],
+    enabledTools: ["select", "create-polygon"],
+    images: [
+      {
+        src: "https://images.unsplash.com/photo-1561518776-e76a5e48f731?ixlib=rb-1.2.1&ixid=eyJhcHBfaWQiOjEyMDd9&auto=format&fit=crop&w=750&q=80",
+        name: "car-image-1",
+      },
+    ],
+  }),
+  Custom: () => loadSavedInput(),
+}
+
+const Editor = ({ onOpenAnnotator, lastOutput }: any) => {
+  const c = useStyles()
+  const [currentError, changeCurrentError] = useState()
+  const [selectedExample, changeSelectedExample] = useState(
+    window.localStorage.getItem("customInput")
+      ? "Custom"
+      : "Simple Bounding Box"
+  )
+  const [outputDialogOpen, changeOutputOpen] = useState(false)
+  const [currentJSONValue, changeCurrentJSONValue] = useState(
+    JSON.stringify(examples[selectedExample](), null, "  ")
+  )
+  return (
+    <ThemeProvider theme={theme}>
+      <div>
+        <div className={c.editBar}>
+          <h3>React Image Annotate</h3>
+          <div style={{ flexGrow: 1 }} />
+          <div>
+            <div style={{ display: "inline-flex" }}>
+              <Select
+                className={c.select}
+                value={{ label: selectedExample, value: selectedExample }}
+                options={Object.keys(examples).map((s) => ({
+                  label: s,
+                  value: s,
+                }))}
+                onChange={(selectedOption) => {
+                  changeSelectedExample(selectedOption.value)
+
+                  changeCurrentJSONValue(
+                    JSON.stringify(
+                      selectedOption.value === "Custom"
+                        ? loadSavedInput()
+                        : examples[selectedOption.value](),
+                      null,
+                      "  "
+                    )
+                  )
+                }}
+              />
+            </div>
+            <Button
+              className="button"
+              disabled={!lastOutput}
+              onClick={() => changeOutputOpen(true)}
+            >
+              View Output
+            </Button>
+            <Button
+              className="button"
+              variant="outlined"
+              disabled={Boolean(currentError)}
+              onClick={() => {
+                onOpenAnnotator(
+                  selectedExample === "Custom"
+                    ? loadSavedInput()
+                    : examples[selectedExample]
+                )
+              }}
+            >
+              Open Annotator
+            </Button>
+          </div>
+        </div>
+        <div
+          className={c.contentArea}
+          style={
+            currentError
+              ? { border: "2px solid #f00" }
+              : { border: "2px solid #fff" }
+          }
+        >
+          <div>
+            <MonacoEditor
+              value={currentJSONValue}
+              language="javascript"
+              onChange={(code) => {
+                try {
+                  window.localStorage.setItem(
+                    "customInput",
+                    JSON.stringify(JSON.parse(code))
+                  )
+                  changeCurrentError(null)
+                } catch (e) {
+                  changeCurrentError(e.toString())
+                }
+                changeCurrentJSONValue(code)
+              }}
+              width="100%"
+              height="550px"
+            />
+          </div>
+        </div>
+        <div className={c.specificationArea}>
+          <h2>React Image Annotate Format</h2>
+          <Code language="javascript">{`
+{
+  taskDescription?: string, // markdown
+  regionTagList?: Array<string>,
+  regionClsList?: Array<string>,
+  imageTagList?: Array<string>,
+  imageClsList?: Array<string>,
+  // all tools are enabled by default
+  enabledTools?: Array< "select" | "create-point" | "create-box" | "create-polygon" | "create-line">,
+  selectedImage?: string, // initial selected image
+  images: Array<{
+    src: string,
+    thumbnailSrc?: string, // use this if you are using high-res images
+    name: string,
+    regions?: Array<{
+      id: string | number,
+      cls?: string,
+      color?: string,
+      tags?: Array<string>,
+
+      // Point
+      type: "point",
+      x: number, // [0-1] % of image width
+      y: number, // [0-1] % of image height
+
+      // Bounding Box
+      type: "box",
+      x: number, // [0-1] % of image width
+      y: number, // [0-1] % of image height
+      w: number, // [0-1] % of image width
+      h: number, // [0-1] % of image height
+
+      // Polygon
+      type: "polygon",
+      open?: boolean, // should last and first points be connected, default: true
+      points: Array<[number, number]> // [0-1] % of image width/height
+    }>
+  }>,
+}
+`}</Code>
+        </div>
+        <Dialog fullScreen open={outputDialogOpen}>
+          <DialogTitle>React Image Annotate Output</DialogTitle>
+          <DialogContent style={{ minWidth: 400 }}>
+            <MonacoEditor
+              value={JSON.stringify(lastOutput, null, "  ")}
+              language="javascript"
+              width="100%"
+              height="550px"
+            />
+          </DialogContent>
+          <DialogActions>
+            <Button onClick={() => changeOutputOpen(false)}>Close</Button>
+          </DialogActions>
+        </Dialog>
+      </div>
+    </ThemeProvider>
+  )
+}
+
+export default Editor

package/src/DemoSite/ErrorBoundaryDialog.js

@@ -0,0 +1,34 @@
+// @flow
+
+import React, { Component } from "react"
+import Dialog from "@mui/material/Dialog"
+import Button from "@mui/material/Button"
+import DialogTitle from "@mui/material/DialogTitle"
+import DialogContent from "@mui/material/DialogContent"
+import DialogActions from "@mui/material/DialogActions"
+
+export default class ErrorBoundaryDialog extends Component {
+  state = { hasError: false, err: "" }
+  componentDidCatch(err, info) {
+    this.setState({
+      hasError: true,
+      err: err.toString() + "\n\n" + info.componentStack,
+    })
+  }
+  render() {
+    if (this.state.hasError) {
+      return (
+        <Dialog open={this.state.hasError} onClose={this.props.onClose}>
+          <DialogTitle>Error Loading Annotator</DialogTitle>
+          <DialogContent>
+            <pre>{this.state.err}</pre>
+          </DialogContent>
+          <DialogActions>
+            <Button onClick={this.props.onClose}>Close</Button>
+          </DialogActions>
+        </Dialog>
+      )
+    }
+    return this.props.children
+  }
+}

package/src/DemoSite/index.js

@@ -0,0 +1,41 @@
+// @flow
+import React, { useState } from "react"
+import ReactDOM from "react-dom"
+import Editor, { examples } from "./Editor"
+import Annotator from "../Annotator"
+import ErrorBoundaryDialog from "./ErrorBoundaryDialog.js"
+
+export default () => {
+  const [annotatorOpen, changeAnnotatorOpen] = useState(false)
+  const [annotatorProps, changeAnnotatorProps] = useState(examples["Custom"]())
+  const [lastOutput, changeLastOutput] = useState()
+
+  return (
+    <div>
+      {annotatorOpen ? (
+        <ErrorBoundaryDialog
+          onClose={() => {
+            changeAnnotatorOpen(false)
+          }}
+        >
+          <Annotator
+            {...(annotatorProps: any)}
+            onExit={(output) => {
+              delete (output: any)["lastAction"]
+              changeLastOutput(output)
+              changeAnnotatorOpen(false)
+            }}
+          />
+        </ErrorBoundaryDialog>
+      ) : (
+        <Editor
+          lastOutput={lastOutput}
+          onOpenAnnotator={(props) => {
+            changeAnnotatorProps(props)
+            changeAnnotatorOpen(true)
+          }}
+        />
+      )}
+    </div>
+  )
+}

package/src/DemoSite/index.story.js

@@ -0,0 +1,10 @@
+// @flow
+
+import React from "react"
+
+import { storiesOf } from "@storybook/react"
+import { action } from "@storybook/addon-actions"
+
+import DemoSite from "./"
+
+storiesOf("DemoSite", module).add("Basic", () => <DemoSite />)