Add user-specified GPT role

Signed-off-by: bghira <bghira@users.github.com>

discord_tron_master/classes/app_config.py

@@ -32,6 +32,7 @@
 
 DEFAULT_USER_CONFIG = {
     "steps": 100,
+    "gpt_role": "You are a Discord bot.",
     "temperature": 0.9,
     "strength": 0.5,
     "resize": 1,

discord_tron_master/classes/openai/text.py

@@ -47,8 +47,12 @@ async def random_image_prompt(self, theme: str = None):
             prompt = prompt + '. Theme: ' + theme
         return await self.turbo_completion("You are a Stable Diffusion Prompt Generator Bot. Respond as one would.", prompt, temperature=1.1)
 
-    async def discord_bot_response(self, prompt):
-        return await self.turbo_completion(self.discord_bot_role, prompt)
+    async def discord_bot_response(self, prompt, ctx = None):
+        user_role = self.discord_bot_role
+        if ctx is not None:
+            user_role = self.config.get_user_setting(ctx.author.id, "gpt_role", self.discord_bot_role)
+            user_temperature = self.config.get_user_setting(ctx.author.id, "temperature")
+        return await self.turbo_completion(user_role, prompt, temperature=user_temperature, max_tokens=2048)
 
     async def turbo_completion(self, role, prompt, **kwargs):
         if kwargs:

discord_tron_master/cogs/image/img2img.py

@@ -58,5 +58,5 @@ async def on_message(self, message):
                 # We were mentioned, but no attachments. They must want to converse.
                 logging.debug("Message contains no attachments. Initiating conversation.")
                 gpt = GPT()
-                response = await gpt.turbo_completion(role=gpt.discord_bot_role, prompt=message.content, max_tokens=2048, temperature=0.9)
+                response = await gpt.discord_bot_response(prompt=message.content, ctx=message)
                 await discord.send_large_message(message, message.author.mention + ' ' + response)

discord_tron_master/cogs/user/settings.py

@@ -69,6 +69,7 @@ async def my_settings(self, ctx, *args):
         guidance_scaling = self.config.get_user_setting(user_id, "guidance_scaling")
         enable_sag = self.config.get_user_setting(user_id, "enable_sag")
         seed = self.config.get_user_setting(user_id, "seed", None)
+        gpt_role = self.config.get_user_setting(user_id, "gpt_role")
         negative_prompt = self.config.get_user_setting(
             user_id,
             "negative_prompt",
@@ -94,6 +95,7 @@ async def my_settings(self, ctx, *args):
             f"🟠 **Self-Assisted Guidance (SAG)**: `{enable_sag}` **Default**: `False`\n❓ Use SAG scaling to make higher quality images. Requires a square aspect ratio on non-SAG models.\n"
             f"🟠 **Negative Prompt:**:\n➡️    `{negative_prompt}`\n❓ Images featuring these keywords are less likely to be generated. Set via `{self.config.get_command_prefix()}negative`.\n"
             f"🟠 **Positive Prompt:**:\n➡️    `{positive_prompt}`\n❓ Added to the end of every prompt, which has a limit of 77 tokens. This can become truncated. Set via `{self.config.get_command_prefix()}positive`.\n"
+            f"🟠 **GPT Role:**:\n➡️    `{gpt_role}`\n❓ Defines how this bot will respond to you when chatting. Use `{self.config.get_command_prefix()}settings gpt_role [new role]`.\n"
             f"🟠 **Resolution:** `{resolution['width']}x{resolution['height']}`\n❓ Lower resolutions render more quickly, and has a relationship with `steps` that can really influence the output. See **{self.config.get_command_prefix()}help resolution** for more information."
         )