Spaces:
Running
Running
| 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) | |