update debugging for image save step

app.py

@@ -62,18 +62,19 @@ def chat(image_in, in_steps, in_guidance_scale, in_img_guidance_scale, image_hid
         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)
-      counter_out += 1
+      #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 = random.choice(add_text_list) + '<img src="/file=' + img_name + '">'
+    if counter_out > 0:
+        response = random.choice(add_text_list) + '<img src="/file=' + temp_img_name + '">'
         history.append((prompt, response))
-        return history, history, edited_image, img_name, counter_out
+        return history, history, edited_image, temp_img_name, counter_out
     else:
-        response = random.choice(add_text_list) + '<img src="/file=' + temp_img_name + '">' 
+        response = random.choice(add_text_list) + '<img src="/file=' + img_name + '">' 
         history.append((prompt, response))
-        return history, history, edited_image, temp_img_name, counter_out
+        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>