from fastapi import FastAPI, File, UploadFile
from fastapi.responses import JSONResponse
from fastapi.middleware.cors import CORSMiddleware
import io
from PIL import Image

from preprocessing.image_preprocessing import prep_image
from model_predictor import prediction

app = FastAPI()

app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],  
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

print("Server is running...")

@app.get("/")
async def root():
    return JSONResponse(
        content={"message": "Welcome to the Food Classification API 🍕🍔🥗! The server is up and running 🚀."}
    )

@app.post("/predict")
async def predict(file: UploadFile = File(...)):
    try:
        image = Image.open(io.BytesIO(await file.read()))

        processed_img = prep_image(image)
        predictions = prediction(processed_img)

        return JSONResponse(content={"predictions": predictions})
    
    except Exception as e:
        return JSONResponse(content={"error": str(e)}, status_code=500)

if __name__ == "__main__":
    import uvicorn
    uvicorn.run(app, host="0.0.0.0", port=7860)