Update app.py

app.py

@@ -2,13 +2,14 @@ import gradio as gr
 import requests
 from PIL import Image
 from io import BytesIO
+import numpy as np
 
 # Hàm xử lý gửi ảnh và mask tới API Magic Studio
 def process_inpaint_via_api(image, mask):
     url = "https://api.magicstudio.com/magiceraser/erase"
     
     headers = {
-        'accessToken': 'eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJjbGllbnRfaWQiOiIyRGZQSm1sY1BLalBuaGh4ajBxSVpmQktfbS12RFhTc1NMRFNOX1gtZlVVIiwiZXhwIjoxNzI3MTY0MTcyLCJhcHBfbmFtZSI6IjE3NTMwODQiLCJtZXRhbnRfdHlwZSI6ImNyZWQifQ.24oHGH9ial_gqeYrpUbYP68MwYly2ArzXbhbE1hGzQo'
+        'accessToken': 'eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJjbGllbnRfaWQiOiIyRGZQSm1sY1BLalBuaGh4ajBxSVpmQktfbS12RFhTc1NMRFNOX1gtZlVVIiwiZXhwIjoxNzI3MTY0MTcyLCJhcHBfbmFtZSI6IjE3NTMwODQiLCJtZXRhZGF0YSI6bnVsbCwiZ3JhbnRfdHlwZSI6ImNyZWQifQ.24oHGH9ial_gqeYrpUbYP68MwYly2ArzXbhbE1hGzQo'
     }
     
     # Chuyển đổi ảnh và mask thành định dạng PNG để gửi tới API
@@ -55,9 +56,11 @@ def process(image, mask):
 with gr.Blocks() as demo:
     gr.Markdown("# AI Object Removal using Magic Studio API")
     
-    # Khu vực tải lên ảnh và vẽ mask
-    with gr.Row():
-        image_input = gr.Image(tool="sketch", type="numpy", label="Upload an image and draw the mask")
+    # Tải ảnh lên từ người dùng
+    image_input = gr.Image(source="upload", type="numpy", label="Upload an image")
+    
+    # Vẽ mask bằng Sketchpad (trên nền ảnh đã tải lên)
+    mask_input = gr.Sketchpad(shape=(512, 512), label="Draw the mask")
 
     # Khu vực hiển thị kết quả
     result_output = gr.Image(label="Output Image")
@@ -65,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, image_input], outputs=[result_output, output_message])
+    process_button.click(process, inputs=[image_input, mask_input], outputs=[result_output, output_message])
 
 # Khởi chạy ứng dụng Gradio
 demo.launch()