Update app.py

app.py

@@ -916,7 +916,7 @@ def display_videos_and_links(num_columns):
             display_glossary_entity(k)  
         col_index += 1  # Increment column index to place the next video in the next column
 
-#@st.cache_resource
+# Image Prompt
 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:
@@ -928,12 +928,29 @@ def display_images_and_wikipedia_summaries(num_columns=4):
     cols = st.columns(num_columns)  # Use specified num_columns for layout
     col_index = 0  # Initialize column index for cycling through columns
 
+    errored = False
     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)
+            try:
+                image = Image.open(image_file)
+                #st.image(image, caption=image_file, use_column_width=True)
+                st.image(image, 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}")
+                #image_text_input = st.text_input(f"Image Prompt:", key=f"image_prompt_{image_file}")
+                image_text_input = st.text_input(key=f"image_prompt_{image_file}")
+                if (len(image_text_input) > 0):
+                    #image_response = process_image(image, image_text_input)
+                    image_response = process_image(image_file, image_text_input)
+
+                    with st.chat_message(name="ai", avatar="🦖"):
+                        st.markdown(image_response)
+            except:
+                errored = True
+                
         col_index += 1  # Increment to move to the next column in the next iteration