uyen22 commited on
Commit
43dffeb
·
verified ·
1 Parent(s): 72ae796

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -74
app.py DELETED
@@ -1,74 +0,0 @@
1
- import gradio as gr
2
- import requests
3
- from PIL import Image, ImageOps
4
- from io import BytesIO
5
- import numpy as np
6
-
7
- # Hàm xử lý gửi ảnh và mask tới API Magic Studio
8
- def process_inpaint_via_api(image, mask):
9
- url = "https://api.magicstudio.com/magiceraser/erase"
10
-
11
- headers = {
12
- 'accessToken': 'eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJjbGllbnRfaWQiOiIyRGZQSm1sY1BLalBuaGh4ajBxSVpmQktfbS12RFhTc1NMRFNOX1gtZlVVIiwiZXhwIjoxNzI3MTY0MTcyLCJhcHBfbmFtZSI6IjE3NTMwODQiLCJtZXRhbnRfdHlwZSI6ImNyZWQifQ.24oHGH9ial_gqeYrpUbYP68MwYly2ArzXbhbE1hGzQo'
13
- }
14
-
15
- # Chuyển đổi ảnh và mask thành định dạng PNG để gửi tới API
16
- image_io = BytesIO()
17
- mask_io = BytesIO()
18
-
19
- image.save(image_io, format="PNG")
20
- mask.save(mask_io, format="PNG")
21
-
22
- image_io.seek(0)
23
- mask_io.seek(0)
24
-
25
- files = [
26
- ('image_file', ('image.png', image_io, 'image/png')),
27
- ('mask_file', ('mask.png', mask_io, 'image/png'))
28
- ]
29
-
30
- # Gửi yêu cầu POST tới API
31
- response = requests.post(url, headers=headers, files=files)
32
-
33
- if response.status_code == 200:
34
- # Nếu API trả về thành công, xử lý và trả về ảnh kết quả
35
- return Image.open(BytesIO(response.content))
36
- else:
37
- # Nếu có lỗi, hiển thị lỗi
38
- return f"API Error: {response.status_code} - {response.text}"
39
-
40
- # Hàm xử lý kết hợp ảnh và mask
41
- def process(image, mask):
42
- img = Image.fromarray(image).convert("RGBA")
43
-
44
- # Mask nhận được từ Image Sketchpad dưới dạng ảnh RGBA, cần chuyển sang ảnh nhị phân
45
- mask = Image.fromarray(mask).convert("L")
46
- mask = ImageOps.invert(mask) # Invert lại mask để đảo ngược vùng vẽ
47
-
48
- # Gọi API với ảnh và mask
49
- result = process_inpaint_via_api(img, mask)
50
-
51
- # Nếu kết quả trả về là lỗi, hiển thị thông báo
52
- if isinstance(result, str):
53
- return None, result
54
-
55
- # Trả về ảnh đã xử lý
56
- return result, "Processing complete."
57
-
58
- # Tạo giao diện Gradio
59
- with gr.Blocks() as demo:
60
- gr.Markdown("# AI Object Removal using Magic Studio API")
61
-
62
- # 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
63
- image_input = gr.Image(type="numpy", tool="sketch", label="Upload an image and draw mask")
64
-
65
- # Khu vực hiển thị kết quả
66
- result_output = gr.Image(label="Output Image")
67
- output_message = gr.Textbox(label="Message", lines=2)
68
-
69
- # Nút xử lý và kết nối các thành phần với hàm process
70
- process_button = gr.Button("Submit")
71
- process_button.click(process, inputs=[image_input, image_input], outputs=[result_output, output_message])
72
-
73
- # Khởi chạy ứng dụng Gradio
74
- demo.launch()