Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import requests
-from PIL import Image
+from PIL import Image, ImageOps
 from io import BytesIO
 import numpy as np
 
@@ -37,13 +37,16 @@ def process_inpaint_via_api(image, mask):
         # Nếu có lỗi, hiển thị lỗi
         return f"API Error: {response.status_code} - {response.text}"
 
-# Hàm xử lý chính trong Gradio
+# Hàm xử lý kết hợp ảnh và mask
 def process(image, mask):
-    img = Image.fromarray(image)
-    mask_img = Image.fromarray(mask)
+    img = Image.fromarray(image).convert("RGBA")
+    
+    # Mask nhận được từ Image Sketchpad dưới dạng ảnh RGBA, cần chuyển sang ảnh nhị phân
+    mask = Image.fromarray(mask).convert("L")
+    mask = ImageOps.invert(mask)  # Invert lại mask để đảo ngược vùng vẽ
     
     # Gọi API với ảnh và mask
-    result = process_inpaint_via_api(img, mask_img)
+    result = process_inpaint_via_api(img, mask)
     
     # Nếu kết quả trả về là lỗi, hiển thị thông báo
     if isinstance(result, str):
@@ -56,11 +59,8 @@ def process(image, mask):
 with gr.Blocks() as demo:
     gr.Markdown("# AI Object Removal using Magic Studio API")
     
-    # Tải ảnh lên từ người dùng
-    image_input = gr.Image(type="numpy", label="Upload an image")
-
-    # Cho phép vẽ mask đè trực tiếp lên ảnh đã tải lên
-    mask_input = gr.Image(type="numpy", tool="sketch", label="Draw the mask on the image")
+    # Tải ảnh lên từ người dùng và cho phép vẽ mask đè trực tiếp lên ảnh đã tải lên
+    image_input = gr.Image(type="numpy", tool="sketch", label="Upload an image and draw mask")
 
     # Khu vực hiển thị kết quả
     result_output = gr.Image(label="Output Image")
@@ -68,7 +68,7 @@ with gr.Blocks() as demo:
     
     # Nút xử lý và kết nối các thành phần với hàm process
     process_button = gr.Button("Submit")
-    process_button.click(process, inputs=[image_input, mask_input], outputs=[result_output, output_message])
+    process_button.click(process, inputs=[image_input, image_input], outputs=[result_output, output_message])
 
 # Khởi chạy ứng dụng Gradio
 demo.launch()