forked from ReidWilliams/GANs
-
Notifications
You must be signed in to change notification settings - Fork 0
/
client_OLD.js
65 lines (56 loc) · 1.47 KB
/
client_OLD.js
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
import PyFiClient from 'pyfi-client'
const py = PyFiClient()
let imgSize = {}
const button = document.getElementById('button')
const button_family = document.getElementById('button_family')
const result = document.getElementById('result')
const input = document.getElementById('input')
button.disabled = true
py._.onReady(()=>{
// wait until PyFi is ready to allow clicks
button.disabled = false
console.log('PyFi Ready!')
py.get_size([])
.then(size => {
console.log(size)
imgSize = size
})
})
button.addEventListener("click", function(){
button.disabled = true
py.generate_random_chair([])
.then(res => {
console.log(res)
loadImage(res.file_name)
result.innerHTML = `success!`
button.disabled = false
})
.catch(error => {
console.log(error)
})
})
button_family.addEventListener("click", function(){
button_family.disabled = true
py.generate_random_chair([])
.then(res => {
console.log(res)
loadImage(res.file_name)
py.generate_similar_chairs([res.latent_vector])
.then(family => {
console.log(family)
family.file_names.forEach(function(element) {
loadImage(element)
})
})
familyResult.innerHTML = `success!`
button_family.disabled = false
})
.catch(error => {
console.log(error)
})
})
const loadImage = function(file_name){
let elem = document.createElement('img')
elem.setAttribute('src', `imgs/${file_name}`)
document.getElementById('imageFrame').appendChild(elem)
}