update image save step

app.py

@@ -51,7 +51,11 @@ def chat(image_in, in_steps, in_guidance_scale, in_img_guidance_scale, image_hid
         os.remove(img_name)
       temp_img_name = img_name
       print(f"temp_img_name is :{temp_img_name}")
-      edited_image.save(temp_img_name) #, overwrite=True)
+      # Create a file-like object
+      with open(temp_img_name, "wb") as fp:
+        # Save the image to the file-like object
+        edited_image.save(fp)
+      #edited_image.save(temp_img_name) #, overwrite=True)
       counter_out += 1
     else:
       print("FIRST PASS")
@@ -61,20 +65,26 @@ def chat(image_in, in_steps, in_guidance_scale, in_img_guidance_scale, image_hid
       if os.path.exists(img_name):
         print("***First PASS:Image exists and will be deleted***")
         os.remove(img_name)
-      edited_image.save(img_name) #, overwrite=True) #("/tmp/edited_image.png") #(img_nm)
+      with open(img_name, "wb") as fp:
+        # Save the image to the file-like object
+        edited_image.save(fp)
+      #edited_image.save(img_name) #, overwrite=True) #("/tmp/edited_image.png") #(img_nm)
       #counter_out += 1
     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?"]
     if counter_out > 0:
+        #response = edited_image
         response = random.choice(add_text_list) + '<img src="/file=' + temp_img_name + '">'
         history.append((prompt, response))
         return history, history, edited_image, temp_img_name, counter_out
     else:
+        #response = edited_image
         response = random.choice(add_text_list) + '<img src="/file=' + img_name + '">' 
         history.append((prompt, response))
         counter_out += 1
         return history, history, edited_image, img_name, counter_out
+        
 
 with gr.Blocks() as demo:
     gr.Markdown("""<h1><center> Chat Interface with InstructPix2Pix: Give Image Editing Instructions [Apologies for inconvenience, this Space is still very much a work in progress...] </h1></center>