main.py

import cv2
import numpy as np
import uvicorn
from starlette.responses import JSONResponse

from utils.clova import clova_ocr
from utils.image_preprocess import PreProcessor
from fastapi import FastAPI, HTTPException, File, UploadFile
from pydantic import BaseModel
import logging
import warnings

from utils.nutrition_runner import nutrition_run
from utils.pororo_ocr import PororoOcr

warnings.filterwarnings('ignore')

class ImageRequest(BaseModel):
    image_key: str

app = FastAPI()

ocr = PororoOcr()
preprocessor = PreProcessor()


logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

@app.get("/")
def health_check():
    return {"ping":"pong"}

@app.exception_handler(Exception)
def handle_unexpected_error(request, exc: Exception):
    return JSONResponse(
        status_code=500,
        content={"detail": f"An unexpected error occurred: {exc}"},
    )

@app.post("/parse_nutrients", status_code=201)
async def read_item(file: UploadFile = File(...)):
    contents = await file.read()

    result = clova_ocr(file, contents)

    if not result:
        raise HTTPException(status_code=422, detail='Text Recognition Fail')
    else:
        return result

if __name__ == '__main__':
    uvicorn.run(
        app='main:app',
        host='0.0.0.0',
        port=8000,
        workers=1,
        access_log=False,
    )