import { useState } from "react"

export const useInputGeneration = () => {
  const [prompt, setPrompt] = useState<string>('')
  const [loading, setLoading] = useState<boolean>(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,
  }

}