From 346a6b0298d4dae36aa7ba1986f324a86459a6a0 Mon Sep 17 00:00:00 2001 From: phixxy Date: Tue, 6 Feb 2024 01:52:03 -0800 Subject: [PATCH] fixed bug where newlines were not being handled correctly in logging --- extensions/chatgpt.py | 3 ++- extensions/stable_diffusion.py | 4 ++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/extensions/chatgpt.py b/extensions/chatgpt.py index 7b9c557..32268e5 100644 --- a/extensions/chatgpt.py +++ b/extensions/chatgpt.py @@ -238,7 +238,8 @@ class ChatGPT(commands.Cog): await self.download_image(image_url, image_filepath) with open(image_filepath, "rb") as fh: f = discord.File(fh, filename=image_filepath) - log_data = f'Author: {ctx.author.name}, Prompt: {prompt.replace('\n',' ')}, Filename: {my_filename}\n' + prompt = prompt.replace('\n',' ') + log_data = f'Author: {ctx.author.name}, Prompt: {prompt}, Filename: {my_filename}\n' with open(f"{self.data_dir}logs/dalle3.log", 'a') as log_filepath: log_filepath.writelines(log_data) await ctx.send(f'Generated by: {ctx.author.name}\nPrompt: {prompt}', file=f) diff --git a/extensions/stable_diffusion.py b/extensions/stable_diffusion.py index 69d94e4..79d4f63 100644 --- a/extensions/stable_diffusion.py +++ b/extensions/stable_diffusion.py @@ -254,8 +254,8 @@ class StableDiffusion(commands.Cog): with open(filepath, "rb") as fh: f = discord.File(fh, filename=filepath) - - log_data = f'Author: {ctx.author.name}, Prompt: {prompt.replace('\n',' ')}, Filename: {my_filename}\n' + prompt = prompt.replace('\n',' ') + log_data = f'Author: {ctx.author.name}, Prompt: {prompt}, Filename: {my_filename}\n' with open(f"{self.data_dir}stable_diffusion.log", 'a') as log_file: log_file.writelines(log_data)