|
from fastapi import FastAPI, HTTPException
|
|
import random
|
|
from pydantic import BaseModel
|
|
from typing import List
|
|
from utils import decode_image
|
|
from mediapipe_preprocess import mediapipe_process
|
|
from model_predict import model_predict
|
|
import numpy as np
|
|
|
|
class Item(BaseModel):
|
|
text: str
|
|
number: int | None = None
|
|
|
|
class RecordFrames(BaseModel):
|
|
images: List[str]
|
|
|
|
app = FastAPI()
|
|
|
|
@app.get("/")
|
|
def return_hello():
|
|
return {"text":"Hello from server"}
|
|
|
|
@app.get("/random")
|
|
def return_random():
|
|
return {"random number":int(random.random()*1000)}
|
|
|
|
@app.post("/receive")
|
|
def return_received(item: Item):
|
|
return {"text":item.text,
|
|
"number":item.number}
|
|
|
|
@app.post("/predict")
|
|
def return_predict(record: RecordFrames):
|
|
if not record.images:
|
|
raise HTTPException(status_code=400, detail="No images provided")
|
|
|
|
frames = [np.array(decode_image(img)) for img in record.images]
|
|
keypoints = mediapipe_process(frames)
|
|
label = model_predict(keypoints)
|
|
return {"label":label} |
