|
import os |
|
|
|
import streamlit as st |
|
from openai import OpenAI |
|
from dotenv import load_dotenv |
|
|
|
load_dotenv() |
|
api_key = os.getenv("api_key") |
|
|
|
client = OpenAI(api_key=api_key) |
|
|
|
|
|
st.title("Text To Image Generator") |
|
st.write("Enter a prompt to generate an image!") |
|
|
|
|
|
user_input = st.text_input("Enter a prompt:") |
|
|
|
|
|
def generate_and_display_image(text, model="dall-e-3", size="1024x1024", quality="standard"): |
|
try: |
|
response = client.images.generate( |
|
model=model, |
|
prompt=text, |
|
size=size, |
|
quality=quality, |
|
n=1 |
|
) |
|
image_url = response.data[0].url |
|
st.image(image_url,width=800) |
|
except Exception as e: |
|
st.error(f"An error occurred: {e}") |
|
|
|
|
|
if st.button("Generate Image"): |
|
generate_and_display_image(user_input) |
|
|
|
|
|
|