import os
import requests
import io
from PIL import Image

API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2-1"


def generate(text_prompt):
    api_key = os.environ.get("HF_API_KEY", "")
    if not api_key:
        raise ValueError("HF_API_KEY environment variable is not set.")
    response = requests.post(
        API_URL,
        headers={"Authorization": f"Bearer {api_key}"},
        json={"inputs": text_prompt},
    )
    response.raise_for_status()
    return Image.open(io.BytesIO(response.content))


"""
text = input("Your sentence here:")
image = generate(text)
image.show()
"""