From 1ac1b1b59cbeb4383f8db480fca450fba1f79525 Mon Sep 17 00:00:00 2001 From: Sourcery AI Date: Fri, 25 Mar 2022 17:11:00 +0000 Subject: [PATCH] 'Refactored by Sourcery' --- ocr.py | 2 +- server.py | 10 ++++++---- 2 files changed, 7 insertions(+), 5 deletions(-) diff --git a/ocr.py b/ocr.py index c5b2794..69be57c 100644 --- a/ocr.py +++ b/ocr.py @@ -37,7 +37,7 @@ async def read_images_from_dir(dir_path, lang='eng', write_to_file=False): converted_text[os.path.join(dir_path, file_)] = text if write_to_file: for file_path, text in converted_text.items(): - _write_to_file(text, os.path.splitext(file_path)[0] + ".txt") + _write_to_file(text, f'{os.path.splitext(file_path)[0]}.txt') return converted_text def _write_to_file(text, file_path): diff --git a/server.py b/server.py index dee9adb..c97f104 100644 --- a/server.py +++ b/server.py @@ -33,10 +33,12 @@ async def bulk_extract_text(request: Request, bg_task: BackgroundTasks): @app.get("/api/v1/bulk_output/{task_id}") async def bulk_output(task_id): - text_map = {} - for file_ in os.listdir(task_id): - if file_.endswith("txt"): - text_map[file_] = open(os.path.join(task_id, file_)).read() + text_map = { + file_: open(os.path.join(task_id, file_)).read() + for file_ in os.listdir(task_id) + if file_.endswith("txt") + } + return {"task_id": task_id, "output": text_map} def _save_file_to_disk(uploaded_file, path=".", save_as="default"):