Create app.py

app.py

@@ -0,0 +1,74 @@
+import gradio as gr
+import requests
+from PIL import Image, ImageOps
+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'
+    }
+    
+    # Chuyển đổi ảnh và mask thành định dạng PNG để gửi tới API
+    image_io = BytesIO()
+    mask_io = BytesIO()
+    
+    image.save(image_io, format="PNG")
+    mask.save(mask_io, format="PNG")
+    
+    image_io.seek(0)
+    mask_io.seek(0)
+    
+    files = [
+        ('image_file', ('image.png', image_io, 'image/png')),
+        ('mask_file', ('mask.png', mask_io, 'image/png'))
+    ]
+    
+    # Gửi yêu cầu POST tới API
+    response = requests.post(url, headers=headers, files=files)
+    
+    if response.status_code == 200:
+        # Nếu API trả về thành công, xử lý và trả về ảnh kết quả
+        return Image.open(BytesIO(response.content))
+    else:
+        # Nếu có lỗi, hiển thị lỗi
+        return f"API Error: {response.status_code} - {response.text}"
+
+# Hàm xử lý kết hợp ảnh và mask
+def process(image, 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)
+    
+    # Nếu kết quả trả về là lỗi, hiển thị thông báo
+    if isinstance(result, str):
+        return None, result
+    
+    # Trả về ảnh đã xử lý
+    return result, "Processing complete."
+
+# Tạo giao diện Gradio
+with gr.Blocks() as demo:
+    gr.Markdown("# AI Object Removal using Magic Studio API")
+    
+    # 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")
+    output_message = gr.Textbox(label="Message", lines=2)
+    
+    # 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])
+
+# Khởi chạy ứng dụng Gradio
+demo.launch()