upate random seed

app.py

@@ -11,36 +11,41 @@ def inversion(image_in):  #, progress=gr.Progress(track_tqdm=True)):
   #progress(0, desc="Starting...")
   seed = random.randint(0, 1000000)
   img_name = f"./input_image_{seed}.png"
+  img_label = f"./input_image_{seed}"
   # saving the input image
   image_in.save(img_name)
   #image_in.save("input_image.png") #("assets/test_images/cats/input_image.png")
   # Run the script file
   subprocess.run(["python", "src/inversion.py", "--input_image", "input_image.png", "--results_folder", "output/test_cat"])
   # Open the text file with blip caption
-  with open("output/test_cat/prompt/input_image.txt", "r") as file:
+  with open(f"output/test_cat/prompt/{img_label}.txt", "r") as file:
       # Read the file 
       prompt = file.read()
-  return "output/test_cat/inversion/input_image.pt", prompt
+  return f"output/test_cat/inversion/{img_label}.pt", prompt, seed
 
 
 # Performing image editing with the editing directions 
 # This will save the edited image as output/test_cat/edit/image-name.png
-def image_edit(task_name): #, progress=gr.Progress(track_tqdm=True)):
+def image_edit(task_name, seed): #, progress=gr.Progress(track_tqdm=True)):
   #progress(0, desc="Starting...")
+  img_label = f"./input_image_{seed}"
   # Run the script file
-  subprocess.run(["python", "src/edit_real.py", "--inversion", "output/test_cat/inversion/input_image.pt", 
-                  "--prompt", "output/test_cat/prompt/input_image.txt", "--task_name", task_name,
+  subprocess.run(["python", "src/edit_real.py", "--inversion", f"output/test_cat/inversion/{img_label}.pt", 
+                  "--prompt", f"output/test_cat/prompt/{img_label}.txt", "--task_name", task_name,
                   "--results_folder", "output/test_cat/"])
-  return "output/test_cat/edit/input_image.png"
+  return f"output/test_cat/edit/{img_label}.png"
 
 
 #Similarly, we can edit the synthetic images generated by Stable Diffusion with the following command.
 def synthetic_image_edit(prompt, task_name): #, progress=gr.Progress(track_tqdm=True)):
   #progress(0, desc="Starting...")
+  seed = random.randint(0, 1000000)
+  #img_name = f"./input_image_{seed}.png"
+  #img_label = f"./input_image_{seed}"
   # Run the script file
   subprocess.run(["python", "src/edit_synthetic.py", "--prompt_str", prompt, 
                   "--task", task_name, "--results_folder", "output/synth_editing"])
-  return "output/synth_editing/reconstruction.png", "output/synth_editing/edit.png"
+  return f"output/synth_editing/reconstruction{seed}.png", f"output/synth_editing/edit{seed}.png"
 
     
 def set_visible_true():
@@ -74,6 +79,7 @@ with gr.Blocks() as demo:
   refresh_direction_html = gr.HTML(value="<h4>👆Click the 'App' button to Refresh the space and try translation again with another Image</h4>", visible=False)
   with gr.Row():
     image_in = gr.Image(type="pil", label="Start by uploading an image of a Cat or a Dog that you want to translate")
+    seed = gr.Number(visible=False, Precision=0)
     #with gr.Box():
     #    gr.Examples( examples=[os.path.join(os.path.dirname(__file__), "dog11.jpg"), os.path.join(os.path.dirname(__file__), "cat11.jpg")],
     #    inputs=image_in,)
@@ -100,7 +106,7 @@ with gr.Blocks() as demo:
 
     image_in.change(set_visible_False, [], direction_html)
       
-    btn_inversion.click(inversion,[image_in],[inversion_file, blip_prompt])
+    btn_inversion.click(inversion,[image_in],[inversion_file, blip_prompt, seed])
     #btn_inversion.click(set_visible_true, [], task_name_radio)  #inversion_file, blip_prompt,
     btn_inversion.click(set_visible_False, [], btn_inversion)
 
@@ -111,7 +117,7 @@ with gr.Blocks() as demo:
     task_name_radio.change(set_visible_true, [], image_out)
     #task_name_radio.change(set_visible_true, [], html_tag)
 
-    btn_imageedit.click(image_edit,[task_name_radio],[image_out])
+    btn_imageedit.click(image_edit,[task_name_radio, seed],[image_out])
     btn_imageedit.click(set_visible_False, [], btn_imageedit)
 
     btn_imageedit.click(set_visible_true, [], html_tag)