Spaces:
Sleeping
Sleeping
| from PIL import Image | |
| import streamlit as st | |
| import os | |
| # Main function | |
| def CREATEGIF(): | |
| st.title("Create photoshoot visual") | |
| # Set the directory where the uploaded images will be saved | |
| UPLOAD_DIR = 'uploaded_images' | |
| # Create the directory if it doesn't exist | |
| if not os.path.exists(UPLOAD_DIR): | |
| os.makedirs(UPLOAD_DIR) | |
| # Streamlit app title | |
| st.title("Image Upload and Save App") | |
| # File uploader allows user to upload an image | |
| uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"]) | |
| if uploaded_file is not None: | |
| # Open the uploaded image | |
| image = Image.open(uploaded_file) | |
| # Display the uploaded image | |
| st.image(image, caption='Uploaded Image.', use_column_width=True) | |
| # Save the uploaded image to the specified directory | |
| image_path = os.path.join(UPLOAD_DIR, uploaded_file.name) | |
| image.save(image_path) | |
| st.write(f"Image saved at: {image_path}") | |
| else: | |
| st.write("No image uploaded yet.") | |
| image_files = [f for f in os.listdir("bgImages") if os.path.isfile(os.path.join("bgImages", f))] | |
| # Create a dropdown with the list of image files | |
| selected_image = st.selectbox("Select an image file", image_files) | |
| if selected_image: | |
| # Display the selected image | |
| image_path = os.path.join("bgImages", selected_image) | |
| image = Image.open(image_path) | |
| st.image(image, caption=f"Selected image: {selected_image}") | |
| if __name__ == "__main__": | |
| CREATEGIF() | |