from typing import Union

from fastapi.responses import JSONResponse

def success_response(data: dict, success: bool = True, status_code: int = 200):
    return JSONResponse(status_code=status_code, content={
        "success": success,
        "error": None,
        "data": data
    })

def error_response(error: Union[str, Exception, dict], status_code: int = 500):
    if isinstance(error, dict):
        return JSONResponse(status_code=status_code, content={
            "success": False,
            "error": error,
            "data": None
        })
    return JSONResponse(status_code=status_code, content={
            "success": False,
            "error": str(error),
            "data": None
        })