summaryrefslogtreecommitdiff
path: root/client/app.js
blob: ba77be65368c06d130ea4db7270fb2394676ffc7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
import React, { Component } from 'react'

import UploadImage from './lib/uploadImage.component'
import { post } from './util'

const initialState = {
  'image': null,
  'res': null,
  'loading': false,
}

export default class PhashApp extends Component {
  state = { ...initialState }

  upload(blob) {
    if (this.state.image) {
      URL.revokeObjectURL(this.state.image)
    }
    const url = URL.createObjectURL(blob)
    this.setState({ image: url, loading: true })

    const fd = new FormData()
    fd.append('q', blob)
    post('/api/v1/match', fd)
    .then(res => {
      console.log(res)
      this.setState({ res, loading: false })
    })
    .catch(err => {
      console.log(err)
      this.setState({ loading: false })
    })
  }

  render() {
    return (
      <div className='app'>
        <h1>Perceptual Hash Demo</h1>
        {this.renderQuery()}
        {this.renderResults()}
      </div>
    )
  }

  renderQuery() {
    const { image } = this.state
    const style = {}
    if (image) {
      style.backgroundImage = 'url(' + image + ')'
      style.backgroundSize = 'cover'
      style.opacity = 1
    }
    return (
      <div className='query'>
        <UploadImage onUpload={this.upload.bind(this)} />
        {image && <div style={style} />}
      </div>
    )
  }
  renderResults() {
    const { loading, res } = this.state
    if (!res) {
      return (
        <div className='results'>
        </div>
      )
    }
    if (loading) {
      return (
        <div className='results'>
          <i>Loading...</i>
        </div>
      )
    }

    const { success, error, match, closest_match } = res
    if (!success) {
      return (
        <div className='results'>
          <b>Error: {error}</b>
        </div>
      )
    }

    if (!match) {
      return (
        <div className='results'>
          No match, image added to database
        </div>
      )
    }

    const { ext, phash, score, sha256 } = closest_match
    return (
      <div className='results'>
        Closest match: {sha256}{'.'}{ext}<br />
        Score: {score}<br />
        Phash: {phash.toString(16)}
      </div>
    )
  }
}