Update app.py

app.py

@@ -31,14 +31,21 @@ def process_image_and_text(subject_image, adapter_dict, prompt, _unused1, _unuse
     seed, guidance_scale, steps = 42, 2.5, 28
 
     adapter_image = adapter_dict["image"] if isinstance(adapter_dict, dict) else adapter_dict
-    if isinstance(adapter_dict, dict) and adapter_dict.get("mask") is not None:
-        m = adapter_dict["mask"].convert("L").point(lambda p: 255 if p else 0)
-        if bbox := m.getbbox():
-            rect = Image.new("L", m.size, 0)
-            ImageDraw.Draw(rect).rectangle(bbox, fill=255)
-            m = rect
-        green = Image.new("RGB", adapter_image.size, "#00FF00")
-        adapter_image = Image.composite(green, adapter_image, m)
+    if isinstance(adapter_dict, dict):  # Image/sketch input
+        adapter_image = adapter_dict["image"]
+        adapter_mask = adapter_dict["mask"]
+        if adapter_mask is not None:
+            # convert mask -> solid green rectangle on copy of adapter_image
+            m = adapter_mask.convert("L").point(lambda p: 255 if p else 0)
+            bbox = m.getbbox()
+            if bbox:
+                rect = Image.new("L", m.size, 0)
+                ImageDraw.Draw(rect).rectangle(bbox, fill=255)
+                m = rect
+            green = Image.new("RGB", adapter_image.size, "#00FF00")
+            adapter_image = Image.composite(green, adapter_image, m)
+    else:
+        adapter_image = adapter_dict
 
     def prep(img: Image.Image):
         w, h = img.size