Hexor/conf_bot
1
0
Fork 0
mirror of https://github.com/house-of-vanity/conf_bot.git synced 2025-07-06 14:24:08 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Improve markov chain text generator.

Browse Source
This commit is contained in:
AB
2020-02-28 10:00:56 +00:00
parent 40364735e0
commit 9b853f6809
2 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
worker.py
View File

@ -188,7 +188,6 @@ class MessageWorker:
else:
chat_title = msg['message']['chat']['title']
self.db.add_conf(conf_id, chat_title)
rand_messages = list()
max_sen = 100
try:
count = int(msg['message']['text'][8:])
@ -201,9 +200,10 @@ class MessageWorker:
except:
use_all = False
if use_all:
rand_messages.append(self.db.get_random_message(count=count))
rand_messages = self.db.get_random_message(count=count)
else:
rand_messages.append(self.db.get_random_message(conf_id, count=count))
rand_messages = self.db.get_random_message(conf_id, count=count)
print(count, rand_messages)
rand_text = " ".join(rand_messages)
gen_text = get(rand_text)
try: