Update app.py

app.py

@@ -236,7 +236,9 @@ def display_images_and_wikipedia_summaries():
     if not image_files:
         st.write("No PNG images found in the current directory.")
         return
-    
+        
+    keyword = image_file.split('.')[0]  # Assumes keyword is the file name without extension
+
     for image_file in image_files:
         image = Image.open(image_file)
         st.image(image, caption=image_file, use_column_width=True)
@@ -255,7 +257,6 @@ def display_images_and_wikipedia_summaries():
         """
         st.markdown("# " + links_md)
         
-        keyword = image_file.split('.')[0]  # Assumes keyword is the file name without extension