Compare commits
10 Commits
0aff4f556b
...
247edd2ced
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
247edd2ced | ||
|
|
edd94aef72 | ||
|
|
b4d3e5a27c | ||
|
|
a1436e3bd2 | ||
|
|
410769b8c6 | ||
|
|
5869016de6 | ||
|
|
0955f6c0c0 | ||
|
|
242f4407df | ||
|
|
814a8ae2f3 | ||
|
|
7c369e72d4 |
@@ -47,7 +47,7 @@ class Reader(threading.Thread):
|
||||
self.name = os.environ.get("MIC_NAME", "pulse_monitor")
|
||||
if not self.name:
|
||||
for index, name in enumerate(sr.Microphone.list_microphone_names()):
|
||||
print("[{0}] Microphone with name \"{1}\" found for `Microphone(device_index={0})`".format(index, name))
|
||||
log("[{0}] Microphone with name \"{1}\" found for `Microphone(device_index={0})`".format(index, name))
|
||||
exit()
|
||||
self.inq = inq
|
||||
self.outq = outq
|
||||
@@ -142,9 +142,13 @@ class Parser(threading.Thread):
|
||||
log("stderr:", proc.stderr.decode().strip())
|
||||
log("raw transcript:", result)
|
||||
result = result.replace(">>", "")
|
||||
result = "".join([i.split("]")[-1] for i in result.split("[")[0]])
|
||||
result = "".join([i.split(")")[-1] for i in result.split("(")[0]])
|
||||
result = "".join([i.split(">")[-1] for i in result.split("<")[0]])
|
||||
for pair in [
|
||||
("[", "]"),
|
||||
("(", ")"),
|
||||
("<", ">"),
|
||||
("*", "*"),
|
||||
]:
|
||||
result = "".join([i.split(pair[1])[-1] for i in result.split(pair[0])[0]])
|
||||
if os.environ.get("DEBUG", None):
|
||||
log("annotation-free transcript:", result)
|
||||
return result
|
||||
@@ -181,6 +185,8 @@ def _load_dot_notation(v, items):
|
||||
else:
|
||||
result.append(subresult)
|
||||
return result
|
||||
elif k == "KEYS":
|
||||
v = [k for k in v]
|
||||
else:
|
||||
if isinstance(v, list):
|
||||
v = v[int(k)]
|
||||
@@ -225,8 +231,15 @@ class Reactor(threading.Thread):
|
||||
def load_hotwords_in_yaml_file():
|
||||
with open(p.split("@")[0], "r") as f:
|
||||
v = yaml.safe_load(f)
|
||||
v = load_dot_notation(v, p.split("@")[-1])
|
||||
return ["".join(i.strip().lower().split()) for i in v if i]
|
||||
if os.environ.get("DEBUG", None):
|
||||
log(f'opened {p.split("@")[0]} and got {v}')
|
||||
result = []
|
||||
for to_find in [i for i in p.split("@")[-1].split(",") if i]:
|
||||
if os.environ.get("DEBUG", None):
|
||||
log(f'finding {to_find} in {v}')
|
||||
v2 = load_dot_notation(v, to_find)
|
||||
result.extend(["".join(i.strip().lower().split()) for i in v2 if i])
|
||||
return result
|
||||
load_hotwords_in_yaml_file()
|
||||
return load_hotwords_in_yaml_file
|
||||
else:
|
||||
@@ -255,13 +268,23 @@ class Reactor(threading.Thread):
|
||||
log("Reactor.run: stop")
|
||||
|
||||
def handle(self, text):
|
||||
try:
|
||||
self._handle(text)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
def _handle(self, text):
|
||||
hotwords = self.load_hotwords()
|
||||
if os.environ.get("DEBUG", None):
|
||||
log(f"seeking {hotwords} in {text}")
|
||||
log(f"seeking {hotwords} in {text}. $HOTWORDS={os.environ.get('HOTWORDS', None)}")
|
||||
if not hotwords:
|
||||
if not os.environ.get("HOTWORDS", None):
|
||||
print(text)
|
||||
if os.environ.get("DEBUG", None):
|
||||
log(f"HOTWORDS is False; {text}")
|
||||
print(text, flush=True)
|
||||
else:
|
||||
if os.environ.get("DEBUG", None):
|
||||
log(f"HOTWORDS is True; {text}")
|
||||
log(text)
|
||||
return
|
||||
cleantext = "".join([i for i in "".join(text.lower().split()) if i.isalpha()])
|
||||
@@ -301,7 +324,7 @@ class Actor(threading.Thread):
|
||||
|
||||
def handle_stdout(self, hotword, context):
|
||||
log(context)
|
||||
print(hotword)
|
||||
print(hotword, flush=True)
|
||||
|
||||
def handle_signal(self, hotword, context):
|
||||
self.handle_stderr(hotword, context)
|
||||
|
||||
2
rust-whisper.d/transcript.sh
Normal file
2
rust-whisper.d/transcript.sh
Normal file
@@ -0,0 +1,2 @@
|
||||
echo "pkill -9 -f hotwords.py; MIC_TIMEOUT=30 MODEL=small.en P=4 DEBUG=true HOTWORDS= python3 ./hotwords.py | tee -a $HOME/Sync/drawful/DnD/bdoob/__log.d/$(date +%Y.%m.%d).transcript.txt"
|
||||
echo "pkill -9 -f hotwords.py; MIC_TIMEOUT=30 MODEL=small.en P=4 DEBUG=true HOTWORDS= python3 ./hotwords.py | tee -a $HOME/Sync/drawful/DnD/nessira.d/_log.d/$(date +%Y.%m.%d).transcript.txt"
|
||||
Reference in New Issue
Block a user