update revert option

app.py

@@ -39,9 +39,23 @@ 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, counter_out, prompt, history, progress=gr.Progress(track_tqdm=True)):
+def previous(image):
+    return image 
+
+def chat(image_in, in_steps, in_guidance_scale, in_img_guidance_scale, image_hid, img_name, counter_out, image_oneup, prompt, history, progress=gr.Progress(track_tqdm=True)):
     progress(0, desc="Starting...")
-    #if message == "revert": --to add revert functionality later
+    if prompt == 'reverse' : #--to add revert functionality later
+        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:
+        temp_img_name = img_name[:-4]+str(int(time.time()))+'.png' 
+        image_oneup.save(temp_img_name)
+        response = 'Reverted to the last image ' + '<img src="/file=' + temp_img_name + '">'  
+        history.append((prompt, response))
+        return history, history, image_oneup, temp_img_name, counter_out
+    # Save the resized image
+    #img.save("resized_example.jpg", optimize=True, quality=95
     if counter_out > 0:
       edited_image = pipe(prompt, image=image_hid, num_inference_steps=int(in_steps), guidance_scale=float(in_guidance_scale), image_guidance_scale=float(in_img_guidance_scale)).images[0]
       if os.path.exists(img_name):
@@ -58,11 +72,17 @@ def chat(image_in, in_steps, in_guidance_scale, in_img_guidance_scale, image_hid
     else:
       seed = random.randint(0, 1000000)
       img_name = f"./edited_image_{seed}.png"
-      #Resizing to handle memory errors
       basewidth = 512
       wpercent = (basewidth/float(image_in.size[0]))
       hsize = int((float(image_in.size[1])*float(wpercent)))
       image_in = image_in.resize((basewidth,hsize), Image.Resampling.LANCZOS)
+      # Get the original size of the image
+      #original_size = image_in.size
+      # Calculate the new size of the image
+      #new_size = (int(original_size[0]/2), int(original_size[1]/2))
+      # Resize the image
+      #img1 = img.resize(new_size)
+      #image_in = image_in.resize(new_size,Image.ANTIALIAS) # resample=Image.LANCZOS) 
       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]
       if os.path.exists(img_name):
         os.remove(img_name)
@@ -84,6 +104,7 @@ def chat(image_in, in_steps, in_guidance_scale, in_img_guidance_scale, image_hid
         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</h1></center>
@@ -100,17 +121,23 @@ with gr.Blocks() as demo:
         image_in = gr.Image(type='pil', label="Original Image")
         text_in = gr.Textbox()
         state_in = gr.State()
+        #with gr.Row():
         b1 = gr.Button('Edit the image!')
+        #b2 = gr.Button('Revert!')
         with gr.Accordion("Advance settings for Training and Inference", open=False):
           gr.Markdown("Advance settings for - Number of Inference steps, Guidanace scale, and Image guidance scale.")
           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(type='pil', visible=False)
+          image_hid = gr.Image(type='pil', visible=True)
+          image_oneup = gr.Image(type='pil', visible=True)
           img_name_temp_out = gr.Textbox(visible=False)
+          #img_revert = gr.Checkbox(visible=True, value=False,label=to track a revert message)
           counter_out = gr.Number(visible=False, value=0, precision=0)
       chatbot = gr.Chatbot() 
-    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)
+    b1.click(chat,[image_in, in_steps, in_guidance_scale, in_img_guidance_scale, image_hid, img_name_temp_out,counter_out, image_oneup,  text_in, state_in], [chatbot, state_in, image_hid, img_name_temp_out, counter_out]) #, queue=True)
+    b1.click(previous, [image_hid], [image_oneup])
+    #b2.click(previous, image_oneup, image_hid)
     gr.Markdown(help_text)
     
 demo.queue(concurrency_count=10)