update restart feature

app.py

@@ -44,18 +44,20 @@ def previous(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 prompt == 'reverse' : #--to add revert functionality later
+    if prompt.lower() == '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 prompt.lower() == 'restart' : #--to add revert functionality later
+        history = history or []
+        temp_img_name = img_name[:-4]+str(int(time.time()))+'.png' 
+        image_in.save(temp_img_name)
+        response = 'Reverted to the last image ' + '<img src="/file=' + temp_img_name + '">'  
+        history.append((prompt, response))
+        return history, history, image_in, temp_img_name, counter_out
     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):
@@ -72,17 +74,11 @@ 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 the image
       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)
@@ -111,10 +107,18 @@ with gr.Blocks() as demo:
     <p>For faster inference without waiting in the queue, you may duplicate the space and upgrade to GPU in settings.<br/>
     <a href="https://huggingface.co/spaces/ysharma/InstructPix2Pix_Chatbot?duplicate=true">
     <img style="margin-top: 0em; margin-bottom: 0em" src="https://bit.ly/3gLdBN6" alt="Duplicate Space"></a>
+    
     **Note: Please be advised that a safety checker has been implemented in this public space. 
     Any attempts to generate inappropriate or NSFW images will result in the display of a black screen 
     as a precautionary measure for the protection of all users. We appreciate your cooperation in 
     maintaining a safe and appropriate environment for all members of our community.**
+
+    New features and bug-fixes: 
+    
+    1. Now use 'reverse' as prompt to get back the previous image after an unwanted edit
+    2. Use 'restart' as prompt to get back to original image and start over!
+    3. Now you can load larger images (~5 mb) as well
+    
     <p/>""")
     with gr.Row():
       with gr.Column():