Update app.py

app.py

@@ -162,7 +162,7 @@ Swap faces using Flux.2 Klein 9B [Alissonerdx/BFS-Best-Face-Swap](https://huggin
                 type="pil"
             )
 
-        run_button = gr.Button("Swap Face", variant="primary", size="lg")
+        run_button = gr.Button("Swap Face", variant="primary", size="lg", visible=False)
 
         with gr.Accordion("Advanced Settings", open=False):
             seed = gr.Slider(
@@ -244,21 +244,23 @@ Swap faces using Flux.2 Klein 9B [Alissonerdx/BFS-Best-Face-Swap](https://huggin
     def auto_swap_wrapper(ref_face, target_img, s, rand_s, w, h, steps, cfg):
         """Only run face swap if both images are provided"""
         if ref_face is not None and target_img is not None:
-            return face_swap(ref_face, target_img, s, rand_s, w, h, steps, cfg)
-        return None, s
+            result = face_swap(ref_face, target_img, s, rand_s, w, h, steps, cfg)
+            # Show the button after first generation
+            return result[0], result[1], gr.update(visible=True)
+        return None, s, gr.update(visible=False)
 
     # Trigger on reference face upload/change
     reference_face.change(
         fn=auto_swap_wrapper,
         inputs=swap_inputs,
-        outputs=swap_outputs,
+        outputs=[comparison_slider, seed_output, run_button],
     )
 
     # Trigger on target image upload/change
     target_image.change(
         fn=auto_swap_wrapper,
         inputs=swap_inputs,
-        outputs=swap_outputs,
+        outputs=[comparison_slider, seed_output, run_button],
     )
 
 if __name__ == "__main__":