Update app.py

app.py

@@ -1,5 +1,7 @@
 import streamlit as st
 import os
+import csv
+import time
 
 uploaded_images = {'characters': {}, 'terrain': {}}
 
@@ -10,6 +12,31 @@ def get_image_path(img, name, image_type):
         img_file.write(img.getbuffer())
     return file_path
 
+def update_csv_file(uploaded_file, name, image_type):
+    csv_file_path = "Resources.csv"
+    with open(csv_file_path, mode='a', newline='') as csv_file:
+        csv_writer = csv.writer(csv_file)
+        csv_writer.writerow([name, uploaded_file.name, image_type])
+
+def get_uploaded_files_info():
+    csv_file_path = "Resources.csv"
+    with open(csv_file_path, mode='r') as csv_file:
+        csv_reader = csv.reader(csv_file)
+        files_info = []
+        for row in csv_reader:
+            files_info.append(row)
+        return files_info
+
+def display_images_from_csv():
+    files_info = get_uploaded_files_info()
+    for row in files_info:
+        if row[2] == 'characters':
+            img_path = f"data/uploadedImages/{row[2]}/{row[0]}/{row[1]}"
+            st.sidebar.image(img_path, width=100, caption=row[0])
+        else:
+            img_path = f"data/uploadedImages/{row[2]}/{row[0]}/{row[1]}"
+            st.image(img_path, width=100, caption=row[0])
+
 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)
@@ -21,6 +48,7 @@ for uploaded_file in uploaded_files:
         uploaded_images[image_type].setdefault(name, [])
         uploaded_images[image_type][name].append(bytes_data)
         st.image(bytes_data, use_column_width=True)
+        update_csv_file(uploaded_file, name, image_type)
 
 if image_type == 'characters':
     if uploaded_images['characters']:
@@ -40,3 +68,8 @@ else:
                     row = []
         if row:
             st.image(row, width=100 * len(row)) # Last row, if not complete
+
+while True:
+    time.sleep(20)
+    st.empty()
+    display_images_from_csv()