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

Commit ec95dcd

Browse files
authored
doc: fixed gemini image code example
1 parent b3b69ca commit ec95dcd

File tree

1 file changed

+40
-22
lines changed

1 file changed

+40
-22
lines changed

README.md

Lines changed: 40 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -148,23 +148,34 @@ Returns images generated by Gemini.
148148
```python
149149
from gemini import Gemini, GeminiImage
150150

151-
prompt = "Create illustrations of Seoul, South Korea."
152-
response = GeminiClient.generate_content(prompt)
153-
151+
response = GeminiClient.generate_content("Create illustrations of Seoul, South Korea.")
154152
generated_images = response.generated_images # Check generated images [Dict]
155153

156-
GeminiImage.save_sync(generated_images, save_path="cached_folder")
154+
GeminiImage.save_sync(generated_images, save_path="cached")
157155

158156
# You can use byte type image dict as follow:
159157
# bytes_images_dict = GeminiImage.fetch_images_dict_sync(generated_images, cookies) # Get bytes images dict
160-
# GeminiImage.save_images_sync(bytes_images_dict, path="cached_folder") # Save to path
158+
# GeminiImage.save_images_sync(bytes_images_dict, path="cached") # Save to path
161159
```
162160
*Async*
161+
163162
```python
163+
response = GeminiClient.generate_content("Create illustrations of Seoul, South Korea.")
164+
165+
generated_images = response.generated_images # Check generated images [Dict]
166+
167+
await GeminiImage.save(generated_images, "cached")
168+
# image_data_dict = await GeminiImage.fetch_images_dict(generated_images)
169+
# await GeminiImage.save_images(image_data_dict, "cached")
170+
```
171+
172+
<details><summary>Async wrapper</summary>
173+
174+
```
164175
import asyncio
165176
from gemini import Gemini, GeminiImage
166177
167-
async def fetch_and_save_images_async(prompt: str, save_path: str="cached_folder"):
178+
async def fetch_and_save_images_async(prompt: str, save_path: str="cached"):
168179
response = await GeminiClient.generate_content_async(prompt)
169180
170181
generated_images = response.generated_images # Check response images [Dict]
@@ -175,13 +186,12 @@ if __name__ == "__main__":
175186
user_prompt = input("Enter your prompt: ")
176187
asyncio.run(fetch_and_save_images_async(user_prompt))
177188
```
178-
<details><summary>GeminiImage.save logics</summary>
179-
180-
```python
189+
`GeminiImage.save` method logic
190+
```
181191
import asyncio
182192
from gemini import Gemini, GeminiImage
183193
184-
async def fetch_and_save_images_async(prompt: str, save_path: str="cached_folder"):
194+
async def fetch_and_save_images_async(prompt: str, save_path: str="cached"):
185195
response = await GeminiClient.generate_content_async(prompt)
186196
187197
generated_images = response.generated_images # Check response images [Dict]
@@ -211,50 +221,58 @@ response = GeminiClient.generate_content(prompt)
211221

212222
response_images = response.web_images # Check response images [Dict]
213223

214-
GeminiImage.save_sync(response_images, save_path="cached_folder")
224+
GeminiImage.save_sync(response_images, save_path="cached")
215225

216226
# You can use byte type image dict as follow:
217227
# bytes_images_dict = GeminiImage.fetch_bytes_sync(response_images, cookies) # Get bytes images dict
218-
# GeminiImage.save_images_sync(bytes_images_dict, path="cached_folder") # Save to path
228+
# GeminiImage.save_images_sync(bytes_images_dict, path="cached") # Save to path
219229
```
220230
*Async*
221231
```python
232+
response = GeminiClient.generate_content("Create illustrations of Seoul, South Korea.")
233+
234+
response_images = response.web_images # Check generated images [Dict]
235+
236+
await GeminiImage.save(response_images, "cached")
237+
# image_data_dict = await GeminiImage.fetch_images_dict(response_images)
238+
# await GeminiImage.save_images(image_data_dict, "cached")
239+
```
240+
241+
<details><summary>Async wrapper</summary>
242+
243+
```
222244
import asyncio
223245
from gemini import Gemini, GeminiImage
224246
225247
async def fetch_and_save_images_async(prompt: str, save_path: str="cached"):
226248
response = await GeminiClient.generate_content_async(prompt)
227249
228250
response_images = response.web_images # Check response images [Dict]
229-
GeminiImage.save(response_images, save_path=save_path)
251+
await GeminiImage.save(response_images, save_path=save_path)
230252
231253
# Run the async function
232254
if __name__ == "__main__":
233255
user_prompt = input("Enter your prompt: ")
234256
asyncio.run(fetch_and_save_images_async(user_prompt))
235257
```
236-
<details><summary>GeminiImage.save logics</summary>
237-
238-
```python
258+
`GeminiImage.save` method logic
259+
```
239260
import asyncio
240261
from gemini import Gemini, GeminiImage
241262
242-
async def fetch_and_save_images_async(prompt: str, save_path: str = "cached_folder"):
263+
async def fetch_and_save_images_async(prompt: str, save_path: str="cached"):
243264
response = await GeminiClient.generate_content_async(prompt)
244265
245266
response_images = response.web_images # Check response images [Dict]
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
267+
image_data_dict = await GeminiImage.fetch_images_dict(response_images) # Get bytes images dict asynchronously
268+
await GeminiImage.save_images(image_data_dict, save_path=save_path) # Save to path asynchronously
248269
249270
# Run the async function
250271
if __name__ == "__main__":
251272
user_prompt = input("Enter your prompt: ")
252273
asyncio.run(fetch_and_save_images_async(user_prompt))
253274
```
254275

255-
</details>
256-
257-
258276
<br>
259277

260278
## Further

0 commit comments

Comments
 (0)