from fastapi import File, UploadFile, HTTPException from ocr.api.message import ocr_router from ocr.api.message.schemas import OcrResponse from ocr.api.message.utils import divide_images, clean_response, extract_text_from_images, clean_text from ocr.core.wrappers import OcrResponseWrapper @ocr_router.post('/parse') async def get_all_chat_messages( file: UploadFile = File(...) ) -> OcrResponseWrapper[OcrResponse]: try: contents = await file.read() if file.filename.endswith('.pdf'): images = divide_images(contents) elif file.filename.endswith(('.jpg', ".jpeg", ".png")): images = [contents] else: raise HTTPException(status_code=400, detail='Unsupported file type.') text_content = extract_text_from_images(images) cleaned_original_text = clean_text(text_content) return OcrResponseWrapper(data=OcrResponse(text=clean_response(text_content), originalText=cleaned_original_text)) finally: await file.close()