from fastapi import FastAPI,UploadFile
import shutil
from fastapi.responses import HTMLResponse
from fastapi.responses import FileResponse
from fastapi.staticfiles import StaticFiles
from starlette.middleware.cors import CORSMiddleware 
import app as predictor
from PIL import Image, ImageFilter

app = FastAPI()

app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_methods=["'GET'","'POST'"],
    allow_headers=["*"]
)

app.mount("/view", StaticFiles(directory="view", html=True), name="view")
app.mount("/public", StaticFiles(directory="public", html=True), name="public")

@app.get("/")
def index() -> FileResponse:
    return FileResponse(path="./view/index.html", media_type="text/html")

@app.post("/prediction")
async def predict(targetImage: UploadFile):
  path = f'public/{targetImage.filename}'# api/filesディレクトリを作成しておく
  with open(path, 'wb+') as buffer:
    shutil.copyfileobj(targetImage.file, buffer)
  im = Image.open(path)
#   todo quality指定できるようにする
  depth_image, mesh_path = predictor.predict(im, 3)
  print(mesh_path)
  return {
  "path":"public",
  "name":mesh_path
  }