import { useState } from "react" export const useInputGeneration = () => { const [prompt, setPrompt] = useState('') const [loading, setLoading] = useState(false) const submit = () => { setLoading(true) return new Promise(async (resolve, reject) => { fetch( 'https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0', { headers: { Authorization: `Bearer {TOKEN}`, 'Content-Type': 'application/json', }, method: 'POST', body: JSON.stringify({ inputs: prompt, }) } ) .then((res) => console.log(res)) .finally(() => { setLoading(false) resolve(true) }) }) } return { prompt, setPrompt, loading, submit, } }