app/services/gemini_error_handling.py

def handle_model_error(code: int, content: str) -> str:
    response = switch_case(code)
    return response


def switch_case(code: int) -> str:
    match code:
        case 503:
            return "Model is overloaded. Please try again later"
        case _:
            return f"Something went wrong while prompting: {code}"