Update app.py

app.py

@@ -198,6 +198,7 @@ def extract_title(text):
         title = re.sub(r'[^a-zA-Z0-9_\-]', ' ', text[-200:])
     return title[-200:]
 
+#@st.cache_resource
 def process_image(image_input, user_prompt):
     SaveNewFile=True
     image_file_name=''
@@ -210,7 +211,8 @@ def process_image(image_input, user_prompt):
         if image_input is None:
             data=False
         else:
-            image_file_name = image_input.name
+            #image_file_name = image_input.name
+            image_file_name = image_input.filename
             image_input = image_input.read()
             SaveNewFile=True
     
@@ -1130,7 +1132,7 @@ def display_glossary(glossary, area):
             for idx, term in enumerate(terms, start=1):
                 st.write(f"{idx}. {term}")
 
-
+# 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:
@@ -1142,6 +1144,7 @@ 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
             try:
@@ -1152,8 +1155,12 @@ def display_images_and_wikipedia_summaries(num_columns=4):
                 
                 # 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)
+                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
                 
@@ -1814,6 +1821,9 @@ def main():
         if (image_input is not None):
             image_response = process_image(image_input, text_input)
 
+            with st.chat_message(name="ai", avatar="🦖"):
+                st.markdown(image_response)
+
     elif option == "Audio":
         text = "You are generating a transcript summary. Create a summary of the provided transcription. Respond in Markdown."
         text_input = st.text_input(label="Enter text prompt to use with Audio context.", value=text)