From d9caececd7b25029d4fed464eaf76b45c6cc3a58 Mon Sep 17 00:00:00 2001
From: xds <iam@voroninv.ru>
Date: Fri, 27 Feb 2026 14:44:14 +0300
Subject: [PATCH] nsfw mark api

---
 api/models/ExternalGenerationDTO.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/api/models/ExternalGenerationDTO.py b/api/models/ExternalGenerationDTO.py
index a56deca..4744015 100644
--- a/api/models/ExternalGenerationDTO.py
+++ b/api/models/ExternalGenerationDTO.py
@@ -12,7 +12,9 @@ class ExternalGenerationRequest(BaseModel):
     # Image can be provided as base64 string OR URL (one must be provided)
     image_data: Optional[str] = Field(None, description="Base64-encoded image data")
     image_url: Optional[str] = Field(None, description="URL to download image from")
-    
+
+    nsfw: bool = False
+
     # Generation metadata
     aspect_ratio: AspectRatios = AspectRatios.NINESIXTEEN  # "1:1","2:3","3:2","3:4","4:3","4:5","5:4","9:16","16:9","21:9"
     quality: Quality = Quality.ONEK