Skip to content

Latest commit

 

History

History
75 lines (49 loc) · 2.35 KB

README.md

File metadata and controls

75 lines (49 loc) · 2.35 KB
Raw

PyWomboAPI

PyWomboAPI

PyWomboAPI is an unofficial Python library designed to generate images from Wombo's dream.ai. This library aims to make generating images easy and fast.

Please remember that this is a personal project I started to learn new things, so expect bugs!

Features

  • No API keys required: The API can automatically generate authentication tokens.
  • Simple and Intuitive: Easy methods for generating images.
  • Regular Updates: I try to maintain and keep the code working efficiently.

Installation

You can easily install PyWomboAPI via pip:

pip install PyWomboAPI

Usage

Here is a quick example of how to use PyWomboAPI to make a request to the Wombo AI API:

import asyncio
import logging
from PyWomboAPI import WomboAPI

async def main() -> None:
    api = WomboAPI(log_level=logging.INFO)  # Set log level to INFO to see what's going on
    api.setup()  # Requesting all the necessary data
    image = await api.create_image(140, "flying car")  # Starting image generation
    print(image.url)

if __name__ == '__main__':
    asyncio.run(main())

This will generate an image:

2024-... - PyWomboAPI.init - INFO - API key not set, fetching it automatically...
2024-... - PyWomboAPI.init - INFO - Auth token not provided, generating one...
2024-... - PyWomboAPI.init - INFO - Credentials: Credentials(api_key='AIzaSyDCvp5MTJLUdtBYEKYWXJrlLzu1zuKM6Xw', auth_token='...')
2024-... - PyWomboAPI.generation - INFO - Constructed POST data for start_task request: {'is_premium': False, 'input_spec': {'aspect_ratio': 'old_vertical_ratio', 'prompt': 'flying car', 'style': 140, 'display_freq': 10}}
2024-... - PyWomboAPI.generation - INFO - Received task data: {'id': '87e7cd9d-af3a-4e71-93ce-58b31b072fb9', ...}
2024-... - PyWomboAPI.generation - INFO - Task 87e7cd9d-af3a-4e71-93ce-58b31b072fb9 completed successfully: https://images.wombo.art/generated/.../final.jpg?Expires=...&Signature=...&Key-Pair-Id=...

More examples can be found here.

To-Do List

  • Add Support for Proxies
  • Better Error Handling
  • Improve Documentation
  • Implement Tests

License

This project is licensed under the MIT License - see the LICENSE file for details.