Skip to content
This repository was archived by the owner on Jun 12, 2024. It is now read-only.

Commit c775797

Browse files
authored
doc: code example
1 parent 8470e6a commit c775797

File tree

1 file changed

+23
-25
lines changed

1 file changed

+23
-25
lines changed

README.md

Lines changed: 23 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -300,12 +300,11 @@ client = Gemini(cookies=cookies)
300300
image_client = GeminiImage(cookies=cookies)
301301

302302
response = client.generate_content("Create illustrations of Seoul, South Korea.")
303-
304303
generated_images = response.generated_images # Check generated images [Dict]
305304

306-
await image_client.save(generated_images, "save_dir", cookies=cookies)
307-
# image_data_dict = await image_client.fetch_images_dict(generated_images, cookies=cookies)
308-
# await image_client.save_images(image_data_dict, "save_dir")
305+
await image_client.save(generated_images, "output")
306+
# image_data_dict = await image_client.fetch_images_dict(generated_images)
307+
# await image_client.save_images(image_data_dict, "output")
309308
```
310309

311310

@@ -322,8 +321,7 @@ await image_client.save(generated_images, "save_dir", cookies=cookies)
322321
cookies = {}
323322
client = Gemini(cookies=cookies)
324323
image_client = GeminiImage(cookies=cookies)
325-
326-
bytes_images_dict = image_client.fetch_images_dict_sync(generated_images, cookies) # Get bytes images dict
324+
bytes_images_dict = image_client.fetch_images_dict_sync(generated_images) # Get bytes images dict
327325

328326
for image_name, image_bytes in bytes_images_dict.items():
329327
print(image_name)
@@ -343,11 +341,11 @@ image_client = GeminiImage(cookies=cookies)
343341
response = client.generate_content("Create illustrations of Seoul, South Korea.")
344342
generated_images = response.generated_images # Check generated images [Dict]
345343

346-
image_client.save_sync(generated_images, save_path="save_dir", cookies=cookies) # Save default folder is `cached`
344+
image_client.save_sync(generated_images, save_path="output")
347345

348346
# You can use byte type image dict for printing images as follow:
349-
# bytes_images_dict = image_client.fetch_images_dict_sync(generated_images, cookies=cookies) # Get bytes images dict
350-
# image_client.save_images_sync(bytes_images_dict, path="save_dir", cookies=cookies) # Save to path
347+
# bytes_images_dict = image_client.fetch_images_dict_sync(generated_images) # Get bytes images dict
348+
# image_client.save_images_sync(bytes_images_dict, path="output") # Save to dir
351349
```
352350

353351
*Async downloader wrapper*
@@ -360,14 +358,14 @@ cookies = {}
360358
client = Gemini(cookies=cookies)
361359
image_client = GeminiImage(cookies=cookies)
362360

363-
async def save_generated_imagse(generated_imagse, save_path="save_dir", cookies=cookies):
364-
await image_client.save(generated_imagse, save_path=save_path, cookies=cookies)
361+
async def save_generated_imagse(generated_imagse, save_path="output"):
362+
await image_client.save(generated_imagse, save_path=save_path)
365363

366364
# Run the async function
367365
if __name__ == "__main__":
368366
cookies = {"key" : "value"}
369367
generated_imagse = response.generated_imagse
370-
asyncio.run(save_generated_imagse(generated_imagse, save_path="save_dir", cookies=cookies))
368+
asyncio.run(save_generated_imagse(generated_imagse, save_path="output"))
371369
```
372370

373371
`GeminiImage.save` method logic
@@ -380,15 +378,15 @@ cookies = {}
380378
client = Gemini(cookies=cookies)
381379
image_client = GeminiImage(cookies=cookies)
382380

383-
async def save_generated_imagse(generated_imagse, save_path="save_dir", cookies=cookies):
384-
image_data_dict = await image_client.fetch_images_dict(generated_imagse, cookies=cookies) # Get bytes images dict asynchronously
381+
async def save_generated_imagse(generated_imagse, save_path="output"):
382+
image_data_dict = await image_client.fetch_images_dict(generated_imagse) # Get bytes images dict asynchronously
385383
await image_client.save_images(image_data_dict, save_path=save_path)
386384

387385
# Run the async function
388386
if __name__ == "__main__":
389387
cookies = {"key" : "value"}
390388
generated_imagse = response.generated_imagse # Check response images [Dict]
391-
asyncio.run(save_generated_imagse(generated_imagse, save_path="save_dir", cookies=cookies))
389+
asyncio.run(save_generated_imagse(generated_imagse, save_path="output"))
392390
```
393391

