update layout, bug fix for parallel usage

app.py

@@ -37,25 +37,27 @@ 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, prompt, history, progress=gr.Progress(track_tqdm=True)):
-    progress(0, desc="Starting...")
+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...")
     global counter 
     counter += 1
     #if message == "revert": --to add revert functionality later
     if counter > 1:
       # Open the image
-      image_in = Image.open("edited_image.png") 
-    #edited_image = pipe(prompt, image=image_in, num_inference_steps=20, image_guidance_scale=1).images[0]
-    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("edited_image.png") #("/tmp/edited_image.png") #(img_nm)
+      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)
+    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)
     history = history or []
-    #Fixed supportive text
+    #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?"]
-    #Resize (or not) the image for better display
-    #response = random.choice(add_text_list) + '<img src="/file=edited_image.png" style="width: 350px; height: 350px;">'
-    response = random.choice(add_text_list) + '<img src="/file=edited_image.png">' # style="width: 350px; height: 350px;">'
+    response = random.choice(add_text_list) + '<img src="/file=' + img_name + '">' # style="width: 200px; height: 200px;">'
     history.append((prompt, response))
-    return history, history
+    return history, history, edited_image, img_name
 
 with gr.Blocks() as demo:
     gr.Markdown("""<h1><center> Chat Interface with InstructPix2Pix: Give Image Editing Instructions </h1></center>
@@ -76,8 +78,10 @@ with gr.Blocks() as demo:
           in_steps = gr.Number(label="Enter the number of Inference steps", value = 20)
           in_guidance_scale = gr.Slider(1,10, step=0.5, label="Set Guidance scale", value=7.5)
           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)
       chatbot = gr.Chatbot() 
-    b1.click(chat,[image_in, in_steps, in_guidance_scale, in_img_guidance_scale,  text_in, state_in], [chatbot, state_in]) #, queue=True)
+    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)
     gr.Markdown(help_text)
     
 demo.queue(concurrency_count=10)