summaryrefslogtreecommitdiff
path: root/client/faceAnalysis/faceAnalysis.result.js
blob: 1c8a2ffbea2d4df57f6279d539c771abae18616c (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
import React, { Component } from 'react'
import { bindActionCreators } from 'redux'
import { connect } from 'react-redux'

import * as actions from './faceAnalysis.actions'
import { Loader } from '../common'

const errors = {
  bbox: (
    <div>
      <h2>No face found</h2>
      {"Sorry, we didn't detect a face in that image. "}
      {"Please choose an image where the face is large and clear."}
    </div>
  ),
  nomatch: (
    <div>
      <h2>{"You're clear"}</h2>
      {"No images in this dataset match your face.  We show only matches above 70% probability."}
    </div>
  ),
  error: (
    <div>
      <h2>{"No matches found"}</h2>
      {"Sorry, an error occured."}
    </div>
  ),
  not_an_image: (
    <div>
      <h2>{"Not an image"}</h2>
      {"Sorry, the file you uploaded was not recognized as an image. Please upload a JPG or PNG image and try again."}
    </div>
  ),
}

class FaceAnalysisResult extends Component {
  render() {
    const { result, timing } = this.props
    const { data, error, loading, message } = result
    let { step, total } = data || {}
    // console.log(step, total)
    if (loading) {
      return (
        <div className='result'>
          <div>
            <Loader /><br />
            <h2>Uploading...</h2>
          </div>
        </div>
      )
    }
    if (error) {
      // console.log(error)
      let errorMessage = errors[error] || errors.error
      return (
        <div className='result'>{errorMessage}</div>
      )
    }
    // console.log(result)
    if (!total) {
      return (
        <div className='result'></div>
      )
    }

    console.log(data.data)
    const results = [
      'blur_fn', 'points_3d_68', 'landmarks_3d_68', 'landmarks_2d_68', 'pose',
      'age_real', 'age_apparent', 'gender'
    ].map(tag => {
      if (tag in data.data) {
        const { title, url } = data.data[tag]
        return (
          <div key={tag}>
            <img src={url} />
            <span>{title}</span>
          </div>
        )        
      }
      return null
    }).filter(a => a)

    return (
      <div>
        {!(step && total && message) ? '' : (<span>Step {step} / {total}: {message}</span>)}
        <div className='results'>
          {results}
        </div>
        <div className="about">
          <small>Query took {(timing / 1000).toFixed(2)} s.</small>
        </div>
      </div>
    )
  }
}

const mapStateToProps = state => ({
  query: state.faceAnalysis.query,
  result: state.faceAnalysis.result,
  timing: state.faceAnalysis.timing,
  options: state.faceAnalysis.options,
})

const mapDispatchToProps = dispatch => ({
  actions: bindActionCreators({ ...actions }, dispatch),
})

export default connect(mapStateToProps, mapDispatchToProps)(FaceAnalysisResult)