394392
</details>
@@ -414,9 +412,9 @@ response = client.generate_content("Create illustrations of Seoul, South Korea."
414412

415413
response_images = response.web_images # Check generated images [Dict]
416414

417-
await image_client.save(response_images, "save_dir")
415+
await image_client.save(response_images, "output")
418416
# image_data_dict = await image_client.fetch_images_dict(response_images)
419-
# await image_client.save_images(image_data_dict, "save_dir")
417+
# await image_client.save_images(image_data_dict, "output")
420418
```
421419

422420
<details><summary>Further</summary>
@@ -433,11 +431,11 @@ image_client = GeminiImage(cookies=cookies)
433431
response = client.generate_content("Please recommend a travel itinerary for Seoul.")
434432
response_images = response.web_images # Check response images [Dict]
435433

436-
GeminiImage.save_sync(response_images, save_path="save_dir")
434+
GeminiImage.save_sync(response_images, save_path="output")
437435

438436
# You can use byte type image dict as follow:
439-
# bytes_images_dict = image_client.fetch_bytes_sync(response_images, cookies) # Get bytes images dict
440-
# image_client.save_images_sync(bytes_images_dict, path="save_dir") # Save to path
437+
# bytes_images_dict = image_client.fetch_bytes_sync(response_images) # Get bytes images dict
438+
# image_client.save_images_sync(bytes_images_dict, save_path="output") # Save to path
441439
```
442440

443441
*Async downloader wrapper*
@@ -449,14 +447,14 @@ cookies = {}
449447
client = Gemini(cookies=cookies)
450448
image_client = GeminiImage(cookies=cookies)
451449

452-
async def save_response_web_imagse(response_images, save_path="save_dir"):
450+
async def save_response_web_imagse(response_images, save_path="output"):
453451
await image_client.save(response_images, save_path=save_path)
454452

455453
# Run the async function
456454
if __name__ == "__main__":
457455
cookies = {"key" : "value"}
458456
response_images = response.web_images
459-
asyncio.run(save_response_web_imagse(response_images, save_path="save_dir"))
457+
asyncio.run(save_response_web_imagse(response_images, save_path="output"))
460458
```
461459

462460
`GeminiImage.save` method logic
@@ -469,14 +467,14 @@ cookies = {}
469467
client = Gemini(cookies=cookies)
470468
image_client = GeminiImage(cookies=cookies)
471469

472-
async def save_response_web_imagse(response_images, save_path="save_dir", cookies=cookies):
473-
image_data_dict = await image_client.fetch_images_dict(response_images, cookies=cookies) # Get bytes images dict asynchronously
470+
async def save_response_web_imagse(response_images, save_path="output"):
471+
image_data_dict = await image_client.fetch_images_dict(response_images) # Get bytes images dict asynchronously
474472
await image_client.save_images(image_data_dict, save_path=save_path)
475473

476474
# Run the async function
477475
if __name__ == "__main__":
478476
response_images = response.web_images # Check response images [Dict]
479-
asyncio.run(save_response_web_imagse(response_images, save_path="save_dir", cookies=cookies))
477+
asyncio.run(save_response_web_imagse(response_images, save_path="output"))
480478
```
481479

482480
</details>

0 commit comments

Comments
 (0)