gogs
/
rust-whisper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

hotwords via $HOTWORDS actually less broken

master
bel 2023-03-25 21:38:53 -06:00
parent 6282972374
commit 8d608c7470
1 changed files with 14 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
whisper-2023/hotwords.py
View File

@ -104,13 +104,15 @@ class Reactor(threading.Thread):
threading.Thread.__init__(self) threading.Thread.__init__(self)
self.inq = inq self.inq = inq
self.outq = outq self.outq = outq
self.load_hotwords = Reactor.new_load_hotwords self.load_hotwords = Reactor.new_load_hotwords()
self.load_hotwords() [i for i in self.load_hotwords()]
def new_load_hotwords(): def new_load_hotwords():
p = environ.get("REACTOR_HOTWORDS", None) p = environ.get("HOTWORDS", None)
if not p: if not p:
return lambda: [] def load_nothing():
return []
return load_nothing
try: try:
def load_hotwords_in_file(): def load_hotwords_in_file():
@ -136,7 +138,14 @@ class Reactor(threading.Thread):
def handle(self, text): def handle(self, text):
hotwords = self.load_hotwords() hotwords = self.load_hotwords()
print(text) if not hotwords:
self.outq.put(text)
return
cleantext = "".join([i for i in "".join(text.lower().split()) if i.isalpha()])
for i in hotwords:
if i in cleantext:
log("hotword", i, "in", text, "as", cleantext)
self.outq.put(i)
class Actor(threading.Thread): class Actor(threading.Thread):
def __init__(self, inq): def __init__(self, inq):