cerbervs/doctr-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

style: reformat

Browse Source
This commit is contained in:
Mark Bailey 2024-10-25 18:12:02 -04:00
parent 3960d106c8
commit fb20c58fe4
1 changed files with 18 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/main.py
View File

@ -47,15 +47,17 @@ 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)
with open("data/"+file_name, "wb") as f: with open("data/" + file_name, "wb") as f:
f.write(b64decode(file.file_contents)) f.write(b64decode(file.file_contents))
print("Working on: " + file_name) print("Working on: " + file_name)
@ -68,25 +70,37 @@ def file(file: Image):
assume_straight_pages=False, assume_straight_pages=False,
preserve_aspect_ratio=True, preserve_aspect_ratio=True,
) )
doc = DocumentFile.from_images("data/"+file_name) doc = DocumentFile.from_images("data/" + file_name)
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))
def convert_to_list(value): 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 = {}