|
from fastapi import FastAPI, File, UploadFile, HTTPException |
|
from fastapi.responses import JSONResponse |
|
from typing import List |
|
from rapidocr_onnxruntime import RapidOCR |
|
from PIL import Image |
|
import numpy as np |
|
import uvicorn |
|
from io import BytesIO |
|
|
|
app = FastAPI( |
|
title="OCR API", |
|
description="API for Optical Character Recognition using RapidOCR", |
|
version="1.0.0" |
|
) |
|
|
|
engine = RapidOCR() |
|
|
|
@app.get("/") |
|
async def index(): |
|
return {"message": "Welcome to the OCR API"} |
|
|
|
@app.post("/ocr") |
|
async def process_ocr(images: List[UploadFile] = File(...)): |
|
if not images: |
|
raise HTTPException(status_code=400, detail="No image files provided") |
|
|
|
text_results = [] |
|
|
|
for img_file in images: |
|
try: |
|
contents = await img_file.read() |
|
img = Image.open(BytesIO(contents)) |
|
|
|
width, height = img.size |
|
crop_top = 201 |
|
crop_area = (0, crop_top, width, height // 2) |
|
img_crop = img.crop(crop_area) |
|
|
|
img_array = np.array(img_crop) |
|
result, _ = engine(img_array) |
|
|
|
text_array = [item[1] for item in result] |
|
text_results.append(text_array) |
|
|
|
except Exception as e: |
|
raise HTTPException(status_code=500, detail=f"Error processing image: {str(e)}") |
|
|
|
return JSONResponse(content={"text": text_results}) |
|
|
|
if __name__ == "__main__": |
|
uvicorn.run("main:app", host="0.0.0.0", port=7860, reload=False) |
