From e050484b570876cdcb27ac9ee05f2c8c04ded7c9 Mon Sep 17 00:00:00 2001
From: dev <dev@cowmotors.com>
Date: Sat, 14 Jun 2025 18:33:16 +0200
Subject: [PATCH] Add size limit

---
 .env              | 1 +
 aperisolve/app.py | 6 ++++++
 2 files changed, 7 insertions(+)

diff --git a/.env b/.env
index 9048488..bdea06b 100644
--- a/.env
+++ b/.env
@@ -1,3 +1,4 @@
+MAX_CONTENT_LENGTH=1048576
 MAX_PENDING_TIME=600
 MAX_STORE_TIME=259200
 FLASK_DEBUG=0
diff --git a/aperisolve/app.py b/aperisolve/app.py
index bf43b1f..c409c11 100644
--- a/aperisolve/app.py
+++ b/aperisolve/app.py
@@ -20,6 +20,7 @@ app: Flask = Flask(__name__)
 app.json.sort_keys = False  # type: ignore
 app.config["SQLALCHEMY_DATABASE_URI"] = os.environ.get("DB_URI")
 app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False
+app.config["MAX_CONTENT_LENGTH"] = os.environ.get("MAX_CONTENT_LENGTH")
 RESULT_FOLDER.mkdir(parents=True, exist_ok=True)
 
 
@@ -29,6 +30,11 @@ redis_conn = Redis(host="redis", port=6379)
 queue = Queue(connection=redis_conn)
 
 
+@app.errorhandler(413)
+def too_large(e):
+    return jsonify({"error": "Image size exceeded"}), 413
+
+
 @app.route("/")
 def index() -> str:
     """Render the main index page."""