o @Tj. @sdZddlmZmZmZmZmZddlmZddl m Z ddl m Z ddl mZeddgd Zejd e d ed ed ddeefdedede de fddZdS)u4Background removal (segmentation / 누끼) endpoint.) APIRouterDependsFileQuery UploadFile)get_segmentation_service)RemoveBackgroundResponse)SegmentationService)save_segmentation_imagez/api/v1/segmentation segmentation)prefixtagsz /remove-bg)response_model.Fz'Center-crop person in 9:16 aspect ratio) descriptionimage center_cropseg_svcreturncCs0|j}|j||d}t|}td|dS)a&Remove background from uploaded image. Accepts any common image format (JPEG, PNG, WebP). Saves the result as PNG with transparent background and returns the URL. When center_crop=true, the result is cropped to 9:16 with the person centered and minimal top/bottom margins. )rz /storage/) image_url)filereadremove_backgroundr r)rrr image_bytes result_bytes relative_pathr8/home/ubuntu/htdocs/ootd-api/app/routers/segmentation.pyr s rN)__doc__fastapirrrrrapp.dependenciesrapp.schemas.segmentationr!app.services.segmentation_servicer app.utils.imager routerpostboolrrrrrs,