-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathuseDetect.js
More file actions
61 lines (50 loc) · 1.38 KB
/
Copy pathuseDetect.js
File metadata and controls
61 lines (50 loc) · 1.38 KB
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
import { useState } from 'react'
export function useDetect() {
const [loading, setLoading] = useState(false)
const [error, setError] = useState(null)
const [results, setResults] = useState(null)
async function detect(imageFile, options = {}) {
setLoading(true)
setError(null)
setResults(null)
const {
useLlm = true,
topK = 5,
enableOcr = false,
confidenceThreshold = 0.25,
} = options
const params = new URLSearchParams({
use_llm: useLlm,
top_k: topK,
enable_ocr: enableOcr,
confidence_threshold: confidenceThreshold,
})
const formData = new FormData()
formData.append('image', imageFile)
const apiBase = import.meta.env.VITE_API_BASE_URL ?? '/api'
try {
const response = await fetch(`${apiBase}/detect-and-recommend?${params}`, {
method: 'POST',
body: formData,
})
if (!response.ok) {
const errBody = await response.text()
throw new Error(`API error ${response.status}: ${errBody}`)
}
const data = await response.json()
setResults(data)
return data
} catch (err) {
setError(err.message || 'Something went wrong')
throw err
} finally {
setLoading(false)
}
}
function reset() {
setResults(null)
setError(null)
setLoading(false)
}
return { detect, loading, error, results, reset }
}