generator.py

import sys
import requests
import time
import nltk

from markovchain import Markovchain, load_data
from secret_settings import * 

Discord_Token = discord_token_secret
ChannelID = discord_channel_id 

def main(): 

    if len(sys.argv) != 3:
        sys.exit("usage: python generator.py n corpus_data")
    # n amount of n-grams
    n = int(sys.argv[1])
    corpus = load_data(sys.argv[2], n)

    ngrams = list(nltk.ngrams(corpus, n))

    model = Markovchain(ngrams, n)
    model.update()
    sentence = model.generate_text()
    print(sentence)

    # send text message to your friends on discord using markovchain
    sendMessage(Discord_Token, ChannelID, sentence)

def sendMessage(token, channel_id, message):
    url = 'https://discord.com/api/v8/channels/{}/messages'.format(channel_id)
    data = {"content": message}
    header = {"authorization": token}
 
    r = requests.post(url, data=data, headers=header)



if __name__ == "__main__":
    main()