import requests
import gradio as gr
import os

# Set API endpoint
api_endpoint = "https://api.huggingface.co/v1/models/llama/versions/1/generate"

# Set API key from environment variable
api_key = os.environ["HF_API_KEY"]

# Define the image generation function
def generate_image(prompt):
	# Set parameters
	parameters = {
		"prompt": prompt,
		"num_images": 1,
		"width": 512,
		"height": 512
	}

	# Set API key in headers
	headers = {
		"Authorization": f"Bearer {api_key}"
	}

	# Send request and get response
	response = requests.post(api_endpoint, headers=headers, json=parameters)

	# Get generated image from response
	image = response.json()["images"][0]

	# Return the generated image
	return image

# Create a Gradio interface for the image generation function
demo = gr.Interface(
	fn=generate_image,
	inputs="text",
	outputs="image",
	title="LLaMA Image Generator"
)

# Launch the Gradio interface
demo.launch()