From 5fd9ed5772b43e605a90c3a3048bfab61653da5e Mon Sep 17 00:00:00 2001
From: Darshan G K <122042809+imDarshanGK@users.noreply.github.com>
Date: Sun, 11 Jan 2026 16:39:51 +0000
Subject: [PATCH] Fix rate limit exception handling in /generate endpoint

---
 web/app.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/web/app.py b/web/app.py
index 8a4aa43..76279ed 100755
--- a/web/app.py
+++ b/web/app.py
@@ -31,6 +31,7 @@
 import metadata_manager
 import build_manager
 from builder import Builder
+from utils.ratelimiter import RateLimitExceededException
 
 # run at lower priority
 os.nice(20)
@@ -220,6 +221,9 @@ def generate():
         app.logger.info('Redirecting to /viewlog')
         return redirect('/viewlog/'+build_id)
 
+    except RateLimitExceededException as ex:
+        app.logger.warning(f"Rate limit exceeded for client: {request.remote_addr}")
+        return render_template('error.html', ex=ex), 429
     except Exception as ex:
         app.logger.error(ex)
         return render_template('error.html', ex=ex)