Update app.py

app.py

@@ -1,20 +1,9 @@
 import gradio as gr
-from PIL import Image, ImageStat
-import numpy as np
+from PIL import Image
 import os
 import shutil
 import random
 
-def get_dominant_color(image):
-    """Get the dominant color of an image as a single RGB value or return None."""
-    try:
-        # Convert the image to a smaller size to simplify color calculation
-        small_image = image.resize((1, 1))  # Reduce to 1x1 pixel
-        dominant_color = small_image.getpixel((0, 0))  # Get the only pixel's color
-        return dominant_color
-    except:
-        return None
-
 def resize_and_pad(image, target_size):
     """Resize the image and pad it to match target size."""
     img = image.convert("RGB")
@@ -33,13 +22,11 @@ def resize_and_pad(image, target_size):
     # Use high-quality resampling for resizing
     img = img.resize((new_width, new_height), Image.Resampling.LANCZOS)
 
-    # Determine padding color
-    dominant_color = get_dominant_color(img)
-    if dominant_color is None:
-        dominant_color = (255, 255, 255) if random.choice([True, False]) else (0, 0, 0)
+    # Randomly select black or white as the background color
+    padding_color = (255, 255, 255) if random.choice([True, False]) else (0, 0, 0)
 
     # Create a new image with the determined background color
-    new_img = Image.new("RGB", target_size, dominant_color)
+    new_img = Image.new("RGB", target_size, padding_color)
     offset = ((target_size[0] - new_width) // 2, (target_size[1] - new_height) // 2)
     new_img.paste(img, offset)
     return new_img