Update app.py

app.py

@@ -23,11 +23,22 @@ def crop_image(image, crop_top_percentage, crop_bottom_percentage):
 
     return cropped_image
 
+def convert_to_jpg(input_image, output_path):
+    avif_image = Image.fromarray(input_image)
+    avif_image.convert('RGB').sav(output_path, 'JPEG')
+
 def resize_and_overlay_image(input_image, reduction_percentage, shift_percentage_lr, shift_percentage_ud, background_color, crop_top_percentage, crop_bottom_percentage):
     # Check if the input image is empty
     if input_image.size == 0:
         return None
 
+    if input_format == 'AVIF' and output_format == 'JPEG':
+        # Convert AVIF to JPEG
+        temp_jpg_path = "temp_image.jpg"
+        convert_to_jpg(input_image, temp_jpg_path)
+        input_image = cv2.imread(temp_jpg_path)
+        os.remove(temp_jpg_path)
+
     img = np.array(input_image)
 
     # Check if the image has shape information