Update app.py

app.py

@@ -38,9 +38,13 @@ def display_images_from_csv():
             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)
+
+# Rewrite for gradio
+image_type = gr.selectbox('Choose image type:', options=['characters', 'terrain'])
+name = gr.text_input('Enter a name for the image:')
+uploaded_files = gr.file_uploader('Upload image(s)', type=['png', 'jpg'], accept_multiple_files=True)
+
+
 
 for uploaded_file in uploaded_files:
     if uploaded_file is not None:
@@ -48,15 +52,22 @@ for uploaded_file in uploaded_files:
         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)
+
+        # Rewrite for gradio
+        gr.image(bytes_data, use_column_width=True)
+        
         update_csv_file(uploaded_file, name, image_type)
 
 if image_type == 'characters':
     if uploaded_images['characters']:
-        st.sidebar.write('**Characters**')
+
+        # Rewrite for gradio
+        gr.sidebar.write('**Characters**')
         for name, files in uploaded_images['characters'].items():
             for file in files:
-                st.sidebar.image(file, width=100, caption=name)
+
+                # Rewrite for gradio
+                gr.sidebar.image(file, width=100, caption=name)
 else:
     if uploaded_images['terrain']:
         st.write('**Terrain**')
@@ -65,12 +76,19 @@ else:
             for file in files:
                 row.append(file)
                 if len(row) == 3:
-                    st.image(row, width=100 * 3)
+
+                    # Rewrite for gradio
+                    gr.image(row, width=100 * 3)
                     row = []
         if row:
-            st.image(row, width=100 * len(row)) # Last row, if not complete
+
+            # Rewrite for gradio
+            gr.image(row, width=100 * len(row)) # Last row, if not complete
 
 while True:
     time.sleep(20)
-    st.empty()
+
+    # Rewrite for gradio
+    gr.empty()
+    
     display_images_from_csv()