""" title: Image Gen author: open-webui version: 0.1 """ import os import requests from datetime import datetime from typing import Callable from open_webui.apps.images.main import image_generations, GenerateImageForm from open_webui.apps.webui.models.users import Users class Tools: def __init__(self): pass async def generate_image( self, prompt: str, __user__: dict, __event_emitter__=None ) -> str: """ Generate an image given a prompt :param prompt: prompt to use for image generation """ await __event_emitter__( { "type": "status", "data": {"description": "Generating an image", "done": False}, } ) try: images = await image_generations( GenerateImageForm(**{"prompt": prompt}), Users.get_user_by_id(__user__["id"]), ) await __event_emitter__( { "type": "status", "data": {"description": "Generated an image", "done": True}, } ) for image in images: await __event_emitter__( { "type": "message", "data": {"content": f""}, } ) return f"Notify the user that the image has been successfully generated" except Exception as e: await __event_emitter__( { "type": "status", "data": {"description": f"An error occured: {e}", "done": True}, } ) return f"Tell the user: {e}"