Update app.py
Browse files
app.py
CHANGED
|
@@ -87,7 +87,13 @@ def process_image(image, prompt):
|
|
| 87 |
else:
|
| 88 |
# Trường hợp input không được hỗ trợ
|
| 89 |
raise ValueError(f"Unsupported image type: {type(image)}")
|
| 90 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 91 |
# 2. Giảm Chất Lượng (quality=80) và Lưu tệp vào tệp tạm thời
|
| 92 |
# Lưu dưới định dạng JPEG để tối ưu hóa việc nén
|
| 93 |
img.save(
|
|
|
|
| 87 |
else:
|
| 88 |
# Trường hợp input không được hỗ trợ
|
| 89 |
raise ValueError(f"Unsupported image type: {type(image)}")
|
| 90 |
+
# KIỂM TRA VÀ CHUYỂN ĐỔI CHẾ ĐỘ MÀU TỪ RGBA SANG RGB
|
| 91 |
+
if img.mode == 'RGBA':
|
| 92 |
+
# Tạo một nền trắng (hoặc bất kỳ màu nào bạn muốn)
|
| 93 |
+
background = Image.new('RGB', img.size, (255, 255, 255))
|
| 94 |
+
# Dán hình ảnh RGBA lên trên nền RGB
|
| 95 |
+
background.paste(img, mask=img.split()[3]) # img.split()[3] là kênh Alpha
|
| 96 |
+
img = background
|
| 97 |
# 2. Giảm Chất Lượng (quality=80) và Lưu tệp vào tệp tạm thời
|
| 98 |
# Lưu dưới định dạng JPEG để tối ưu hóa việc nén
|
| 99 |
img.save(
|