Update app.py

app.py

@@ -22,21 +22,21 @@ for uploaded_file in uploaded_files:
         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
+if image_type == 'characters':
+    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)
+else:
+    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