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

fixed short output

Browse source
This commit is contained in:
lickthecheese 2020-04-23 13:50:05 -04:00
parent 3b8044dbac
commit a16fade5af
2 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

BIN
modules/__pycache__/nlp.cpython-36.pyc
View file

Binary file not shown.

7
modules/nlp.py
View file

@ -31,13 +31,14 @@ async def genOut(self, noun):
prew = self.db['prew']
beg = [ i['word'] for i in self.db['beg'].find() ]
end = [ i['word'] for i in self.db['end'].find() ]
nouns = [i['word'] for i in self.db['noun'].find()]
iter=0
out = [noun]
while out[0] not in beg and iter < 7:
while (out[0] not in beg or nouns.count(out[0])-1 > iter * self.enmul) and iter < 7:
out = [ random.choice(list(prew.find(pro=out[0])))['pre'] ] + out
iter += 1
iter = 0
while out[-1] not in end and iter < 7:
while (out[-1] not in end or nouns.count(out[-1])-1 > iter * self.enmul) and iter < 7:
out.append(random.choice(list(prew.find(pre=out[-1])))['pro'])
iter += 1
return out
@ -55,5 +56,7 @@ async def filter(self, c, n, m):
async def init(self):
self.db = dataset.connect('sqlite:///database.db')
self.enmul = 1
self.raw['nlp'] = filter