Update app.py

app.py

@@ -76,7 +76,7 @@ def add_overlay(base_img, overlay_img, margin=20):
 
 
 @spaces.GPU
-def infer(input_image, input_image_upload, overlay_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5, steps=28, progress=gr.Progress(track_tqdm=True)):
+def infer(input_image, input_image_upload, overlay_image, prompt="make it real", seed=42, randomize_seed=False, guidance_scale=2.5, steps=28, progress=gr.Progress(track_tqdm=True)):
     """
     Perform image editing using the FLUX.1 Kontext pipeline.
     
@@ -147,7 +147,7 @@ def infer(input_image, input_image_upload, overlay_image, prompt, seed=42, rando
             generator=torch.Generator().manual_seed(seed),
         ).images[0]
         
-    return image, processed_input_image, seed, gr.Button(visible=True)
+    return image, processed_input_image, seed, gr.Button(visible=False)
     
 @spaces.GPU
 def infer_example(input_image, prompt):
@@ -166,12 +166,12 @@ css=""
 with gr.Blocks(css=css) as demo:
     
     with gr.Column(elem_id="col-container"):
-        gr.Markdown(f"""# FLUX.1 Kontext [dev]
+        gr.Markdown(f"""# FLUX.1 Kontext [dev] + Draw2Photo LoRA
 Image editing and manipulation model guidance-distilled from FLUX.1 Kontext [pro], [[blog]](https://bfl.ai/announcements/flux-1-kontext-dev) [[model]](https://huggingface.co/black-forest-labs/FLUX.1-Kontext-dev)
         """)
         with gr.Row():
             with gr.Column():
-                gr.Markdown("Select/Upload/Draw a person ⬇️")
+                gr.Markdown("Step 1.  Select/Upload/Draw a person ⬇️")
                 # input_image = gr.Image(label="Upload drawing", type="pil")
                 with gr.Row():
                     with gr.Tabs() as tabs:
@@ -191,7 +191,7 @@ Image editing and manipulation model guidance-distilled from FLUX.1 Kontext [pro
                 )
 
             with gr.Column():
-                gr.Markdown("Select/Upload a face photo ⬇️")
+                gr.Markdown("Step 1.  Select/Upload a face photo ⬇️")
                 with gr.Row():
                     overlay_image = gr.Image(label="Upload face photo", type="pil")
                 gr.Examples(
@@ -200,7 +200,7 @@ Image editing and manipulation model guidance-distilled from FLUX.1 Kontext [pro
                 )
                     
             with gr.Column():
-                gr.Markdown("Press “Run” to get results")
+                gr.Markdown("Step 1.  Press “Run” to get results")
                 with gr.Row():
                     run_button = gr.Button("Run")
                 with gr.Accordion("Advanced Settings", open=False):
@@ -243,17 +243,17 @@ Image editing and manipulation model guidance-distilled from FLUX.1 Kontext [pro
                 reuse_button = gr.Button("Reuse this image", visible=False)
         
             
-        examples = gr.Examples(
-            examples=[
-                ["flowers.png", "turn the flowers into sunflowers"],
-                ["monster.png", "make this monster ride a skateboard on the beach"],
-                ["cat.png", "make this cat happy"]
-            ],
-            inputs=[input_image_upload, prompt],
-            outputs=[result, seed],
-            fn=infer_example,
-            cache_examples="lazy"
-        )
+        # examples = gr.Examples(
+        #     examples=[
+        #         ["flowers.png", "turn the flowers into sunflowers"],
+        #         ["monster.png", "make this monster ride a skateboard on the beach"],
+        #         ["cat.png", "make this cat happy"]
+        #     ],
+        #     inputs=[input_image_upload, prompt],
+        #     outputs=[result, seed],
+        #     fn=infer_example,
+        #     cache_examples="lazy"
+        # )
             
     gr.on(
         triggers=[run_button.click, prompt.submit],
@@ -261,10 +261,10 @@ Image editing and manipulation model guidance-distilled from FLUX.1 Kontext [pro
         inputs = [input_image, input_image_upload, overlay_image, prompt, seed, randomize_seed, guidance_scale, steps],
         outputs = [result, result_input, seed, reuse_button]
     )
-    reuse_button.click(
-        fn = lambda image: image,
-        inputs = [result],
-        outputs = [input_image]
-    )
+    # reuse_button.click(
+    #     fn = lambda image: image,
+    #     inputs = [result],
+    #     outputs = [input_image]
+    # )
 
 demo.launch(mcp_server=True)