main.py

from fastapi import FastAPI, Request, UploadFile, File
from fastapi.templating import Jinja2Templates
from fastapi.staticfiles import StaticFiles
import uvicorn
from src import const, preprocess
import os
import shutil
from pathlib import Path
import json

templates = Jinja2Templates(directory="./templates")

app = FastAPI()

app.mount(
    "/static",
    StaticFiles(directory=Path(__file__).parent.parent.absolute() / "static"),
    name="static",
)


@app.get('/')
async def home(request: Request):
    return templates.TemplateResponse("index.html", {"request": request})


@app.post('/predict')
async def predict(image: UploadFile = File(...)):
    temp_file = save_to_disk(image, path="temp", save_as='temp')
    result = preprocess.predict(temp_file)
    with open(const.diagnosis_dir + const.diseases[result]+".json", 'r', encoding='utf-8') as f:
    	diagnosis = json.load(f)
    return diagnosis
   


def save_to_disk(uploadedfile, path='.', save_as='default'):
    extension = os.path.splitext(uploadedfile.filename)[-1]
    temp_file = os.path.join(path, save_as+extension)
    with open(temp_file, 'wb') as buffer:
        shutil.copyfileobj(uploadedfile.file, buffer)
    return temp_file