88 lines
2.5 KiB
TypeScript
88 lines
2.5 KiB
TypeScript
import { Rect, Settings } from '../store/Atoms'
|
|
import { dataURItoBlob } from '../utils'
|
|
|
|
export const API_ENDPOINT = `${process.env.REACT_APP_INPAINTING_URL}`
|
|
|
|
export default async function inpaint(
|
|
imageFile: File,
|
|
maskBase64: string,
|
|
settings: Settings,
|
|
croperRect: Rect,
|
|
prompt?: string,
|
|
sizeLimit?: string
|
|
) {
|
|
// 1080, 2000, Original
|
|
const fd = new FormData()
|
|
fd.append('image', imageFile)
|
|
const mask = dataURItoBlob(maskBase64)
|
|
fd.append('mask', mask)
|
|
|
|
const hdSettings = settings.hdSettings[settings.model]
|
|
fd.append('ldmSteps', settings.ldmSteps.toString())
|
|
fd.append('ldmSampler', settings.ldmSampler.toString())
|
|
fd.append('zitsWireframe', settings.zitsWireframe.toString())
|
|
fd.append('hdStrategy', hdSettings.hdStrategy)
|
|
fd.append('hdStrategyCropMargin', hdSettings.hdStrategyCropMargin.toString())
|
|
fd.append(
|
|
'hdStrategyCropTrigerSize',
|
|
hdSettings.hdStrategyCropTrigerSize.toString()
|
|
)
|
|
fd.append(
|
|
'hdStrategyResizeLimit',
|
|
hdSettings.hdStrategyResizeLimit.toString()
|
|
)
|
|
|
|
fd.append('prompt', prompt === undefined ? '' : prompt)
|
|
fd.append('croperX', croperRect.x.toString())
|
|
fd.append('croperY', croperRect.y.toString())
|
|
fd.append('croperHeight', croperRect.height.toString())
|
|
fd.append('croperWidth', croperRect.width.toString())
|
|
fd.append('useCroper', settings.showCroper ? 'true' : 'false')
|
|
fd.append('sdStrength', settings.sdStrength.toString())
|
|
fd.append('sdSteps', settings.sdSteps.toString())
|
|
fd.append('sdGuidanceScale', settings.sdGuidanceScale.toString())
|
|
fd.append('sdSampler', settings.sdSampler.toString())
|
|
fd.append('sdSeed', settings.sdSeedFixed ? settings.sdSeed.toString() : '-1')
|
|
|
|
if (sizeLimit === undefined) {
|
|
fd.append('sizeLimit', '1080')
|
|
} else {
|
|
fd.append('sizeLimit', sizeLimit)
|
|
}
|
|
|
|
try {
|
|
const res = await fetch(`${API_ENDPOINT}/inpaint`, {
|
|
method: 'POST',
|
|
body: fd,
|
|
})
|
|
if (res.ok) {
|
|
const blob = await res.blob()
|
|
const seed = res.headers.get('x-seed')
|
|
return { blob: URL.createObjectURL(blob), seed }
|
|
}
|
|
} catch {
|
|
throw new Error('Something went wrong on server side.')
|
|
}
|
|
}
|
|
|
|
export function switchModel(name: string) {
|
|
const fd = new FormData()
|
|
fd.append('name', name)
|
|
return fetch(`${API_ENDPOINT}/model`, {
|
|
method: 'POST',
|
|
body: fd,
|
|
})
|
|
}
|
|
|
|
export function currentModel() {
|
|
return fetch(`${API_ENDPOINT}/model`, {
|
|
method: 'GET',
|
|
})
|
|
}
|
|
|
|
export function modelDownloaded(name: string) {
|
|
return fetch(`${API_ENDPOINT}/model_downloaded/${name}`, {
|
|
method: 'GET',
|
|
})
|
|
}
|