Updated .background remove to actually work

Author: cakedan

src/api/index.ts

@@ -687,7 +687,7 @@ export async function imageOverlayShutterstock(
 
 export async function imageToolsBackgroundRemove(
   context: RequestContext,
-  options: RestOptions.ImageBaseOptions,
+  options: RestOptions.ImageBackgroundRemoveOptions,
 ) {
   return raw.imageToolsBackgroundRemove(context, options);
 }

src/api/raw.ts

@@ -1392,9 +1392,10 @@ export async function imageOverlayShutterstock(
 
 export async function imageToolsBackgroundRemove(
   context: RequestContext,
-  options: RestOptions.ImageBaseOptions,
+  options: RestOptions.ImageBackgroundRemoveOptions,
 ): Promise<Response> {
   const query = {
+    model: options.model,
     url: options.url,
   };
   return request(context, {

src/api/types.ts

@@ -136,6 +136,10 @@ export namespace RestOptions {
     url: string,
   }
 
+  export interface ImageBackgroundRemoveOptions extends ImageBaseOptions {
+    model?: string,
+  }
+
   export interface ImageCreateTombstone {
     line1?: string,
     line2?: string,

src/commands/prefixed/image/background.remove.ts

@@ -1,17 +1,19 @@
 import { Command, CommandClient } from 'detritus-client';
 
 import { imageToolsBackgroundRemove } from '../../../api';
-import { CommandTypes } from '../../../constants';
+import { CommandTypes, ImageBackgroundRemovalModels } from '../../../constants';
 import { imageReply } from '../../../utils';
 
 import { BaseImageCommand } from '../basecommand';
 
 
 export interface CommandArgsBefore {
+  model?: string,
   url?: null | string,
 }
 
 export interface CommandArgs {
+  model?: string,
   url: string,
 }
 
@@ -23,14 +25,18 @@ export default class GrayscaleCommand extends BaseImageCommand<CommandArgs> {
       name: COMMAND_NAME,
 
       aliases: ['background rm', 'bg remove', 'bg rm'],
+      args: [
+        {name: 'model', choices: Object.values(ImageBackgroundRemovalModels), help: `Must be one of: (${Object.values(ImageBackgroundRemovalModels).join(', ')})`},
+      ],
       metadata: {
         description: 'Remove the background of an image',
         examples: [
           COMMAND_NAME,
           `${COMMAND_NAME} notsobot`,
+          `${COMMAND_NAME} notsobot -model U2NETP`,
         ],
         type: CommandTypes.IMAGE,
-        usage: '?<emoji,user:id|mention|name,url>',
+        usage: '?<emoji,user:id|mention|name,url> (-model <ImageBackgroundRemovalModels>)',
       },
     });
   }

src/constants.ts

@@ -482,6 +482,13 @@ export enum GuildPremiumTypes {
 };
 
 
+export enum ImageBackgroundRemovalModels {
+  ALPHA_MATTING = 'ALPHA_MATTING',
+  U2NET = 'U2NET',
+  U2NETP = 'U2NETP',
+  U2NET_HUMAN_SEG = 'U2NET_HUMAN_SEG',
+}
+
 
 export enum ImageEyeTypes {
   BIG = 'BIG',