Update app.py

app.py

@@ -1644,16 +1644,16 @@ def process_image(image_input):
         )
         image_response = response.choices[0].message.content
         st.markdown(image_response)
-
-
-        # Save the image with a new filename based on the response
-        #save_image(image_input, image_response)   
         
         filename_txt = generate_filename(image_response, "md")  # Save markdown on image AI output from gpt4o
         create_file(filename_txt, image_response, '', True)          #create_file() # create_file()  3 required positional arguments: 'filename', 'prompt', and 'response'
 
         filename_txt = generate_filename(image_response, "png")
         save_image(image_input, filename_txt)     # Save copy of image with new filename   
+        st.rerun() # rerun to show new image and new markdown files
+
+        reurn image_response
+
 
 def process_audio(audio_input):
     if audio_input:
@@ -1767,7 +1767,8 @@ def main():
         process_text()
     elif option == "Image":
         image_input = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
-        process_image(image_input)
+        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)