app.py

from fastapi import FastAPI
from fastapi.responses import StreamingResponse
import fastapi as _fapi
import schemas as _schemas
import services as _services
import traceback


app = FastAPI()

@app.get("/")
def read_root():
    return {"message": "Welcome to AI Photo Enhancer API"}


# Endpoint to test the backend
@app.get("/api")
async def root():
    return {"message": "Welcome to the AI Photo Enhancer with FastAPI"}


@app.post("/api/enhance/")
async def enhance_image(enhanceBase: _schemas._EnhanceBase = _fapi.Depends()):
    
    try:
        encoded_img = await _services.enhance(enhanceBase=enhanceBase)
    except Exception as e:
        print(traceback.format_exc())
        return {"message": f"{e.args}"}
    
    payload = {
        "mime" : "image/jpg",
        "image": encoded_img
        }
    
    return payload