import os
from dotenv import load_dotenv
import openai
import gradio as gr

# Load environment variables
load_dotenv()

# Get the API key from Hugging Face Secrets
api_key = os.getenv("OPENAI_API_KEY")

# Ensure API key is set
if not api_key:
    raise ValueError("OPENAI_API_KEY is not set. Please check your Hugging Face Secrets.")

openai.api_key = api_key

def generate_image(prompt):
    response = openai.Image.create(
        prompt=prompt,
        n=1,
        size="1024x1024"
    )
    return response["data"][0]["url"]

# Create Gradio Interface
gr.Interface(
    fn=generate_image,
    inputs=gr.Textbox(label="Describe your image"),
    outputs=gr.Image(label="Generated Image"),
    title="DALL-E Image Generator"
).launch()