updated code for bug while second pass

app.py

@@ -38,12 +38,12 @@ be to the input. This pipeline requires a value of at least `1`. It's possible y
     * Cropping the image so the face takes up a larger portion of the frame.
 """
 
-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)):
+def chat(image_in, in_steps, in_guidance_scale, in_img_guidance_scale, image_hid, img_name, counter_out, prompt, history, progress=gr.Progress(track_tqdm=True)):
     progress(0, desc="Starting...")
-    global counter 
-    counter += 1
+    #global counter 
+    #counter += 1
     #if message == "revert": --to add revert functionality later
-    if counter > 1:
+    if counter_out > 0:
       # 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]
@@ -54,13 +54,14 @@ def chat(image_in, in_steps, in_guidance_scale, in_img_guidance_scale, image_hid
       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, overwrite=True) #("/tmp/edited_image.png") #(img_nm)
+      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?"]
     response = random.choice(add_text_list) + '<img src="/file=' + img_name + '">' # style="width: 200px; height: 200px;">'
     history.append((prompt, response))
-    return history, history, edited_image, img_name
+    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 </h1></center>
@@ -83,8 +84,9 @@ with gr.Blocks() as demo:
           in_img_guidance_scale = gr.Slider(1,10, step=0.5, label="Set Image Guidance scale", value=1.5)
           image_hid = gr.Image(visible=False,)
           img_name_temp_out = gr.Textbox(visible=False)
+          counter_out = gr.Number(visible=False, value=0)
       chatbot = gr.Chatbot() 
-    b1.click(chat,[image_in, in_steps, in_guidance_scale, in_img_guidance_scale, image_hid, img_name_temp_out,  text_in, state_in], [chatbot, state_in, image_hid, img_name_temp_out]) #, queue=True)
+    b1.click(chat,[image_in, in_steps, in_guidance_scale, in_img_guidance_scale, image_hid, img_name_temp_out,counter_out,  text_in, state_in], [chatbot, state_in, image_hid, img_name_temp_out, counter_out]) #, queue=True)
     gr.Markdown(help_text)
     
 demo.queue(concurrency_count=10)