Update app.py

app.py

@@ -176,7 +176,7 @@ def generate_pose_transfer(
     
     # Enhance pose if requested
     if enhance_pose:
-        pose_image = extract_pose_edges(pose_image)
+        pose_image = process_pose_for_control(pose_image)
     
     # Prepare concatenated input with fixed size
     concatenated_input = prepare_images_for_kontext(reference_image, pose_image, target_size=512)
@@ -231,7 +231,7 @@ def generate_pose_transfer(
         if has_lora:
             try:
                 # Increase LoRA strength for pose control
-                actual_lora_scale = lora_scale * 1.0  # Boost LoRA influence
+                actual_lora_scale = lora_scale * 1.5  # Boost LoRA influence
                 pipe.set_adapters(["refcontrol"], adapter_weights=[actual_lora_scale])
                 print(f"LoRA adapter set with boosted strength: {actual_lora_scale}")
             except Exception as e:
@@ -562,13 +562,14 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
             guidance_scale,
             num_inference_steps,
             lora_scale,
+            controlnet_scale,
             enhance_pose
         ],
         outputs=[result_image, seed_used, concat_preview]
     )
     
     extract_btn.click(
-        fn=extract_pose_edges,
+        fn=process_pose_for_control,
         inputs=[extract_source],
         outputs=[pose_image]
     )
@@ -580,7 +581,7 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
     )
     
     reuse_pose_btn.click(
-        fn=extract_pose_edges,
+        fn=process_pose_for_control,
         inputs=[result_image],
         outputs=[pose_image]
     )