Update app.py

app.py

@@ -1141,15 +1141,18 @@ def display_images_and_wikipedia_summaries(num_columns=4):
 
     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)
-            
-            # Add text input for image file
-            image_text_input = st.text_input(f"Image Prompt for {image_file}", key=f"image_prompt_{image_file}")
-            if image_text_input:
-                process_image(image_file, image_text_input)
+            try:
+                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)
+                
+                # Add text input for image file
+                image_text_input = st.text_input(f"Image Prompt for {image_file}", key=f"image_prompt_{image_file}")
+                if image_text_input:
+                    process_image(image_file, image_text_input)
+            except:
+                errored = True
                 
         col_index += 1  # Increment to move to the next column in the next iteration