Update app.py

app.py

@@ -616,41 +616,19 @@ def display_images_and_wikipedia_summaries():
 
     # Create a map for number of columns to use for each size
     num_columns_map = {"small": 4, "medium": 3, "large": 2}
-    current_grid_size = 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 Wikipedia and Google search links
-            k = image_file.split('.')[0]  # Assumes keyword is the file name without extension
-
-            display_glossary_entity(k)
-            
-            #wikipedia_url = create_search_url_wikipedia(keyword)
-            #google_url = create_search_url_google(keyword)
-            #youtube_url = create_search_url_youtube(keyword)
-            #bing_url = create_search_url_bing(keyword)
-            #ai_url = create_search_url_ai(keyword)
-            #ai_url2 = create_search_url_ai(keyword + ' ' + PromptPrefix)
-            #ai_url3 = create_search_url_ai(keyword + ' ' + PromptPrefix2)
-            #links_md = f"""
-            #[Wikipedia]({wikipedia_url}) | 
-            #[Google]({google_url}) | 
-            #[YouTube]({youtube_url}) | 
-            #[Bing]({bing_url}) | 
-            #[AI]({ai_url})|
-            #[AI Novel]({ai_url2} )|
-            #[AI Novel App]({ai_url3})
-            #"""
-            #st.markdown(links_md)
-            #col_index += 1
+            with cols[col_index % current_grid_size]:
+                image = Image.open(image_file)
+                st.image(image, caption=image_file, use_column_width=True)
+                # Display search links
+                k = image_file.split('.')[0]  # Assumes keyword is the file name without extension
+                display_glossary_entity(k)
+                #col_index += 1
 
 
 def get_all_query_params(key):