Update gradio_imager.py

gradio_imager.py

@@ -11,9 +11,9 @@ def apply_standard_settings(setting):
         "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, "dimgray"),
-        "M dark": (True, True, "480x480", 96, "dimgray"),
-        "L dark": (True, True, "960x960", 128, "dimgray"),
+        "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))
@@ -56,33 +56,33 @@ def gradio_interface(image, standard_settings, crop=False, remove_bg=False, resi
             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)
 
-    # Use a temporary file to save the input image from Gradio
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp_input:
-        image.save(tmp_input, format="PNG")
-        tmp_input_path = tmp_input.name
-
-    # Prepare a temporary file for the output image
-    tmp_output_path = tempfile.mktemp(suffix=".png")
+    # Generate settings description
+    applied_settings = settings_description(
+        crop, remove_bg, resize, padding, background)
 
-    # Process the image
-    process_image(tmp_input_path, tmp_output_path, crop,
-                  remove_bg, resize_dimensions, padding, background)
+    return processed_image, applied_settings
 
-    # Load and return the processed image
-    processed_image = Image.open(tmp_output_path)
 
-    # Clean up temporary files
-    os.remove(tmp_input_path)
-    os.remove(tmp_output_path)
+example_images = [
+    [os.path.join("data", "examples", "supermario.png"),
+     "S light", True, True, "480x420", 10, "whitesmoke"],
+    [os.path.join("data", "examples",
+                  "depositphotos_520707962-stock-photo-fujifilm-s10-body-black-fujifilm.jpg"), "None", True, True, "480x320", 48, "blue"],
+    [os.path.join("data", "examples", "batman_b_c_320x280_bg.png"),
+     "None", True, True, "360x360", 48, "yellow"],
 
-    return processed_image, applied_settings
 
+]
 
 # Define the Gradio interface
 interface = gr.Interface(fn=gradio_interface,
                          inputs=[
-                             gr.components.Image(type="pil"),
+                             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"),
@@ -98,6 +98,7 @@ interface = gr.Interface(fn=gradio_interface,
                              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.",)