Spaces:
Running
Running
File size: 2,030 Bytes
fb96795 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
import streamlit as st
# import openai
from PIL import Image
# from src.utils import get_width_height, resize_image
# from rembg import remove
def page4():
st.title("OpenAI DALL·E Image Editing")
st.info("""#### NOTE: you can download image by \
right clicking on the image and select save image as option""")
with st.form(key='form'):
uploaded_file = st.file_uploader("Choose an image file", type=['png', 'jpg'])
mask_file = st.file_uploader("Choose an mask file", type=['png', 'jpg'])
prompt = st.text_input("Enter a text prompt")
size = st.selectbox('Select size of the images', ('256x256', '512x512', '1024x1024'))
num_images = st.selectbox('Enter number of images to be generated', (1,2,3,4))
submit_button = st.form_submit_button(label='Submit')
if submit_button:
if (uploaded_file is not None) and (mask_file is not None) and prompt:
our_image = Image.open(uploaded_file)
mask_image = Image.open(mask_file)
width, height = get_width_height(size)
our_image = resize_image(our_image, width, height)
mask_image = resize_image(mask_image, width, height)
st.image(our_image, caption="Uploaded image", use_column_width=True)
st.image(mask_image, caption="Uploaded mask", use_column_width=True)
backround_removed_mask = remove(mask_image)
st.image(backround_removed_mask, caption="backround_removed_mask",
use_column_width=True)
# response = openai.Image.create_edit(
# image=our_image,
# mask=backround_removed_mask,
# prompt=prompt,
# n=num_images,
# size=size
# )
# for idx in range(num_images):
# image_url = response["data"][idx]["url"]
# st.image(image_url, caption=f"Generated image: {idx+1}",
# use_column_width=True)
|