@@ -164,13 +164,11 @@ GeminiImage.save_sync(generated_images, save_path="cached_folder")
164164import asyncio
165165from 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
176174if __name__ == " __main__" :
@@ -183,12 +181,12 @@ if __name__ == "__main__":
183181import asyncio
184182from 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
194192if __name__ == " __main__" :
@@ -224,13 +222,11 @@ GeminiImage.save_sync(response_images, save_path="cached_folder")
224222import asyncio
225223from 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
236232if __name__ == " __main__" :
@@ -243,12 +239,12 @@ if __name__ == "__main__":
243239import asyncio
244240from 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
254250if __name__ == " __main__" :
0 commit comments