diff --git a/src/api/src/main.py b/src/api/src/main.py index 0c8076a..cca7941 100644 --- a/src/api/src/main.py +++ b/src/api/src/main.py @@ -16,6 +16,7 @@ from create_book import ( import tempfile from io import BytesIO from fastapi.staticfiles import StaticFiles +from aiohttp import ClientResponseError app = FastAPI() BUILD_PATH = Path(__file__).parent / "build" @@ -107,6 +108,13 @@ async def handle_download( content='The story you tried to download does not exist or has been deleted. Support is available on the Discord', ) + except ClientResponseError: + # Rate-limit by Wattpad + return HTMLResponse( + status_code=429, + content='Unfortunately, the downloader got rate-limited by Wattpad. Please try again later. Support is available on the Discord', + ) + app.mount("/", StaticFiles(directory=BUILD_PATH), "static")