diff --git a/server/app.py b/server/app.py index 5ac1e0c..83bb776 100644 --- a/server/app.py +++ b/server/app.py @@ -18,6 +18,7 @@ from dotenv import load_dotenv app = Flask(__name__) +app.config.from_object("config") # Get the CLIENT_URL environment variable, set a default if not found @@ -341,7 +342,7 @@ def download_image_mask(): response = requests.get(image_url) image = Image.open(BytesIO(response.content)) width, height = image.size - mask = Image.new('RGB', (width, height), (0, 0, 0)) # 'RGB' mode for colored masks + mask = Image.new('RGB', (width, height), app.config["MASK_BACKGROUND_COLOR"]) # 'RGB' mode for colored masks draw = ImageDraw.Draw(mask) for region in image_info.get("regions", []): diff --git a/server/config.py b/server/config.py new file mode 100644 index 0000000..a709cf7 --- /dev/null +++ b/server/config.py @@ -0,0 +1 @@ +MASK_BACKGROUND_COLOR = (0, 0, 0) \ No newline at end of file