Update app.py

app.py

@@ -24,7 +24,9 @@ pipeline.enable_vae_slicing()
 
 # DDIM inversion
 
-def run(ref_path, ref_style, ref_prompt, prompt1, prompt2, prompt3):
+global zts, src_style, src_prompt
+
+def load_refs(ref_path, ref_style, ref_prompt, prompt1, prompt2, prompt3):
 
     src_style = f"{ref_style}"
     src_prompt = f"{ref_prompt}, {src_style}."
@@ -34,7 +36,9 @@ def run(ref_path, ref_style, ref_prompt, prompt1, prompt2, prompt3):
     x0 = np.array(load_image(image_path).resize((1024, 1024)))
     zts = inversion.ddim_inversion(pipeline, x0, src_prompt, num_inference_steps, 2)
     #mediapy.show_image(x0, title="innput reference image", height=256)
+    return "ready"
 
+def run(prompt1, prompt2, prompt3):
     # run StyleAligned
     prompts = [
         src_prompt,
@@ -88,16 +92,29 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             ref_path = gr.Image(type="filepath", value="./example_image/medieval-bed.jpeg")
+            ref_style = gr.Textbox(label="Reference style", value="medieval painting")
+            ref_prompt = gr.Textbox(label="Referéce prompt", value="Man laying on bed")
+            load_ref_btn = gr.Button("Load Reference")
         with gr.Column():
-            ref_style = gr.Textbox(value="medieval painting")
-            ref_prompt = gr.Textbox(value="Man laying on bed")
-            prompt1 = gr.Textbox(value="A man working on a laptop")
-            prompt2 = gr.Textbox(value="A man eating pizza")
-            prompt3 = gr.Textbox(value="A woman playing on saxophone")
+            is_ready = gr.Textbox(label = "Ref status")
+            prompt1 = gr.Textbox(label="Prompt1", value="A man working on a laptop")
+            prompt2 = gr.Textbox(label="Prompt2", value="A man eating pizza")
+            prompt3 = gr.Textbox(label="Prompt3", value="A woman playing on saxophone")
             run_button = gr.Button("Submit")
         with gr.Column():
             results = gr.Gallery()
-
+    load_ref_btn.click(
+        fn = load_refs,
+        inputs = [
+            ref_path,
+            ref_style,
+            ref_prompt
+        ],
+        outputs = [
+            is_ready
+        ],
+        queue = False
+    )
     run_button.click(
         fn = run,
         inputs = [