Spaces:
Sleeping
Sleeping
tesalonikahtp commited on
Commit ·
e82342b
1
Parent(s): 88cf2e3
feat: add ratio size
Browse files
server.py
CHANGED
|
@@ -232,6 +232,7 @@ def create_app() -> Flask:
|
|
| 232 |
def generate_passport():
|
| 233 |
data = request.get_json()
|
| 234 |
bg_color_name = data.get('bg_color_name', 'white')
|
|
|
|
| 235 |
if not data.get("raw_photo"):
|
| 236 |
return jsonify({"error": "Missing required field: 'raw_photo'"}), 400
|
| 237 |
response = requests.get(data['raw_photo'], stream=True)
|
|
@@ -262,8 +263,9 @@ def create_app() -> Flask:
|
|
| 262 |
angle = info.get("chin_angle", 0.0)
|
| 263 |
|
| 264 |
# 4. Process
|
|
|
|
| 265 |
selected_bg = PASSPORT_COLORS.get(bg_color_name, (255, 255, 255))
|
| 266 |
-
cropper = PassportCropper(output_size=
|
| 267 |
|
| 268 |
img_clean = cropper.composite(img_rgb, mask)
|
| 269 |
img_rot, rot_box = cropper.rotate_and_expand_face(img_clean, angle, (x,y,x+w,y+h))
|
|
|
|
| 232 |
def generate_passport():
|
| 233 |
data = request.get_json()
|
| 234 |
bg_color_name = data.get('bg_color_name', 'white')
|
| 235 |
+
output_size = data.get('output_size', (600, 800))
|
| 236 |
if not data.get("raw_photo"):
|
| 237 |
return jsonify({"error": "Missing required field: 'raw_photo'"}), 400
|
| 238 |
response = requests.get(data['raw_photo'], stream=True)
|
|
|
|
| 263 |
angle = info.get("chin_angle", 0.0)
|
| 264 |
|
| 265 |
# 4. Process
|
| 266 |
+
# TODO: adjust bg color and size based on visa requirements
|
| 267 |
selected_bg = PASSPORT_COLORS.get(bg_color_name, (255, 255, 255))
|
| 268 |
+
cropper = PassportCropper(output_size=output_size, bg_color=selected_bg)
|
| 269 |
|
| 270 |
img_clean = cropper.composite(img_rgb, mask)
|
| 271 |
img_rot, rot_box = cropper.rotate_and_expand_face(img_clean, angle, (x,y,x+w,y+h))
|