api: Adding a profile picture for the bot

okbrandon · Oct 16, 2024 · 3c1c559 · 3c1c559
1 parent e3b4382
commit 3c1c559
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/backend/api/apps.py b/backend/api/apps.py
@@ -1,3 +1,5 @@
+import base64
+
 from django.apps import AppConfig
 from django.db.models.signals import post_migrate
 
@@ -32,13 +34,18 @@ def create_ai_account(sender, **kwargs):
     from .models import User
 
     if not User.objects.filter(userID="user_ai").exists():
+        encoded_avatar = None
+        with open("static/ai_avatar.jpg", "rb") as f:
+            encoded_avatar = base64.b64encode(f.read()).decode('utf-8')
+
         User.objects.create(
             userID="user_ai",
             username='ai',
             displayName='Prune',
             email='prune@brandoncodes.dev',
             password='',
             lang='EN',
+            avatarID=f"data:image/jpeg;base64,{encoded_avatar}",
             flags=3
         )
 

diff --git a/backend/api/consumers.py b/backend/api/consumers.py
@@ -396,7 +396,10 @@ async def handle_identify(self, data):
                 await self.close()
         else:
             self.user = await self.get_user_from_id('user_ai')
-            await self.send_json({"e": "READY", "d": get_safe_profile(UserSerializer(self.user).data, me=True)})
+            user_data = UserSerializer(self.user).data
+            safe_profile = get_safe_profile(user_data, me=True)
+            safe_profile.pop('avatarID', None)
+            await self.send_json({"e": "READY", "d": safe_profile})
             logger.info(f"[{self.__class__.__name__}] AI identified: {self.user.userID}")
 
     async def get_user_profile(self, token):

diff --git a/backend/static/ai_avatar.jpg b/backend/static/ai_avatar.jpg