Matrix bot that generates messages based off of messages of other users using a neural network.
Go to file
array-in-a-matrix 32a6ba2c35 updated prompt/ added twim 2022-09-06 16:06:06 -04:00
.gitignore ignore AI related files 2022-08-15 23:38:00 -04:00
LICENSE Initial commit 2022-08-15 00:50:58 -04:00
README.md updated prompt/ added twim 2022-09-06 16:06:06 -04:00
example.config.json removed size variable 2022-08-23 13:39:54 -04:00
index.js updated prompt/ added twim 2022-09-06 16:06:06 -04:00
package.json now using pythonshell lib 2022-08-22 01:55:10 -04:00
pnpm-lock.yaml now using pythonshell lib 2022-08-22 01:55:10 -04:00
textgen.py updated prompt/ added twim 2022-09-06 16:06:06 -04:00

README.md

text-gen-bot

Matrix bot that generates messages based off of messages of other users using a neural network. The first Matrix AI? Featured on TWIM!

Note: Project is still being developed and some functionality is not fully implemented yet.

Table of content

Usage

First install the needed using the instructions here. Then copy example.config.json and rename it config.json. Replace the items in angled brackets with their respective values of the bot account (e.g. replace <DOMAIN.TLD> with the homeserver url like https://matrix.org or https://matrix.arrayinamatrix.xyz). An explanation of each configurable string is located in configurations section of this document. To obtain the token of an account follow the instructions here.

Once the config file has been populated with valid data, execute the index.js file (Warning: executing for the first time will be slow.).

$ node index.js
...
<some warnings show up, ignore them>
...
Client has started!
...

List of directories and files created by the project and its dependencies:

  • aitextgen/
  • node_modules/
  • trained_model/
  • aitextgen.tokenizer.json
  • storage.json
  • training-matrix.txt

Commands

The bot has several commands:

► generate   ⇢     Generates a message in the room. 
► train      ⇢     Trains the AI's mini GPT-2 model.

Setup

The project is split into 2 parts index.js and textgen.py. The index.js file contains the code that interacts with the user on Matrix and sends text generated by the textgen.py file.

Install matrix-bot-sdk and python-shell (JS):

> pnpm add matrix-bot-sdk
> pnpm add python-shell

Install aitextgen (PY):

> pip3 install aitextgen

Configurations

Before a bot can be used the fields in the config.json file must be populated with valid information. Values in angled brackets (stared below) must be supplied before usage.

► homeserver*   ⇢     URL of the bot's homeserver.

► token*        ⇢     Account token used to sign in the bot.

► user*         ⇢     Account's User ID.

► file          ⇢     Path of file used for training the AI (.txt file only).

► prefix        ⇢     Bot listens to commands that start with this prefix.

► frequency     ⇢     How often the bot sends a message (keep high to prevent spam).

► retrain       ⇢      The bot retrains itself after this many extra lines of messages are recorded in the text file.

Using existing messages

If you do not want to wait until the bot creates its own dataset from new messages, you can export chat history of a room easily using Element. In this case, you will need to manually remove the time stamps from the text file then place the path of the file in the file variable in the configuration file.