hueso/Alpaca
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed added check so it doesn't fail if there isn't data in a message when converting

Browse Source
This commit is contained in:
jeffser 2024-06-04 20:00:45 -06:00
parent 887a7645d4
commit 60dd1de39d
1 changed files with 14 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
src/update_history.py
View File

@ -15,19 +15,20 @@ def update(self):
if not os.path.exists(directory): os.makedirs(directory) if not os.path.exists(directory): os.makedirs(directory)
new_messages = {} new_messages = {}
for message in content['messages']: for message in content['messages']:
message_id = self.generate_uuid() if message:
if 'images' in message: message_id = self.generate_uuid()
if not os.path.exists(os.path.join(directory, message_id)): os.makedirs(os.path.join(directory, message_id)) if 'images' in message:
new_images = [] if not os.path.exists(os.path.join(directory, message_id)): os.makedirs(os.path.join(directory, message_id))
for image in message['images']: new_images = []
file_name = f"{self.generate_uuid()}.png" for image in message['images']:
decoded = base64.b64decode(image) file_name = f"{self.generate_uuid()}.png"
buffer = io.BytesIO(decoded) decoded = base64.b64decode(image)
im = Image.open(buffer) buffer = io.BytesIO(decoded)
im.save(os.path.join(directory, message_id, file_name)) im = Image.open(buffer)
new_images.append(file_name) im.save(os.path.join(directory, message_id, file_name))
message['images'] = new_images new_images.append(file_name)
new_messages[message_id] = message message['images'] = new_images
new_messages[message_id] = message
new_data['chats'][chat_name] = {} new_data['chats'][chat_name] = {}
new_data['chats'][chat_name]['messages'] = new_messages new_data['chats'][chat_name]['messages'] = new_messages