"""
title: Image Gen
author: open-webui
author_url: https://github.com/open-webui
funding_url: https://github.com/open-webui
version: 0.1
required_open_webui_version: 0.5.3
"""
import os
import requests
from datetime import datetime
from typing import Callable
from fastapi import Request
from open_webui.routers.images import image_generations, GenerateImageForm
from open_webui.models.users import Users
class Tools:
def __init__(self):
pass
async def generate_image(
self, prompt: str, __request__: Request, __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(
request=__request__,
form_data=GenerateImageForm(**{"prompt": prompt}),
user=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}"
