Update app.py

app.py

@@ -752,12 +752,14 @@ def display_videos_and_links():
         col_index += 1  # Increment column index to place the next video in the next column
 
 @st.cache_resource
-def display_images_and_wikipedia_summaries():
+def display_images_and_wikipedia_summariesold():
     image_files = [f for f in os.listdir('.') if f.endswith('.png')]
     if not image_files:
         st.write("No PNG images found in the current directory.")
         return
     image_files_sorted = sorted(image_files, key=lambda x: len(x.split('.')[0]))
+    num_columns=4
+
     grid_sizes = [len(f.split('.')[0]) for f in image_files_sorted]
     col_sizes = ['small' if size <= 4 else 'medium' if size <= 8 else 'large' for size in grid_sizes]
     num_columns_map = {"small": 4, "medium": 3, "large": 2}
@@ -772,7 +774,30 @@ def display_images_and_wikipedia_summaries():
                 st.image(image, caption=image_file, use_column_width=True)
                 k = image_file.split('.')[0]  # Assumes keyword is the file name without extension
                 display_glossary_entity(k)
-                
+
+@st.cache_resource
+def display_images_and_wikipedia_summaries(num_columns=4):
+    image_files = [f for f in os.listdir('.') if f.endswith('.png')]
+    if not image_files:
+        st.write("No PNG images found in the current directory.")
+        return
+
+    image_files_sorted = sorted(image_files, key=lambda x: len(x.split('.')[0]))
+
+    cols = st.columns(num_columns)  # Use specified num_columns for layout
+    col_index = 0  # Initialize column index for cycling through columns
+
+    for image_file in image_files_sorted:
+        with cols[col_index % num_columns]:  # Cycle through columns based on num_columns
+            image = Image.open(image_file)
+            st.image(image, caption=image_file, use_column_width=True)
+            k = image_file.split('.')[0]  # Assumes keyword is the file name without extension
+            display_glossary_entity(k)
+        col_index += 1  # Increment to move to the next column in the next iteration
+
+
+
+
 def get_all_query_params(key):
     return st.query_params().get(key, [])