import streamlit as st
import openai
import os


openai.api_key = os.getenv("openapikey")

def generate_image(prompt):
    try:
        response = openai.images.generate(
            model="dall-e-3",
            prompt=prompt,
            n=1,
            size="1024x1024"
        )
        return response.data[0].url

    except openai.OpenAIError as e:
        st.error(f"Error generating image: {e}")
        return None
    except Exception as e:
        st.error(f"An unexpected error occurred: {e}")
        return None

st.title(" Image Generator")
prompt = st.text_input("Enter a prompt:")

if st.button("Generate"):
    if prompt:
        with st.spinner("Generating image..."):
            image_url = generate_image(prompt)
            if image_url:
                st.image(image_url)