python --去除人像背景(抠图)

https://www.remove.bg/zh/api#api-reference

import time
import requests
import win32ui
import pyautogui
import PySimpleGUI as sg
import os

a = pyautogui.confirm(text='是否现在开始去除人像背景', title='小毅欧巴', buttons=['开始', '关闭'])
if a == '关闭':
    exit()

dlg = win32ui.CreateFileDialog(1)  # 参数 1 表示打开文件对话框
dlg.SetOFNInitialDir('C:')  # 设置打开文件对话框中的初始显示目录
dlg.DoModal()
filename = dlg.GetPathName()
file, ext = os.path.splitext(filename)
if ext not in ('.jpg', '.JPG', '.PNG', '.png'):
    pyautogui.alert(text='文件仅支持(jpg, png)', title='提示', button='关闭')
    exit()

pyautogui.alert(text=filename, title='确认路径', button='我已确认')

key = 'c6vLHWNS7fDFAvoMXk3uo4mB'
# rmbg = RemoveBg(key, "error.log")
# rmbg.remove_background_from_img_file(filename)
new_file_name = os.path.join(os.path.join(os.path.expanduser('~'), "Desktop"), 'no-bg.png')
with open(filename, 'rb') as f:
    response = requests.post(
        'https://api.remove.bg/v1.0/removebg',
        files={'image_file': f},
        data={
            'size': "regular",
            'bg_color': None
        },
        headers={'X-Api-Key': key})
    with open(new_file_name, 'wb') as removed_bg_file:
        removed_bg_file.write(response.content)

mylist = [1, 2, 3, 4, 5, 6, 7, 8]
for i, item in enumerate(mylist):
    sg.one_line_progress_meter('AI', i + 1, len(mylist), '正在智能清除背景')
    time.sleep(0.1)