Update app.py

Add crop option

app.py

@@ -10,7 +10,7 @@ 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 resize_and_overlay_image(input_image, reduction_percentage, shift_pixels, shift_pixels_ud, background_color):
+def resize_and_overlay_image(input_image, reduction_percentage, shift_pixels, shift_pixels_ud, background_color, crop_top, crop_bottom):
     # Check if the input image is empty
     if input_image.size == 0:
         return None
@@ -24,6 +24,9 @@ def resize_and_overlay_image(input_image, reduction_percentage, shift_pixels, sh
     # Get the image dimensions
     height, width = img.shape[:2]
 
+    # Crop the top and/or bottom of the image
+    img = img[crop_top:height - crop_bottom, :]
+
     # Calculate the new dimensions based on the reduction percentage
     new_height = int(height * reduction_percentage / 100)
     new_width = int(width * reduction_percentage / 100)
@@ -55,12 +58,14 @@ iface = gr.Interface(
         gr.inputs.Slider(minimum=0, maximum=100, step=10, default=80, label="Percentage of Original"),
         gr.inputs.Slider(minimum=-150, maximum=150, step=10, default=0, label="Shift Pixels Left / Right"),
         gr.inputs.Slider(minimum=-150, maximum=250, step=10, default=0, label="Shift Pixels Up / Down"),
-        gr.inputs.Textbox(default="#ffffff", label="Background Color (Hex Code)")
+        gr.inputs.Textbox(default="#ffffff", label="Background Color (Hex Code)"),
+        gr.inputs.Slider(minimum=0, maximum=100, step=10, default=0, label="Crop Top (pixels)"),
+        gr.inputs.Slider(minimum=0, maximum=100, step=10, default=0, label="Crop Bottom (pixels)")
     ],
     outputs=gr.outputs.Image(type="numpy", label="Result"),
     title="Image Resizer",
-    description="Reduce the size of an image and overlay it on a colored background and shift it to the right."
+    description="Reduce the size of an image, overlay it on a colored background, shift it, and crop the top and/or bottom."
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()