tesalonikahtp commited on
Commit
e82342b
·
1 Parent(s): 88cf2e3

feat: add ratio size

Browse files
Files changed (1) hide show
  1. server.py +3 -1
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=(600, 800), bg_color=selected_bg)
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))