import uvicorn
from fastapi import FastAPI, File, UploadFile
from starlette.responses import RedirectResponse

from application.components import predict, read_imagefile
from application.schema import Symptom
from application.components.prediction import symptom_check

app_desc = """<h2>Use this 🦠  API by uploading any image with `predict/image`</h2>
<h4> 🦠 Brain Tumor Prediction API - it is just for research learning</h4>
<br>by Olusegun Odewole - AIOT lab"""

app = FastAPI(title=' 🦠 Brain Tumor Prediction API', description=app_desc)


@app.get("/", include_in_schema=False)
async def index():
    return RedirectResponse(url="/docs")


@app.post("/predict/image")
async def predict_api(file: UploadFile = File(...)):
    extension = file.filename.split(".")[-1] in ("jpg", "jpeg", "png")
    if not extension:
        return "Image must be jpg or png format!"
    image = read_imagefile(await file.read())
    prediction = predict(image)

    return prediction




if __name__ == "__main__":
    uvicorn.run(app, debug=True)