Synced repo using 'sync_with_huggingface' Github Action

app.py

@@ -1,7 +1,101 @@
 import gradio as gr
+from PIL import Image
+import tempfile
+import os
+from image_processor import process_image
 
-def greet(name):
-    return "Hello " + name + "!!"
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+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()