File size: 4,439 Bytes
786cdff c942e23 786cdff c942e23 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 |
import gradio as gr
from PIL import Image
import tempfile
import os
from image_processor import process_image
def apply_standard_settings(setting):
"""Returns the parameters for the selected standard setting."""
settings_dict = {
"S light": (True, True, "240x240", 48, "whitesmoke"),
"M light": (True, True, "480x480", 96, "whitesmoke"),
"L light": (True, True, "960x960", 128, "whitesmoke"),
"S dark": (True, True, "240x240", 48, "#2A373D"),
"M dark": (True, True, "480x480", 96, "#2A373D"),
"L dark": (True, True, "960x960", 128, "#2A373D"),
}
# Default to no special settings
return settings_dict.get(setting, (None, None, None, None, None))
def settings_description(crop, remove_bg, resize, padding, background):
"""Generate an HTML text description of the current settings in a smaller font and list format."""
description = f"""
<ul style="font-size:small;">
<li>Crop: {crop}</li>
<li>Remove Background: {remove_bg}</li>
<li>Resize: {resize if resize else 'No resize'}</li>
<li>Padding: {padding}</li>
<li>Background: {background}</li>
</ul>
"""
return description
def gradio_interface(image, standard_settings, crop=False, remove_bg=False, resize=None, padding=0, background="white"):
# Apply standard settings if selected and not "None"
if image is None:
# Load the standard image from the specified path if no image is uploaded
standard_image_path = 'https://f.uguu.se/FMSiaqYp.png'
image = Image.open(standard_image_path)
if standard_settings and standard_settings != "None":
crop, remove_bg, resize, padding, background = apply_standard_settings(
standard_settings)
# Generate settings description
applied_settings = settings_description(
crop, remove_bg, resize, padding, background)
# Convert resize string to tuple (if provided)
resize_dimensions = None
if resize:
try:
width, height = map(int, resize.split('x'))
resize_dimensions = (width, height)
except ValueError:
return "Invalid format for resize dimensions. Please use 'WxH'.", "original", applied_settings
# Process the image directly
processed_image = process_image(
image, crop, remove_bg, resize_dimensions, padding, background)
# Generate settings description
applied_settings = settings_description(
crop, remove_bg, resize, padding, background)
return processed_image, applied_settings
example_images = [
["https://f.uguu.se/FMSiaqYp.png", "S light", True, True, "480x420", 10, "whitesmoke"],
["https://f.uguu.se/RKKYgeDC.png", "None", True, True, "480x320", 48, "blue"],
["https://f.uguu.se/WMoKEkYc.png", "None", True, True, "360x360", 48, "yellow"],
]
# Define the Gradio interface
interface = gr.Interface(fn=gradio_interface,
inputs=[
gr.components.Image(
type="pil", label="Input Image"),
gr.components.Radio(choices=[
"None", "S light", "M light", "L light", "S dark", "M dark", "L dark"], label="Settings"),
gr.components.Checkbox(label="Crop"),
gr.components.Checkbox(label="Remove Background"),
gr.components.Textbox(
label="Resize (WxH)", placeholder="Example: 100x100"),
gr.components.Slider(
minimum=0, maximum=200, label="Padding"),
gr.components.Textbox(
label="Background", placeholder="Color name or hex code")
],
outputs=[
gr.components.Image(type="pil"),
gr.components.HTML(label="Applied Settings")
],
examples=example_images,
title="IMAGER ___ Image Processor",
description="Upload an image and select processing options or choose a standard setting. Supports crop, autoremove background, resize, add padding, and set the background color.",)
if __name__ == "__main__":
interface.launch() |