from fastapi import FastAPI, File, UploadFile
from fastapi.staticfiles import StaticFiles
from fastapi.responses import FileResponse
from transformers import pipeline,AutoModel
import streamlit as st
from PIL import Image
import torch
import numpy as np
import urllib.request
import io
import os
from ofa_ocr import ofa_ocr_gr ##导入ofa_ocr中的大函数,方便使用其获取坐标的子函数

app = FastAPI()

@app.get("/inference")
def inference():
    return "<p>Hello, World!</p>"
    
@app.get("/infer_t5")
def t5(input):
    
    return {"output": input}

    
@app.post("/getCoordinates")
async def getCoordinates(image: UploadFile = File(...)):
    img = await image.read()
    image = Image.open(io.BytesIO(img)).convert("L").convert("RGB")
    path_to_image = "save.jpg"
    image.save(path_to_image)
    ocr_api_func = ofa_ocr_gr()
    result = ocr_api_func(path_to_image)
    os.remove(path_to_image)
    return result