Update app.py

app.py

@@ -599,11 +599,43 @@ def create_search_url_ai(keyword):
     base_url = "https://huggingface.co/spaces/awacke1/GraphicAINovel?q="
     return base_url + keyword.replace(' ', '+')
 
+ 
+
 def display_images_and_wikipedia_summaries():
     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
+
+    # Sort image_files based on the length of the keyword to create a visually consistent grid
+    image_files_sorted = sorted(image_files, key=lambda x: len(x.split('.')[0]))
+    # Calculate the grid size based on the sorted keywords
+    grid_sizes = [len(f.split('.')[0]) for f in image_files_sorted]
+    # Dynamically adjust column size based on keyword length
+    col_sizes = ['small' if size <= 4 else 'medium' if size <= 8 else 'large' for size in grid_sizes]
+
+    # Create a map for number of columns to use for each size
+    num_columns_map = {"small": 4, "medium": 3, "large": 2}
+    current_grid_size = 0
+
+    for image_file, col_size in zip(image_files_sorted, col_sizes):
+        if current_grid_size != num_columns_map[col_size]:
+            cols = st.columns(num_columns_map[col_size])
+            current_grid_size = num_columns_map[col_size]
+            col_index = 0
+
+        with cols[col_index % current_grid_size]:
+            image = Image.open(image_file)
+            st.image(image, caption=image_file, use_column_width=True)
+            # Display of links or other information follows here
+
+        col_index += 1
+
+def display_images_and_wikipedia_summaries_deprecated():
+    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
     
     for image_file in image_files:
         image = Image.open(image_file)