summaryrefslogtreecommitdiff
path: root/client/app.js
blob: d18a5aa34038b930b788985712e9fb35a1a0863b (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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
import React, { Component } from 'react'

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

const initialState = {
  image: null,
  url: "",
  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 })
    })
  }

  submit() {
    const { url } = this.state
    if (!url || url.indexOf('http') !== 0) return
    this.setState({ image: url, loading: true })

    const fd = new FormData()
    fd.append('url', url)
    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'>
        <label>
          <span>Upload image</span>
          <UploadImage onUpload={this.upload.bind(this)} />
        </label>
        <label>
          <span>Enter URL</span>
          <input
            type='text'
            value={this.state.url}
            onChange={e => this.setState({ url: e.target.value })}
            onKeyDown={e => e.keyCode === 13 && this.submit()}
            placeholder='https://'
          />
        </label>
        {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 { phash, score, sha256, url} = closest_match
    return (
      <div className='results'>
        <img src={url} /><br />
        Closest match: {sha256}<br />
        Score: {score}<br />
        Phash: {phash.toString(16)}
      </div>
    )
  }
}