From 3c4d658099987c7441bb4152f47a753570462bd1 Mon Sep 17 00:00:00 2001 From: "l.gabrysiak" Date: Sun, 2 Mar 2025 17:48:04 +0100 Subject: [PATCH] mod user letter --- backend/open_webui/models/users.py | 6 +----- backend/open_webui/utils/oauth.py | 15 +++++++-------- 2 files changed, 8 insertions(+), 13 deletions(-) diff --git a/backend/open_webui/models/users.py b/backend/open_webui/models/users.py index 3e352b9..ce7d1fb 100644 --- a/backend/open_webui/models/users.py +++ b/backend/open_webui/models/users.py @@ -397,11 +397,7 @@ class UsersTable: text_width = bbox[2] - bbox[0] text_hight = bbox[3] - bbox[1] x = (size[0] - text_width) // 2 - y = (size[1] - text_hight - 133) // 2 # Poprawione obliczenie pozycji y - print(str(text_hight)) - print(str(size[1] - bbox[1])) - print(str(size[1] - text_hight - 133)) - print(str(y)) + y = (size[1] - text_hight - 135) // 2 # Poprawione obliczenie pozycji y draw.text((x, y), letter, font=font, fill=text_color) buffered = io.BytesIO() diff --git a/backend/open_webui/utils/oauth.py b/backend/open_webui/utils/oauth.py index 5835f5f..eceb09d 100644 --- a/backend/open_webui/utils/oauth.py +++ b/backend/open_webui/utils/oauth.py @@ -308,24 +308,20 @@ class OAuthManager: Users.update_user_oauth_sub_by_id(user.id, provider_sub) if user: - letter = "A" + name = user_data.get("name", "") + letter = name[0] base64_image = Users.generate_image_base64(letter) - print(base64_image) Auths.update_user_profile( user.id, user_data.get("name", ""), - user_data.get("profile_image_url", ""), + user_data.get("profile_image_url", base64_image), user_data.get("role", self.get_user_role(None, user_data)), user_data.get("permissions", ""), user_data.get("subscription", "") ) if not user: - letter = "A" - base64_image = Users.generate_image_base64(letter) - print(base64_image) - user_count = Users.get_num_users() # If the user does not exist, check if signups are enabled @@ -378,6 +374,9 @@ class OAuthManager: if not name: log.warning("Username claim is missing, using email as name") name = email + + letter = name[0] + base64_image = Users.generate_image_base64(letter) #role = self.get_user_role(None, user_data) role = user_data.get("role", self.get_user_role(None, user_data)) @@ -388,7 +387,7 @@ class OAuthManager: str(uuid.uuid4()) ), # Random password, not used name=name, - profile_image_url=picture_url, + profile_image_url=user_data.get("profile_image_url", base64_image), role=role, oauth_sub=provider_sub, permissions=user_data.get("permissions", ""),