Merge branch 'main' of https://github.com/dsdanielpark/Gemini-API

Author: dsdanielpark

README.md

@@ -148,47 +148,55 @@ Returns images generated by Gemini.
 ```python
 from gemini import Gemini, GeminiImage
 
-prompt = "Create illustrations of Seoul, South Korea."
-response = GeminiClient.generate_content(prompt)
-
+response = GeminiClient.generate_content("Create illustrations of Seoul, South Korea.")
 generated_images = response.generated_images # Check generated images [Dict]
 
-GeminiImage.save_sync(generated_images, save_path="cached_folder")
+GeminiImage.save_sync(generated_images, save_path="cached")
 
 # You can use byte type image dict as follow:
 # bytes_images_dict = GeminiImage.fetch_images_dict_sync(generated_images, cookies) # Get bytes images dict
-# GeminiImage.save_images_sync(bytes_images_dict, path="cached_folder") # Save to path
+# GeminiImage.save_images_sync(bytes_images_dict, path="cached") # Save to path
 ```
 *Async*
+
 ```python
+response = GeminiClient.generate_content("Create illustrations of Seoul, South Korea.")
+
+generated_images = response.generated_images # Check generated images [Dict]
+
+await GeminiImage.save(generated_images, "cached")
+# image_data_dict = await GeminiImage.fetch_images_dict(generated_images)
+# await GeminiImage.save_images(image_data_dict, "cached")
+```
+
+<details><summary>Async wrapper</summary>
+
+```
 import asyncio
 from gemini import Gemini, GeminiImage
 
-async def fetch_and_save_images_async(prompt: str):
+async def fetch_and_save_images_async(prompt: str, save_path: str="cached"):
     response = await GeminiClient.generate_content_async(prompt)
 
     generated_images = response.generated_images  # Check response images [Dict]
-    for image in generated_images:
-        gemini_image = GeminiImage(url=image.url)
-        await gemini_image.save(path="cached_folder")  # Save to path asynchronously
+    await GeminiImage.save(generated_images, save_path=save_path)
 
 # Run the async function
 if __name__ == "__main__":
     user_prompt = input("Enter your prompt: ")
     asyncio.run(fetch_and_save_images_async(user_prompt))
 ```
-<details><summary>GeminiImage.save logics</summary>
-  
-```python
+`GeminiImage.save` method logic
+```
 import asyncio
 from gemini import Gemini, GeminiImage
 
-async def fetch_and_save_images_async(prompt: str):
+async def fetch_and_save_images_async(prompt: str, save_path: str="cached"):
     response = await GeminiClient.generate_content_async(prompt)
 
     generated_images = response.generated_images  # Check response images [Dict]
-    bytes_images_dict = await GeminiImage.fetch_images_dict(generated_images, cookies)  # Get bytes images dict asynchronously
-    await GeminiImage.save_images(bytes_images_dict, path="cached_folder")  # Save to path asynchronously
+    image_data_dict = await GeminiImage.fetch_images_dict(generated_images)  # Get bytes images dict asynchronously
+    await GeminiImage.save_images(image_data_dict, save_path=save_path)  # Save to path asynchronously
 
 # Run the async function
 if __name__ == "__main__":
@@ -213,52 +221,58 @@ response = GeminiClient.generate_content(prompt)
 
 response_images = response.web_images # Check response images [Dict]
 
-GeminiImage.save_sync(response_images, save_path="cached_folder")
+GeminiImage.save_sync(response_images, save_path="cached")
 
 # You can use byte type image dict as follow:
 # bytes_images_dict = GeminiImage.fetch_bytes_sync(response_images, cookies) # Get bytes images dict
-# GeminiImage.save_images_sync(bytes_images_dict, path="cached_folder") # Save to path
+# GeminiImage.save_images_sync(bytes_images_dict, path="cached") # Save to path
 ```
 *Async*
 ```python
+response = GeminiClient.generate_content("Create illustrations of Seoul, South Korea.")
+
+response_images = response.web_images # Check generated images [Dict]
+
+await GeminiImage.save(response_images, "cached")
+# image_data_dict = await GeminiImage.fetch_images_dict(response_images)
+# await GeminiImage.save_images(image_data_dict, "cached")
+```
+
+<details><summary>Async wrapper</summary>
+
+```
 import asyncio
 from gemini import Gemini, GeminiImage
 
-async def fetch_and_save_images_async(prompt: str):
+async def fetch_and_save_images_async(prompt: str, save_path: str="cached"):
     response = await GeminiClient.generate_content_async(prompt)
 
     response_images = response.web_images  # Check response images [Dict]
-    for image in response_images:
-        gemini_image = GeminiImage(url=image.url)
-        await gemini_image.save(path="cached_folder")  # Save to path asynchronously
+    await GeminiImage.save(response_images, save_path=save_path)
 
 # Run the async function
 if __name__ == "__main__":
     user_prompt = input("Enter your prompt: ")
     asyncio.run(fetch_and_save_images_async(user_prompt))
 ```
-<details><summary>GeminiImage.save logics</summary>
-  
-```python
+`GeminiImage.save` method logic
+```
 import asyncio
 from gemini import Gemini, GeminiImage
 
-async def fetch_and_save_images_async(prompt: str):
+async def fetch_and_save_images_async(prompt: str, save_path: str="cached"):
     response = await GeminiClient.generate_content_async(prompt)
 
     response_images = response.web_images  # Check response images [Dict]
-    bytes_images_dict = await GeminiImage.fetch_images_dict(response_images, cookies)  # Get bytes images dict asynchronously
-    await GeminiImage.save_images(bytes_images_dict, path="cached_folder")  # Save to path asynchronously
+    image_data_dict = await GeminiImage.fetch_images_dict(response_images)  # Get bytes images dict asynchronously
+    await GeminiImage.save_images(image_data_dict, save_path=save_path)  # Save to path asynchronously
 
 # Run the async function
 if __name__ == "__main__":
     user_prompt = input("Enter your prompt: ")
     asyncio.run(fetch_and_save_images_async(user_prompt))
 ```
 
-</details>
-
-
 <br>
 
 ## Further