Update app.py

app.py

@@ -171,9 +171,13 @@ Image editing and manipulation model guidance-distilled from FLUX.1 Kontext [pro
         """)
         with gr.Row():
             with gr.Column():
-                # input_image = gr.Image(label="Upload the image for editing", type="pil")
+                gr.Markdown("Select/Upload/Draw a person ⬇️")
+                # input_image = gr.Image(label="Upload drawing", type="pil")
                 with gr.Row():
                     with gr.Tabs() as tabs:
+                        with gr.TabItem("Upload"):
+                            input_image_upload = gr.Image(label="Upload drawing", type="pil")
+                            
                         with gr.TabItem("Draw"):
                             input_image = gr.Paint(
                                     type="pil",
@@ -181,23 +185,33 @@ Image editing and manipulation model guidance-distilled from FLUX.1 Kontext [pro
                                     canvas_size = (1200,1200),
                                     layers = False
                                 )
-                        with gr.TabItem("Upload"):
-                            input_image_upload = gr.Image(label="Upload the drawing", type="pil")
                 gr.Examples(
                     examples=[[img] for img in BASE_EXAMPLES],
                     inputs=[input_image_upload],
                 )
+
+            with gr.Column():
+                gr.Markdown("Select/Upload a face photo ⬇️")
+                with gr.Row():
+                    overlay_image = gr.Image(label="Upload face photo", type="pil")
+                gr.Examples(
+                    examples=[[img] for img in FACE_EXAMPLES],
+                    inputs=[overlay_image],
+                )
+                    
+            with gr.Column():
+                gr.Markdown("Press “Run” to get results")
                 with gr.Row():
+                    run_button = gr.Button("Run")
+                with gr.Accordion("Advanced Settings", open=False):
+
                     prompt = gr.Text(
                         label="Prompt",
-                        show_label=False,
                         max_lines=1,
                         value = "make it real",
                         placeholder="Enter your prompt for editing (e.g., 'Remove glasses', 'Add a hat')",
                         container=False,
                     )
-                    run_button = gr.Button("Run", scale=0)
-                with gr.Accordion("Advanced Settings", open=False):
                     
                     seed = gr.Slider(
                         label="Seed",
@@ -224,15 +238,6 @@ Image editing and manipulation model guidance-distilled from FLUX.1 Kontext [pro
                         value=28,
                         step=1
                     )
-            with gr.Column():
-                with gr.Row():
-                    overlay_image = gr.Image(label="Upload face photo", type="pil")
-                gr.Examples(
-                    examples=[[img] for img in FACE_EXAMPLES],
-                    inputs=[overlay_image],
-                )
-                    
-            with gr.Column():
                 result = gr.Image(label="Result", show_label=False, interactive=False)
                 result_input = gr.Image(label="Result", show_label=False, interactive=False)
                 reuse_button = gr.Button("Reuse this image", visible=False)
@@ -244,7 +249,7 @@ Image editing and manipulation model guidance-distilled from FLUX.1 Kontext [pro
                 ["monster.png", "make this monster ride a skateboard on the beach"],
                 ["cat.png", "make this cat happy"]
             ],
-            inputs=[input_image, prompt],
+            inputs=[input_image_upload, prompt],
             outputs=[result, seed],
             fn=infer_example,
             cache_examples="lazy"