LapStore
commited on
Commit
·
473fa66
1
Parent(s):
0c38a00
modified back mGE STEP 1
Browse files- app.py +13 -4
- index.html +5 -3
- server.py +3 -0
app.py
CHANGED
@@ -9,6 +9,7 @@ import numpy as np
|
|
9 |
import ast
|
10 |
from server import *
|
11 |
import cv2
|
|
|
12 |
|
13 |
#http://localhost:8000
|
14 |
app = FastAPI()
|
@@ -22,11 +23,19 @@ def main():
|
|
22 |
|
23 |
|
24 |
@app.post('/imageStep1')
|
25 |
-
async def image_step1(image_file: UploadFile = File(...),type_of_filters: str = Form(...), blur_radius: str = Form(...)):#--->,background_image: UploadFile = File(...)):
|
26 |
-
|
27 |
-
|
28 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
29 |
|
|
|
30 |
produced_image=SegmenterBackground().Back_step1(image,type_of_filters,int(blur_radius))[0]#---->
|
31 |
|
32 |
# Save the processed image to a temporary file
|
|
|
9 |
import ast
|
10 |
from server import *
|
11 |
import cv2
|
12 |
+
from typing import Optional
|
13 |
|
14 |
#http://localhost:8000
|
15 |
app = FastAPI()
|
|
|
23 |
|
24 |
|
25 |
@app.post('/imageStep1')
|
26 |
+
async def image_step1(image_file: UploadFile = File(...),background_image: Optional [UploadFile] = File(None),type_of_filters: str = Form(...), blur_radius: str = Form(...)):#--->,background_image: UploadFile = File(...)):
|
27 |
+
|
28 |
+
input_to_type_of_filters=None
|
29 |
+
if background_image:
|
30 |
+
# Process the image if provided
|
31 |
+
return {"filename": background_image.filename}
|
32 |
+
else:
|
33 |
+
# Handle the case where no image is provided
|
34 |
+
return {"message": "No image provided"}
|
35 |
+
contents_img = await image_file.read()
|
36 |
+
image = Image.open(io.BytesIO(contents_img))
|
37 |
|
38 |
+
|
39 |
produced_image=SegmenterBackground().Back_step1(image,type_of_filters,int(blur_radius))[0]#---->
|
40 |
|
41 |
# Save the processed image to a temporary file
|
index.html
CHANGED
@@ -2,11 +2,13 @@
|
|
2 |
<body bgcolor="#00cccc">
|
3 |
<center>
|
4 |
<br><br><br>
|
5 |
-
<form action="https://taha454-backg.hf.space/
|
6 |
<p><h3>Enter Image:</h3></p>
|
7 |
-
|
8 |
blur_radius :<p><input type="text" name="blur_radius" /></p><br>
|
9 |
-
|
|
|
|
|
10 |
<p><input type="submit" value="submit" /></p>
|
11 |
</form>
|
12 |
</center>
|
|
|
2 |
<body bgcolor="#00cccc">
|
3 |
<center>
|
4 |
<br><br><br>
|
5 |
+
<form action="https://taha454-backg.hf.space/imageStep1" method="post" enctype="multipart/form-data">
|
6 |
<p><h3>Enter Image:</h3></p>
|
7 |
+
type_of_filters :<p><input type="text" name="type_of_filters" /></p><br>
|
8 |
blur_radius :<p><input type="text" name="blur_radius" /></p><br>
|
9 |
+
Image : <input type="file" name="image_file" required><br><br>
|
10 |
+
Background_Image : <input type="file" name="background_image" required><br><br>
|
11 |
+
|
12 |
<p><input type="submit" value="submit" /></p>
|
13 |
</form>
|
14 |
</center>
|
server.py
CHANGED
@@ -10,6 +10,9 @@ import matplotlib.pyplot as plt
|
|
10 |
|
11 |
warnings.filterwarnings("ignore", category=FutureWarning)
|
12 |
|
|
|
|
|
|
|
13 |
class SegmenterBackground():
|
14 |
def __init__(self) -> None:
|
15 |
self.segment_names = {} # This dictionary will store names for segments across multiple inferences
|
|
|
10 |
|
11 |
warnings.filterwarnings("ignore", category=FutureWarning)
|
12 |
|
13 |
+
## use class to make each time inference not conflic with other object
|
14 |
+
|
15 |
+
|
16 |
class SegmenterBackground():
|
17 |
def __init__(self) -> None:
|
18 |
self.segment_names = {} # This dictionary will store names for segments across multiple inferences
|