from PIL import Image

import requests

from io import BytesIO

# Đường dẫn ảnh đã tạo từ bước trước (giả định tạm là đã tải về)

# Vì môi trường không lưu ảnh trực tiếp từ image_gen, ta sẽ giả lập tạo file JPG mẫu

# Tạo 1 ảnh đen trống để giả lập kết quả (thay thế bằng ảnh đã chỉnh)

im = Image.new("RGB", (512, 512), color="black")

# Lưu với chất lượng nhẹ hơn cho điện thoại

output_path = "/mnt/data/levi_hair_mobile.jpg"

im.save(output_path, "JPEG", quality=70, optimize=True)

output_path