Update app.py

app.py

@@ -10,6 +10,18 @@ def hex_to_rgb(hex_code):
     # Convert the hex code to RGB
     return tuple(int(hex_code[i:i+2], 16) for i in (0, 2, 4))
 
+def crop_image(image, crop_top_percentage, crop_bottom_percentage):
+    height, width = image.shape[:2]
+
+    # Calculate the number of pixels to crop from the top and bottom as a percentage of the image height
+    crop_top = int(height * crop_top_percentage / 100)
+    crop_bottom = int(height * crop_bottom_percentage / 100)
+
+    # Crop the input image
+    cropped_image = image[crop_top:height - crop_bottom, :]
+
+    return cropped_image
+
 def resize_and_overlay_image(input_image, reduction_percentage, shift_pixels, shift_pixels_ud, background_color, crop_top_percentage, crop_bottom_percentage):
     # Check if the input image is empty
     if input_image.size == 0:
@@ -21,19 +33,18 @@ def resize_and_overlay_image(input_image, reduction_percentage, shift_pixels, sh
     if img.ndim < 2:
         return None
 
-    # Get the image dimensions
-    height, width = img.shape[:2]
+    # Crop the input image
+    cropped_img = crop_image(img, crop_top_percentage, crop_bottom_percentage)
 
-    # Calculate the number of pixels to crop from the top and bottom as a percentage of the image height
-    crop_top = int(height * crop_top_percentage / 100)
-    crop_bottom = int(height * crop_bottom_percentage / 100)
+    # Get the dimensions of the cropped image
+    height, width = cropped_img.shape[:2]
 
     # Calculate the new dimensions based on the reduction percentage
     new_height = int(height * reduction_percentage / 100)
     new_width = int(width * reduction_percentage / 100)
 
-    # Resize the image
-    resized_img = cv2.resize(img, (new_width, new_height))
+    # Resize the cropped image
+    resized_img = cv2.resize(cropped_img, (new_width, new_height))
 
     # Convert the hex code to RGB
     background_rgb = hex_to_rgb(background_color)
@@ -68,7 +79,7 @@ iface = gr.Interface(
         gr.outputs.Textbox(label="Image Info")
     ],
     title="Image Resizer",
-    description="Overlay the resized input image on a new background of the specified color, shift it, and crop the top and/or bottom as a percentage."
+    description="Crop the input image, overlay it on a new background of the specified color, shift it, and resize it as a percentage."
 )
 
 if __name__ == "__main__":