xfnw/min
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Merge pull request #2 from cheapie/main

Browse source 
Improve message generation speed
This commit is contained in:
owen 2021-09-26 10:04:15 -04:00 committed by GitHub
commit ee1240b818
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
modules/nlp.py
View file

@ -28,10 +28,10 @@ async def getNoun(self, words, c):
shared.db['remsg'].insert_ignore(dict(noun=oldnoun,msg=' '.join(words)),['id'])
nouns = [i['word'] for i in shared.db['noun'].find()]
nouns = shared.db['noun']
out = {}
for i in words:
out[i] = nouns.count(i)
out[i] = nouns.count(word=i)
noun = min(out, key=out.get)
conversation = shared.db['conver']
@ -51,13 +51,13 @@ async def genOut(self, noun):
if len(oldresponses) > 0:
return random.choice(oldresponses).split(' ')
prew = shared.db['prew']
beg = [ i['word'] for i in shared.db['beg'].find() ]
end = [ i['word'] for i in shared.db['end'].find() ]
nouns = [i['word'] for i in shared.db['noun'].find()]
beg = shared.db['beg']
end = shared.db['end']
nouns = shared.db['noun']
iter=0
coun=0
out = [noun]
while (out[0] not in beg or nouns.count(out[0])-1 > iter * shared.enmul) and iter < 7:
while (beg.find_one(word=out[0]) is None or nouns.count(word=out[0])-1 > iter * shared.enmul) and iter < 7:
try:
out = [ random.choice(list(prew.find(pro=out[0])))['pre'] ] + out
except IndexError:
@ -65,7 +65,7 @@ async def genOut(self, noun):
iter += 1
coun += 1
iter = 0
while (out[-1] not in end or nouns.count(out[-1])-1 > iter * shared.enmul) and iter < 7:
while (end.find_one(word=out[-1]) is None or nouns.count(word=out[-1])-1 > iter * shared.enmul) and iter < 7:
try:
out.append(random.choice(list(prew.find(pre=out[-1])))['pro'])