Update app.py

app.py

@@ -2,7 +2,6 @@ import gradio as gr
 import requests
 from PIL import Image
 import numpy as np
-import io
 
 # Hàm gọi API Magic Studio để thực hiện xóa đối tượng
 def erase_object(image, mask, access_token):
@@ -10,6 +9,8 @@ def erase_object(image, mask, access_token):
     
     # Chuyển đổi hình ảnh và mask thành định dạng thích hợp để gửi qua API
     image_file = ('image_file', ('image.jpg', image, 'image/jpeg'))
+    
+    # Mask là một hình ảnh do người dùng vẽ trực tiếp từ Gradio Sketchpad
     mask_file = ('mask_file', ('mask.png', mask, 'image/png'))
     
     # Payload và headers
@@ -21,39 +22,32 @@ def erase_object(image, mask, access_token):
     
     if response.status_code == 200:
         # Trả về hình ảnh đã xóa đối tượng nếu thành công
-        return response.content  # Trả về nội dung hình ảnh từ API
+        return response.json()
     else:
         # Trả về thông báo lỗi nếu thất bại
         return f"Error: {response.status_code} - {response.text}"
 
 # Hàm xử lý cho Gradio
 def process(image, mask):
-    # Chuyển đổi hình ảnh và mask từ Gradio thành dữ liệu nhị phân để gửi qua API
-    image_bytes = io.BytesIO()
-    mask_bytes = io.BytesIO()
-    image.save(image_bytes, format="JPEG")
-    mask.save(mask_bytes, format="PNG")
+    # Chuyển đổi hình ảnh từ Gradio thành định dạng 'bytes'
+    image_bytes = image.convert("RGB").tobytes()
+    mask_bytes = mask.convert("RGB").tobytes()
     
     # Gọi API với hình ảnh và mask (token được cố định trong code)
-    access_token = 'eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJjbGllbnRfaWQiOiIyRGZQSm1sY1BLalBuaGh4ajBxSVpmQktfbS12RFhTc1NMRFNOX1gtZlVVIiwiZXhwIjoxNzI3MTY0MTcyLCJhcHBfbmFtZSI6IjE3NTMwODQiLCJtZXRhZGF0YSI6bnVsbCwiZ3JhbnRfdHlwZSI6ImNyZWQifQ.24oHGH9ial_gqeYrpUbYP68MwYly2ArzXbhbE1hGzQo'  # Sửa token của bạn ở đây
+    access_token = 'eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJjbGllbnRfaWQiOiIyRGZQSm1sY1BLalBuaGh4ajBxSVpmQktfbS12RFhTc1NMRFNOX1gtZlVVIiwiZXhwIjoxNzI3MTY0MTcyLCJhcHBfbmFtZSI6IjE3NTMwODQiLCJtZXRhZGF0YSI6bnVsbCwiZ3JhbnRfdHlwZSI6ImNyZWQifQ.24oHGH9ial_gqeYrpUbYP68MwYly2ArzXbhbE1hGzQo'
     
-    # Gửi hình ảnh và mask để xóa đối tượng
-    result = erase_object(image_bytes.getvalue(), mask_bytes.getvalue(), access_token)
+    result = erase_object(image_bytes, mask_bytes, access_token)
     
-    # Trả về ảnh kết quả (nếu có)
-    if isinstance(result, bytes):
-        return Image.open(io.BytesIO(result))
-    else:
-        return None
+    return result
 
 # Tạo giao diện Gradio
 with gr.Blocks() as demo:
     gr.Markdown("# Magic Eraser - Xóa đối tượng khỏi ảnh")
 
-    # Input cho ảnh và cho phép vẽ mask trực tiếp lên ảnh
+    # Input cho ảnh và Sketchpad để vẽ mask
     with gr.Row():
-        # Sử dụng gr.Image với chế độ vẽ mask
-        image_input = gr.Image(type="pil", tool="sketch", label="Tải ảnh lên và vẽ mask")
+        image_input = gr.Image(type="pil", label="Tải ảnh lên")
+        mask_input = gr.Sketchpad(shape=(512, 512), label="Vẽ mask để xóa đối tượng")
 
     # Nút thực hiện xử lý xóa đối tượng
     submit_button = gr.Button("Xóa đối tượng")
@@ -64,7 +58,7 @@ with gr.Blocks() as demo:
     # Kết nối các thành phần với nhau
     submit_button.click(
         process,
-        inputs=[image_input, image_input],  # Sử dụng cùng một ảnh cho cả đầu vào và mask
+        inputs=[image_input, mask_input],
         outputs=result_output
     )