update with synth image translte

app.py

@@ -27,7 +27,17 @@ def image_edit(task_name, progress=gr.Progress(track_tqdm=True)):
                   "--prompt", "output/test_cat/prompt/input_image.txt", "--task_name", task_name,
                   "--results_folder", "output/test_cat/"])
   return "output/test_cat/edit/input_image.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...")
+  # 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"
+
+    
 def set_visible_true():
     return gr.update(visible=True)
 
@@ -54,29 +64,41 @@ with gr.Blocks() as demo:
                 Please visit their website and <a href="https://github.com/pix2pixzero/pix2pix-zero" target="_blank">github repo</a> for more details.
                 </p></div>""")
   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")
+    image_in = gr.Image(type="pil", label="Upload the image of a cat or a dog you want to translate")
     with gr.Column():
-      btn_inversion = gr.Button("Get input noise and caption",visible=False )
+      btn_inversion = gr.Button("Get input noise",visible=False )
       with gr.Row():
         blip_prompt = gr.Textbox(visible=False)
         inversion_file = gr.File(visible=False)
         #task_name = gr.Textbox()
   with gr.Row():
-    image_out = gr.Image(visible=False)
+    image_out = gr.Image(visible=False, label="Trnaslated Image output")
     with gr.Column():
       task_name_radio  = gr.Radio(choices = ["cat2dog", "dog2cat",], type="value", visible=False, label="Select a task that you want to accomplish") #, value="cat2dog"),
-      btn_imageedit = gr.Button(value="Translate the image!",visible=False)
-    
-    
+      btn_imageedit = gr.Button(value="Edit the image!",visible=False)
+      prompt_synth = gr.Textbox(visible=False)
+      btn_synth_image = gr.Button(value="Generate & Translate SD image",visible=False)
+  with gr.Row():
+    image_synth = gr.Image(visible=False, label="Synthetic image generated by Stable Diffusion on the fly")
+    image_synth_translated = gr.Image(visible=False, label="Translated synthetic image")
+
     btn_inversion.click(inversion,[image_in],[inversion_file, blip_prompt])
     btn_inversion.click(set_visible_true, [], task_name_radio)  #inversion_file, blip_prompt,
     btn_inversion.click(set_visible_False, [], btn_inversion)
 
+    #task_initial_radio.change(set_visible_true, [], btn_imageedit)
+    
     task_name_radio.change(set_visible_true, [], btn_imageedit)
     task_name_radio.change(set_visible_true, [], image_out)
 
     btn_imageedit.click(image_edit,[task_name_radio],[image_out])
-    
+    btn_imageedit.click(set_visible_False, [], btn_imageedit)
+    btn_imageedit.click(set_visible_true, [], prompt_synth)
+    btn_imageedit.click(set_visible_true, [], btn_synth_image)
+    btn_imageedit.click(set_visible_true, [], image_synth)
+    btn_imageedit.click(set_visible_true, [], image_synth_translated)
+    btn_synth_image.click(synthetic_image_edit,[prompt_synth, task_name_radio],[image_synth, image_synth_translated])
+
     image_in.clear(set_visible_true, [], btn_inversion)
     image_in.change(set_visible_true, [], btn_inversion)
     image_in.change(set_visible_true, [], blip_prompt)