mirror of
https://github.com/mealie-recipes/mealie.git
synced 2025-07-24 23:59:45 +02:00
39adea4ee3
* Add pytesseract * Add simple ocr endpoint replace extension argument * feat/ocr-editor gui * fix frontend linting issues * Add service unit tests * Add split text modes & single ingredient/instruction editing * make split mode really reactive * Remove default step and ingredient * make the linter haappy * Accept only image uploads * Add automatic recipe title suggestion * Correct regex * fix incorrect array.map method usage * make the linter happy again * Swap route to use asset name * Rearange buttons * fix test data * feat: Allow making image the recipe image * Add translation * Make the linter happy * Restrict function setPropertyValueByPath generic * Restrict template literal type * Add a more friendly icon to creation page * update poetry lock file * Correct sloppy ocr classes * Make MyPy happy * Rewrite safer tests * Add tesseract to backend test CI container dependencies * Make canvas element a component global * Remove unwanted spaces in selected text * Add way to know if recipe was created with ocr * Access to ocr-editor for ocr recipes * Update Alembic revision * Make the frontend build * Fix scrolling offset bug * Allow creation of recipes with custom settings * Fix rebasing mistakes * Add format_tsv_output test * Exclude the tests data directory only * Enforce camelCase for frontend functions * Remove import of unused component * Fix type and class initialization * Add multi-language support * Highlight words in mount * Fix image ratio bug * Better ocr creation page * Revert awkward feature to scroll in Selection mode * Rebasing alembic migrations sux * Remove obsolete getShared function * Add function docstring * Move down ocr creation option * Make toolbar icons more generic * Show help at the bottom of the page * move ocr types to own file * Use template ref for the canvas * Use i18n.tc to get strings directly * Correct naming mistake * Move Ocr editor to own directory * Create Ocr Editor parts * Safeguard recipe properties access * Add loading frontend animation due to longer request time * minor cleanup chores Co-authored-by: Miroito <alban.vachette@gmail.com>
37 lines
1.4 KiB
Python
37 lines
1.4 KiB
Python
from fastapi import APIRouter, File
|
|
|
|
from mealie.routes._base import BaseUserController, controller
|
|
from mealie.schema.ocr.ocr import OcrAssetReq, OcrTsvResponse
|
|
from mealie.services.ocr.pytesseract import OcrService
|
|
from mealie.services.recipe.recipe_data_service import RecipeDataService
|
|
from mealie.services.recipe.recipe_service import RecipeService
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@controller(router)
|
|
class OCRController(BaseUserController):
|
|
def __init__(self):
|
|
self.ocr_service = OcrService()
|
|
|
|
@router.post("/", response_model=str)
|
|
def image_to_string(self, file: bytes = File(...)):
|
|
return self.ocr_service.image_to_string(file)
|
|
|
|
@router.post("/file-to-tsv", response_model=list[OcrTsvResponse])
|
|
def file_to_tsv(self, file: bytes = File(...)):
|
|
tsv = self.ocr_service.image_to_tsv(file)
|
|
return self.ocr_service.format_tsv_output(tsv)
|
|
|
|
@router.post("/asset-to-tsv", response_model=list[OcrTsvResponse])
|
|
def asset_to_tsv(self, req: OcrAssetReq):
|
|
recipe_service = RecipeService(self.repos, self.user, self.group)
|
|
recipe = recipe_service._get_recipe(req.recipe_slug)
|
|
if recipe.id is None:
|
|
return []
|
|
data_service = RecipeDataService(recipe.id, recipe.group_id)
|
|
asset_path = data_service.dir_assets.joinpath(req.asset_name)
|
|
file = open(asset_path, "rb")
|
|
tsv = self.ocr_service.image_to_tsv(file.read())
|
|
|
|
return self.ocr_service.format_tsv_output(tsv)
|