Update app.py

app.py

@@ -190,7 +190,19 @@ def update_selection(selected_state: gr.SelectData, sdxl_loras, is_new=False):
         gr.Gallery(selected_index=None)
     )
 
-
+def center_crop_image_as_square(img):
+    square_size = min(img.size)  # Use the smaller dimension of the image
+    
+    # Calculate the coordinates of the crop box
+    left = (img.width - square_size) / 2
+    top = (img.height - square_size) / 2
+    right = (img.width + square_size) / 2
+    bottom = (img.height + square_size) / 2
+    
+    # Perform the crop
+    img_cropped = img.crop((left, top, right, bottom))
+    return img_cropped
+    
 def check_selected(selected_state):
     if not selected_state:
         raise gr.Error("You must select a LoRA")
@@ -425,6 +437,12 @@ with gr.Blocks(css="custom.css") as demo:
         inputs=[selected_state],
         queue=False,
         show_progress=False
+    ).success(
+        fn=center_crop_image_as_square,
+        inputs=[photo],
+        outputs=[photo],
+        queue=False,
+        show_progress=False,
     ).success(
         fn=run_lora,
         inputs=[photo, prompt, negative, weight, selected_state, face_strength, image_strength, guidance_scale, depth_control_scale, gr_sdxl_loras, gr_sdxl_loras_new],
@@ -435,6 +453,12 @@ with gr.Blocks(css="custom.css") as demo:
         inputs=[selected_state],
         queue=False,
         show_progress=False
+    ).success(
+        fn=center_crop_image_as_square,
+        inputs=[photo],
+        outputs=[photo],
+        queue=False,
+        show_progress=False,
     ).success(
         fn=run_lora,
         inputs=[photo, prompt, negative, weight, selected_state, face_strength, image_strength, guidance_scale, depth_control_scale, gr_sdxl_loras, gr_sdxl_loras_new],