Update app.py

app.py

@@ -1552,6 +1552,7 @@ def save_image(image, filename):
     filename = f"{truncated_stem}{extension}"
     with open(filename, "wb") as f:
         f.write(image.getbuffer())
+    return filename
         
 def extract_boldface_terms(text):
     return re.findall(r'\*\*(.*?)\*\*', text)
@@ -1587,16 +1588,12 @@ def process_image(image_input):
             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)
-
             # Extract boldface terms from image_response then autoname save file
             boldface_terms = extract_boldface_terms(image_response)
             filename_stem, extension = os.path.splitext(image_input.name)
             filename_img = f"{filename_stem}_{' '.join(boldface_terms)}{extension}"
-            save_image(image_input, filename_img)
-            filename_md = f"{filename_stem}_{'_'.join(boldface_terms)}.md"
+            newfilename = save_image(image_input, filename_img)
+            filename_md = newfilename.replace('.png', '.md')
             create_file(filename_md, '', image_response, True)
             
             return image_response