Spaces:
Sleeping
Sleeping
Update render/canvas.py
Browse files- render/canvas.py +5 -7
render/canvas.py
CHANGED
|
@@ -16,7 +16,7 @@ class Canvas:
|
|
| 16 |
dots = [dots]
|
| 17 |
for dot in dots:
|
| 18 |
if dot[0]>=self.height or dot[1]>=self.width or dot[0]<0 or dot[1]<0:
|
| 19 |
-
|
| 20 |
continue
|
| 21 |
self.img.putpixel(dot, color + (255,))
|
| 22 |
|
|
@@ -27,23 +27,21 @@ class Canvas:
|
|
| 27 |
|
| 28 |
|
| 29 |
alpha = self.img.getchannel("A")
|
| 30 |
-
|
| 31 |
dilated_alpha = alpha.filter(ImageFilter.MaxFilter(size=5))
|
| 32 |
-
|
| 33 |
white_area = Image.new("RGBA", self.img.size, (255, 255, 255, 255))
|
| 34 |
white_area.putalpha(dilated_alpha)
|
| 35 |
|
| 36 |
|
| 37 |
result = Image.alpha_composite(white_area, self.img)
|
| 38 |
-
|
| 39 |
-
# white_border = Image.new("RGBA", image.size, (255, 255, 255, 255))
|
| 40 |
-
# white_border.putalpha(alpha)
|
| 41 |
return result
|
| 42 |
|
| 43 |
def __enter__(self):
|
| 44 |
return self
|
| 45 |
|
| 46 |
def __exit__(self, type, value, traceback):
|
| 47 |
-
|
| 48 |
self.img.save(self.filename)
|
| 49 |
pass
|
|
|
|
| 16 |
dots = [dots]
|
| 17 |
for dot in dots:
|
| 18 |
if dot[0]>=self.height or dot[1]>=self.width or dot[0]<0 or dot[1]<0:
|
| 19 |
+
|
| 20 |
continue
|
| 21 |
self.img.putpixel(dot, color + (255,))
|
| 22 |
|
|
|
|
| 27 |
|
| 28 |
|
| 29 |
alpha = self.img.getchannel("A")
|
| 30 |
+
|
| 31 |
dilated_alpha = alpha.filter(ImageFilter.MaxFilter(size=5))
|
| 32 |
+
|
| 33 |
white_area = Image.new("RGBA", self.img.size, (255, 255, 255, 255))
|
| 34 |
white_area.putalpha(dilated_alpha)
|
| 35 |
|
| 36 |
|
| 37 |
result = Image.alpha_composite(white_area, self.img)
|
| 38 |
+
|
|
|
|
|
|
|
| 39 |
return result
|
| 40 |
|
| 41 |
def __enter__(self):
|
| 42 |
return self
|
| 43 |
|
| 44 |
def __exit__(self, type, value, traceback):
|
| 45 |
+
|
| 46 |
self.img.save(self.filename)
|
| 47 |
pass
|