basic gpt

basic-gpt.py

import random

model_training = [
 
]

tokens = {}

for i in range(0, len(model_training)):
    phrase_tokens = model_training[i].split()
    for x in range(0, len(phrase_tokens)):
        tokens[phrase_tokens[x].lower()] = []
       
for i in range(0, len(model_training)):
    phrase_tokens = model_training[i].split()
    for x in range(0, len(phrase_tokens)):
        if (x + 1 < len(phrase_tokens)):
            tokens[phrase_tokens[x].lower()].append(phrase_tokens[x + 1].lower())


print(tokens)

def findNextToken(token):
    return tokens[token][random.randint(0, len(tokens[token]) - 1)]

phrase = input('> ')

phrase_tokens = phrase.split()

result = []
for i in range(0, len(phrase_tokens)):
    result.append(findNextToken(phrase_tokens[i].lower()))

result_tokens = []
for x in range(0, len(phrase.split())):
    result_tokens = [w for w in result if w!= phrase.split()[x]]

print(result_tokens[0])