Update app.py

app.py

@@ -1499,33 +1499,40 @@ def save_image(image_input, filename):
 
 def process_image(image_input):
     if image_input:
-        base64_image = base64.b64encode(image_input.read()).decode("utf-8")
-        response = client.chat.completions.create(
-            model=MODEL,
-            messages=[
-                {"role": "system", "content": "You are a helpful assistant that responds in Markdown."},
-                {"role": "user", "content": [
-                    {"type": "text", "text": "Help me understand what is in this picture and list ten facts as markdown outline with appropriate emojis that describes what you see."},
-                    {"type": "image_url", "image_url": {
-                        "url": f"data:image/png;base64,{base64_image}"}
-                    }
-                ]}
-            ],
-            temperature=0.0,
-        )
-        image_response = response.choices[0].message.content
-        st.markdown(image_response)
-        
-        # Save markdown on image AI output from gpt4o
-        filename_md = f"{image_input.name}.md"
-        with open(filename_md, "w", encoding="utf-8") as f:
-            f.write(image_response)
-
-        # Save copy of image with original filename
-        filename_img = image_input.name
-        save_image(image_input, filename_img)
-        
-        return image_response
+        st.markdown('Processing image:  ' + image_input.name )
+        if image_input:
+            base64_image = base64.b64encode(image_input.read()).decode("utf-8")
+            response = client.chat.completions.create(
+                model=MODEL,
+                messages=[
+                    {"role": "system", "content": "You are a helpful assistant that responds in Markdown."},
+                    {"role": "user", "content": [
+                        {"type": "text", "text": "Help me understand what is in this picture and list ten facts as markdown outline with appropriate emojis that describes what you see."},
+                        {"type": "image_url", "image_url": {
+                            "url": f"data:image/png;base64,{base64_image}"}
+                        }
+                    ]}
+                ],
+                temperature=0.0,
+            )
+            image_response = response.choices[0].message.content
+            st.markdown(image_response)
+            
+            # Save markdown on image AI output from gpt4o
+            filename_md = generate_filename(image_input.name + '- ' + image_response, "md")
+            # Save markdown on image AI output from gpt4o 
+            filename_png = filename_md.replace('.md', '.' + image_input.name.split('.')[-1])
+                    
+            create_file(filename_md, image_response, '', True)          #create_file() # create_file()  3 required positional arguments: 'filename', 'prompt', and 'response'
+
+            with open(filename_md, "w", encoding="utf-8") as f:
+                f.write(image_response)
+
+            # Save copy of image with original filename
+            filename_img = image_input.name
+            save_image(image_input, filename_img)
+            
+            return image_response
 
 def save_imageold(image_input, filename_txt):
     # Save the uploaded video file
@@ -1675,11 +1682,12 @@ def main():
         text_input = st.text_input("Enter your text:")
         if (text_input > ''):
             textResponse = process_text(text_input)
-
     elif option == "Image":
         image_input = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
         image_response = process_image(image_input)
-        #st.markdown(image_response)
+
+
+
     elif option == "Audio":
         audio_input = st.file_uploader("Upload an audio file", type=["mp3", "wav"])
         process_audio(audio_input)
@@ -1688,6 +1696,9 @@ def main():
         process_audio_and_video(video_input)
 
     # Image and Video Galleries
+    num_columns_images=st.slider(key="num_columns_images", label="Choose Number of Image Columns", min_value=1, max_value=15, value=5)
+    display_images_and_wikipedia_summaries(num_columns_images)   # Image Jump Grid
+
     num_columns_video=st.slider(key="num_columns_video", label="Choose Number of Video Columns", min_value=1, max_value=15, value=5)
     display_videos_and_links(num_columns_video)   # Video Jump Grid
 
@@ -1699,8 +1710,6 @@ if showExtendedTextInterface:
     num_columns_text=st.slider(key="num_columns_text", label="Choose Number of Text Columns", min_value=1, max_value=15, value=4)
     display_buttons_with_scores(num_columns_text)  # Feedback Jump Grid
     st.markdown(personality_factors)
-    num_columns_images=st.slider(key="num_columns_images", label="Choose Number of Image Columns", min_value=1, max_value=15, value=5)
-    display_images_and_wikipedia_summaries(num_columns_images)   # Image Jump Grid