-
Notifications
You must be signed in to change notification settings - Fork 0
/
api.py
61 lines (43 loc) · 1.24 KB
/
api.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
from fastapi import FastAPI, File, UploadFile
import requests
from bg import remove
from PIL import Image
from io import BytesIO
import base64
from pydantic import BaseModel
class Item(BaseModel):
imageName: str
imageData: str
# Convert Image to Base64
def im_2_b64(image):
buff = BytesIO()
image.save(buff, format="PNG")
img_str = base64.b64encode(buff.getvalue())
#print(img_str)
return img_str
# Convert Base64 to Image
def b64_2_img(data):
buff = BytesIO(base64.b64decode(data))
return Image.open(buff)
app = FastAPI()
@app.get("/")
def read_root():
return {"Hello": "World"}
@app.post("/data")
def function1(data:Item):
return {"Hello": data}
@app.post("/removeBg")
def handler(data:Item):
#url = "https://unsplash.com/photos/E3LcqpQxtTU/download?force=true&w=640"
#response = requests.get(url)
#input_path = 'muh.jpg'
#output_path = 'output.png'
input = b64_2_img(data.imageData)
output = remove(input)
#buffered = BytesIO()
#output.save(buffered)
im_b64 = im_2_b64(output)
#print(im_b64)
#img_str = base64.b64encode(buffered.getvalue())
return "data:image/png:base64," + im_b64.decode("utf-8")
#return {"content": im_b64.decode("utf-8")}