diff --git a/backend/open_webui/models/users.py b/backend/open_webui/models/users.py index 61262d7..7934b20 100644 --- a/backend/open_webui/models/users.py +++ b/backend/open_webui/models/users.py @@ -346,10 +346,13 @@ class UsersTable: users = db.query(User).filter(User.id.in_(user_ids)).all() return [user.id for user in users] + @staticmethod def generate_image_base64(letter, size=200): # Upewniamy się, że size jest krotką if isinstance(size, int): size = (size, size) + elif not isinstance(size, (list, tuple)) or len(size) != 2: + raise ValueError("Size must be an integer or a tuple of two integers") # Tworzenie obrazu z losowym kolorem tła background_color = (random.randint(0, 255), random.randint(0, 255), random.randint(0, 255)) @@ -363,7 +366,6 @@ class UsersTable: try: font = ImageFont.truetype("arial.ttf", font_size) except IOError: - # Jeśli nie ma czcionki Arial, użyj domyślnej font = ImageFont.load_default() # Obliczanie pozycji tekstu