diff --git a/lib/ocr.py b/lib/ocr.py index 8d1ed65..a6269f4 100644 --- a/lib/ocr.py +++ b/lib/ocr.py @@ -87,4 +87,7 @@ def get_room_data(img: bytes) -> list[Room]: return rooms def get_image_string(img: bytes) -> str: - return image_to_string(Image.open(BytesIO(img))) \ No newline at end of file + img_string = image_to_string(Image.open(BytesIO(img))) + img_string = correct_newlines(img_string) + img_string = correct_commas(img_string) + return img_string \ No newline at end of file