Spaces:
Sleeping
Sleeping
File size: 1,918 Bytes
96c3387 669df7a 1be5f36 ab59b01 669df7a 96c3387 669df7a d5a57f1 669df7a 96c3387 669df7a 96c3387 5fec972 96c3387 52b559d 669df7a 5fec972 669df7a 52b559d 669df7a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 |
import os
import tempfile
import shutil
import logging
os.system('pip install paddlepaddle==2.4.2')
os.system('pip install paddleocr')
from paddleocr import PaddleOCR, draw_ocr
from PIL import Image
import gradio as gr
import torch
from fastapi import FastAPI, File, UploadFile, Form
from fastapi.responses import FileResponse
import uvicorn
# Configure logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)
CUSTOM_PATH = "/gradio"
app = FastAPI()
@app.get("/")
def read_main():
return {"message": "This is your main app"}
io = gr.Interface(lambda x: "Hello, " + x + "!", "textbox", "textbox")
torch.hub.download_url_to_file('https://i.imgur.com/aqMBT0i.jpg', 'example.jpg')
@app.post("/ocr/")
async def ocr_endpoint(img: UploadFile = File(...), lang: str = Form(...)):
logger.info("Processing OCR request")
# Save the uploaded image to a temporary file
with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as temp_img:
shutil.copyfileobj(img.file, temp_img)
img_path = temp_img.name
# Perform OCR
ocr = PaddleOCR(use_angle_cls=True, lang=lang, use_gpu=False)
result = ocr.ocr(img_path, cls=True)[0]
boxes = [line[0] for line in result]
txts = [line[1][0] for line in result]
scores = [line[1][1] for line in result]
image = Image.open(img_path).convert('RGB')
im_show = draw_ocr(image, boxes, txts=None, scores=None, font_path='simfang.ttf')
im_show = Image.fromarray(im_show)
result_img_path = 'result.jpg'
im_show.save(result_img_path)
# Prepare the response
response_data = {
"result_image": result_img_path,
"ocr_result": result,
"extracted_text": '\n'.join(txts)
}
logger.info("OCR request processed successfully")
return response_data
app = gr.mount_gradio_app(app, io, path=CUSTOM_PATH)
uvicorn.run(app, host="0.0.0.0", port=7860) |