EdgeGPT —— Bing Chat 逆向工程

optional arguments:
-h, –help show this help message and exit
-U U Auth cookie from browser
–cookie-file COOKIE_FILE
File containing auth cookie
–prompt PROMPT Prompt to generate images for
–output-dir OUTPUT_DIR
Output directory
–quiet Disable pipeline messages
–asyncio Run ImageGen using asyncio

async def async_image_gen(args) -> None:
async with ImageGenAsync(args.U, args.quiet) as image_generator:
images = await image_generator.get_images(args.prompt)
await image_generator.save_images(images, output_dir=args.output_dir)

if __name__ == “__main__”:
parser = argparse.ArgumentParser()
parser.add_argument(“-U”, help=“Auth cookie from browser”, type=str)
parser.add_argument(“–cookie-file”, help=“File containing auth cookie”, type=str)
parser.add_argument(
“–prompt”,
help=“Prompt to generate images for”,
type=str,
required=True,
)
parser.add_argument(
“–output-dir”,
help=“Output directory”,
type=str,
default=“./output”,
)
args = parser.parse_args()
# Load auth cookie
with open(args.cookie_file, encoding=“utf-8”) as file:
cookie_json = json.load(file)
for cookie in cookie_json:
if cookie.get(“name”) == “_U”:
args.U = cookie.get(“value”)
break

if args.U is None:
raise Exception(“Could not find auth cookie”)

if not args.asyncio:
# Create image generator
image_generator = ImageGen(args.U, args.quiet)
image_generator.save_images(
image_generator.get_images(args.prompt),
output_dir=args.output_dir,
)
else:
asyncio.run(async_image_gen(args))