style: reformat

This commit is contained in:
Mark Bailey 2024-10-25 18:12:02 -04:00
parent 3960d106c8
commit fb20c58fe4

View File

@ -47,10 +47,12 @@ def root():
return Response(renderTemplate("main.html", {"results": result})) return Response(renderTemplate("main.html", {"results": result}))
class Image(BaseModel): class Image(BaseModel):
file_name: str file_name: str
file_contents: str file_contents: str
@app.post("/detect/") @app.post("/detect/")
def file(file: Image): def file(file: Image):
file_name = unquote(file.file_name) file_name = unquote(file.file_name)
@ -72,7 +74,11 @@ def file(file: Image):
pred_res = predictor(doc) pred_res = predictor(doc)
json_res = pred_res.export() json_res = pred_res.export()
converted = convert_dict_items_to_list(json_res) converted = convert_dict_items_to_list(json_res)
os.unlink("data/" + file_name)
return JSONResponse(content=converted) return JSONResponse(content=converted)
except Exception as e: except Exception as e:
raise HTTPException(status_code=422, detail=str(e)) raise HTTPException(status_code=422, detail=str(e))
@ -81,12 +87,20 @@ def convert_to_list(value):
if isinstance(value, dict): if isinstance(value, dict):
return {k: convert_to_list(v) for k, v in value.items()} return {k: convert_to_list(v) for k, v in value.items()}
elif isinstance(value, list): 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): elif isinstance(value, np.ndarray):
return value.tolist() return value.tolist()
else: else:
return value return value
def convert_dict_items_to_list(d: dict): def convert_dict_items_to_list(d: dict):
converted = {} converted = {}