style: reformat
This commit is contained in:
parent
3960d106c8
commit
fb20c58fe4
16
src/main.py
16
src/main.py
@ -47,10 +47,12 @@ def root():
|
||||
|
||||
return Response(renderTemplate("main.html", {"results": result}))
|
||||
|
||||
|
||||
class Image(BaseModel):
|
||||
file_name: str
|
||||
file_contents: str
|
||||
|
||||
|
||||
@app.post("/detect/")
|
||||
def file(file: Image):
|
||||
file_name = unquote(file.file_name)
|
||||
@ -72,7 +74,11 @@ def file(file: Image):
|
||||
pred_res = predictor(doc)
|
||||
json_res = pred_res.export()
|
||||
converted = convert_dict_items_to_list(json_res)
|
||||
|
||||
os.unlink("data/" + file_name)
|
||||
|
||||
return JSONResponse(content=converted)
|
||||
|
||||
except Exception as e:
|
||||
raise HTTPException(status_code=422, detail=str(e))
|
||||
|
||||
@ -81,12 +87,20 @@ def convert_to_list(value):
|
||||
if isinstance(value, dict):
|
||||
return {k: convert_to_list(v) for k, v in value.items()}
|
||||
elif isinstance(value, list):
|
||||
return [convert_to_list(item) if isinstance(item, (dict, np.ndarray)) else item.tolist() if isinstance(item, np.ndarray) else item for item in value]
|
||||
return [
|
||||
convert_to_list(item)
|
||||
if isinstance(item, (dict, np.ndarray))
|
||||
else item.tolist()
|
||||
if isinstance(item, np.ndarray)
|
||||
else item
|
||||
for item in value
|
||||
]
|
||||
elif isinstance(value, np.ndarray):
|
||||
return value.tolist()
|
||||
else:
|
||||
return value
|
||||
|
||||
|
||||
def convert_dict_items_to_list(d: dict):
|
||||
converted = {}
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user