shukdevdatta123 commited on
Commit
a72052a
·
verified ·
1 Parent(s): 94f39df

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +45 -0
app.py ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ from openai import OpenAI
3
+ from dotenv import load_dotenv
4
+ import os
5
+
6
+ # Streamlit UI for input
7
+ st.title("Generate an Image with DALL-E 3")
8
+
9
+ # Input for OpenAI API key
10
+ api_key = st.text_input("Enter your OpenAI API Key:")
11
+
12
+ # Check if API key is provided
13
+ if api_key:
14
+ # Set the API key for OpenAI
15
+ os.environ["OPENAI_API_KEY"] = api_key
16
+ load_dotenv() # Load environment variables if needed
17
+ client = OpenAI(api_key=api_key)
18
+
19
+ # Prompt input field
20
+ prompt = st.text_area("Enter your prompt:")
21
+
22
+ if prompt:
23
+ st.write("Generating image...")
24
+
25
+ # Call the OpenAI API to generate the image
26
+ try:
27
+ response = client.images.generate(
28
+ model="dall-e-3",
29
+ prompt=prompt,
30
+ size="1024x1024", # Available size: 1024x1024, 1024x1792, or 1792x1024
31
+ quality="standard", # Available quality: "standard" or "hd"
32
+ n=1, # Number of images to generate (limit is 1)
33
+ )
34
+
35
+ image_url = response.data[0].url
36
+
37
+ # Display the image
38
+ st.image(image_url, caption="Generated Image", use_column_width=True)
39
+ st.write("Image URL: " + image_url)
40
+ except Exception as e:
41
+ st.error(f"An error occurred: {e}")
42
+ else:
43
+ st.warning("Please enter a prompt.")
44
+ else:
45
+ st.warning("Please enter your OpenAI API Key.")