import gradio as gr
import requests
from PIL import Image
import io
import os

API_URL = "https://api-inference.huggingface.co/models/GenAIJake/d3xt3r-l1tt13-dachshund"
headers = {"Authorization": f"Bearer {os.getenv('HF_TOKEN')}"}

def generate(prompt):
    try:
        payload = {
            "inputs": prompt,
            "parameters": {
                "num_inference_steps": 5
            }
        }
        response = requests.post(API_URL, headers=headers, json=payload)
        image = Image.open(io.BytesIO(response.content))
        return image
    except Exception as e:
        return f"Error generating image: {str(e)}"

# Create the Gradio interface
demo = gr.Interface(
    fn=generate,
    inputs=gr.Textbox(label="Enter your prompt"),
    outputs=gr.Image(label="Generated Image"),
    title="D3xt3r L1tt13 Dachshund Generator",
    description="Generate images of a cute dachshund character."
)

demo.launch()