Updated code to save the image, remove bug causing same image to pick up

app.py

@@ -38,7 +38,7 @@ be to the input. This pipeline requires a value of at least `1`. It's possible y
 """
 
 def chat(image_in, in_steps, in_guidance_scale, in_img_guidance_scale, image_hid, img_name, prompt, history, progress=gr.Progress(track_tqdm=True)):
-    #progress(0, desc="Starting...")
+    progress(0, desc="Starting...")
     global counter 
     counter += 1
     #if message == "revert": --to add revert functionality later
@@ -46,12 +46,12 @@ def chat(image_in, in_steps, in_guidance_scale, in_img_guidance_scale, image_hid
       # Open the image
       image_in = Image.open(img_name) #("edited_image.png") #(img_nm)
       edited_image = pipe(prompt, image=image_in, num_inference_steps=int(in_steps), guidance_scale=float(in_guidance_scale), image_guidance_scale=float(in_img_guidance_scale)).images[0]
-      edited_image.save(img_name)
+      edited_image.save(img_name, overwrite=True)
     else:
       seed = random.randint(0, 1000000)
       img_name = f"edited_image_{seed}.png"
       edited_image = pipe(prompt, image=image_in, num_inference_steps=int(in_steps), guidance_scale=float(in_guidance_scale), image_guidance_scale=float(in_img_guidance_scale)).images[0]
-      edited_image.save(img_name) #("/tmp/edited_image.png") #(img_nm)
+      edited_image.save(img_name, overwrite=True) #("/tmp/edited_image.png") #(img_nm)
     history = history or []
     #Resizing (or not) the image for better display and adding supportive sample text
     add_text_list = ["There you go", "Enjoy your image!", "Nice work! Wonder what you gonna do next!", "Way to go!", "Does this work for you?", "Something like this?"]