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

Commit b3b69ca

Browse files
authored
doc: update readme
1 parent 24ce19f commit b3b69ca

File tree

1 file changed

+10
-14
lines changed

1 file changed

+10
-14
lines changed

README.md

Lines changed: 10 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -164,13 +164,11 @@ GeminiImage.save_sync(generated_images, save_path="cached_folder")
164164
import asyncio
165165
from gemini import Gemini, GeminiImage
166166

167-
async def fetch_and_save_images_async(prompt: str):
167+
async def fetch_and_save_images_async(prompt: str, save_path: str="cached_folder"):
168168
response = await GeminiClient.generate_content_async(prompt)
169169

170170
generated_images = response.generated_images # Check response images [Dict]
171-
for image in generated_images:
172-
gemini_image = GeminiImage(url=image.url)
173-
await gemini_image.save(path="cached_folder") # Save to path asynchronously
171+
await GeminiImage.save(generated_images, save_path=save_path)
174172

175173
# Run the async function
176174
if __name__ == "__main__":
@@ -183,12 +181,12 @@ if __name__ == "__main__":
183181
import asyncio
184182
from gemini import Gemini, GeminiImage
185183

186-
async def fetch_and_save_images_async(prompt: str):
184+
async def fetch_and_save_images_async(prompt: str, save_path: str="cached_folder"):
187185
response = await GeminiClient.generate_content_async(prompt)
188186

189187
generated_images = response.generated_images # Check response images [Dict]
190-
bytes_images_dict = await GeminiImage.fetch_images_dict(generated_images, cookies) # Get bytes images dict asynchronously
191-
await GeminiImage.save_images(bytes_images_dict, path="cached_folder") # Save to path asynchronously
188+
image_data_dict = await GeminiImage.fetch_images_dict(generated_images) # Get bytes images dict asynchronously
189+
await GeminiImage.save_images(image_data_dict, save_path=save_path) # Save to path asynchronously
192190

193191
# Run the async function
194192
if __name__ == "__main__":
@@ -224,13 +222,11 @@ GeminiImage.save_sync(response_images, save_path="cached_folder")
224222
import asyncio
225223
from gemini import Gemini, GeminiImage
226224

227-
async def fetch_and_save_images_async(prompt: str):
225+
async def fetch_and_save_images_async(prompt: str, save_path: str="cached"):
228226
response = await GeminiClient.generate_content_async(prompt)
229227

230228
response_images = response.web_images # Check response images [Dict]
231-
for image in response_images:
232-
gemini_image = GeminiImage(url=image.url)
233-
await gemini_image.save(path="cached_folder") # Save to path asynchronously
229+
GeminiImage.save(response_images, save_path=save_path)
234230

235231
# Run the async function
236232
if __name__ == "__main__":
@@ -243,12 +239,12 @@ if __name__ == "__main__":
243239
import asyncio
244240
from gemini import Gemini, GeminiImage
245241

246-
async def fetch_and_save_images_async(prompt: str):
242+
async def fetch_and_save_images_async(prompt: str, save_path: str = "cached_folder"):
247243
response = await GeminiClient.generate_content_async(prompt)
248244

249245
response_images = response.web_images # Check response images [Dict]
250-
bytes_images_dict = await GeminiImage.fetch_images_dict(response_images, cookies) # Get bytes images dict asynchronously
251-
await GeminiImage.save_images(bytes_images_dict, path="cached_folder") # Save to path asynchronously
246+
bytes_images_dict = await GeminiImage.fetch_images_dict(response_images) # Get bytes images dict asynchronously
247+
await GeminiImage.save_images(bytes_images_dict, save_path=save_path) # Save to path asynchronously
252248

253249
# Run the async function
254250
if __name__ == "__main__":

0 commit comments

Comments
 (0)