Add chatbot model configuration #58

Author: mak448a

README.md

@@ -138,7 +138,18 @@ Go to `config.json` and set the key `"sync"` to `false`.
 It should look like this:
 
 ```json
-"sync": false
+"sync": false,
+```
+</details>
+
+<details><summary>Change chatbot drawing model</summary>
+
+Go to `config.json` and set the key `"image_model"` to a model that you find [here](https://aqualxx.github.io/stable-ui/).
+
+It should look like this:
+
+```json
+"image_model": "stable_diffusion_2.1"
 ```
 </details>
 

example_config.json

@@ -2,5 +2,6 @@
   "chatbot": true,
   "model": "gpt-3.5-turbo",
   "loading_gif": "https://tenor.com/view/loading-gif-9212724",
-  "sync": true
+  "sync": true,
+  "image_model": "stable_diffusion_2.1"
 }

main.py

@@ -88,7 +88,7 @@ async def on_message(message):
         await imagine_horde(
             FakeCtx(message),  # NOQA
             prompt=prompt,
-            model=app_commands.Choice(name="Stable Diffusion", value="stable_diffusion_2.1")
+            model=app_commands.Choice(name=config["image_model"], value=config["image_model"])
         )
 
 

run.py

@@ -39,7 +39,8 @@ def configure():
         "chatbot": True,
         "model": "gpt-3.5-turbo",
         "loading_gif": "https://tenor.com/view/loading-gif-9212724",
-        "sync": True
+        "sync": True,
+        "image_model": "stable_diffusion_2.1"
     }
     # Set a variable to store new config
     config = default_config  # add .copy() if you want it to be separate