Spaces:
Runtime error
Runtime error
import streamlit as st | |
import os | |
uploaded_images = {'characters': {}, 'terrain': {}} | |
def get_image_path(img, name, image_type): | |
file_path = f"data/uploadedImages/{image_type}/{name}/{img.name}" | |
os.makedirs(os.path.dirname(file_path), exist_ok=True) | |
with open(file_path, "wb") as img_file: | |
img_file.write(img.getbuffer()) | |
return file_path | |
image_type = st.selectbox('Choose image type:', options=['characters', 'terrain']) | |
name = st.text_input('Enter a name for the image:') | |
uploaded_files = st.file_uploader('Upload image(s)', type=['png', 'jpg'], accept_multiple_files=True) | |
for uploaded_file in uploaded_files: | |
if uploaded_file is not None: | |
# Get actual image file | |
bytes_data = get_image_path(uploaded_file, name, image_type) | |
uploaded_images[image_type].setdefault(name, []) | |
uploaded_images[image_type][name].append(bytes_data) | |
st.image(bytes_data, use_column_width=True) | |
# Display character images on the left and terrain images on the right | |
if uploaded_images['characters']: | |
st.sidebar.write('**Characters**') | |
for name, files in uploaded_images['characters'].items(): | |
for file in files: | |
st.sidebar.image(file, width=100, caption=name) | |
if uploaded_images['terrain']: | |
st.write('**Terrain**') | |
row = [] | |
for name, files in uploaded_images['terrain'].items(): | |
for file in files: | |
row.append(file) | |
if len(row) == 3: | |
st.image(row, width=100 * 3) | |
row = [] | |
if row: | |
st.image(row, width=100 * len(row)) # Last row, if not complete |