diff --git a/hugvey/story.py b/hugvey/story.py index 0c269ab..ae484a5 100644 --- a/hugvey/story.py +++ b/hugvey/story.py @@ -13,6 +13,7 @@ import faulthandler from zmq.asyncio import Context import zmq import wave +import sox from pythonosc import udp_client import random @@ -26,18 +27,18 @@ class Utterance(object): self.endTime = None self.text = "" self.lastUpdateTime = startTime - + def setText(self, text, now): self.text = text self.lastUpdateTime = now - + def setFinished(self, endTime): self.endTime = endTime - + def isFinished(self): return self.endTime is not None - + class Message(object): def __init__(self, id, text): self.id = id @@ -58,7 +59,7 @@ class Message(object): self.parseForVariables() self.uuid = None # Have a unique id each time the message is played back. self.color = None - + def setStory(self, story): self.story = story self.logger = story.logger.getChild("message") @@ -80,7 +81,7 @@ class Message(object): # prevent clipping on some Lyrebird tracks msg.params['vol'] = .8 return msg - + def parseForVariables(self): """ Find variables in text @@ -88,20 +89,20 @@ class Message(object): self.variables = re.findall('\$(\w+)', self.text) for var in self.variables: self.variableValues[var] = None - + def hasVariables(self) -> bool: return len(self.variables) > 0 - + def setVariable(self, name, value): if name not in self.variables: self.logger.critical("Set nonexisting variable") return - + if self.variableValues[name] == value: return - + self.variableValues[name] = value - + self.logger.warn(f"Set variable, fetch {name}") if not None in self.variableValues.values(): self.logger.warn(f"now fetch {name}") @@ -116,10 +117,10 @@ class Message(object): # self.logger.debug(f"Getting text for {self.id}") for var in self.variables: self.logger.debug(f"try replacing ${var} with {self.variableValues[var]} in {text}") - replacement = self.variableValues[var] if (self.variableValues[var] is not None) else "nothing" #TODO: translate nothing to each language + replacement = self.variableValues[var] if (self.variableValues[var] is not None) else "nothing" #TODO: translate nothing to each language text = text.replace('$'+var, replacement) return text - + def setReply(self, reply): self.reply = reply @@ -132,16 +133,16 @@ class Message(object): "Getting reply while there is none! {0}".format(self.id)) return self.reply - + def isFinished(self): return self.finishTime is not None - + def setFinished(self, currentTime): self.finishTime = currentTime - + def getFinishedTime(self): return self.finishTime - + def getParams(self): return self.params @@ -152,18 +153,18 @@ class Message(object): 'text': self.getText(), 'replyText': None if self.reply is None else [u.text for u in self.reply.utterances] } - + async def getAudioFilePath(self): if self.audioFile is not None: return self.audioFile - + text = self.getText() self.logger.debug(f"Fetching audio for {text}") - + # return "test"; async with self.filenameFetchLock: # print(threading.enumerate()) - + info = { 'text': text, 'variable': True if self.hasVariables() else False @@ -174,10 +175,10 @@ class Message(object): await s.send_json(info) filename = await s.recv_string() s.close() - - + + # print(threading.enumerate()) - + self.logger.debug(f"Fetched audio for {text}: {filename}") return filename @@ -187,20 +188,20 @@ class Reply(object): self.forMessage = None self.utterances = [] self.setForMessage(message) - + def setForMessage(self, message: Message): self.forMessage = message message.setReply(self) - + def getLastUtterance(self) -> Utterance: if not self.hasUtterances(): return None u = self.utterances[-1] #: :type u: Utterance - + # attempt to fix a glitch that google does not always send is_finished if u.isFinished(): return u - + now = self.forMessage.story.timer.getElapsed() diff = now - u.lastUpdateTime if diff > 5: # time in seconds to force silence in utterance @@ -209,23 +210,23 @@ class Reply(object): f"Set finish time for utterance after {diff}s {u.text}" ) u.setFinished(now) - + return u - + def getFirstUtterance(self) -> Utterance: if not self.hasUtterances(): return None return self.utterances[0] - + def hasUtterances(self) -> bool: return len(self.utterances) > 0 def addUtterance(self, utterance: Utterance): self.utterances.append(utterance) - + def getText(self) -> str: return ". ".join([u.text for u in self.utterances]) - + def getActiveUtterance(self, currentTime) -> Utterance: """ If no utterance is active, create a new one. Otherwise return non-finished utterance for update @@ -242,13 +243,13 @@ class Reply(object): if u is not None and not u.isFinished(): return True return False - + def getTimeSinceLastUtterance(self): if not self.hasUtterances(): return None - + return self.forMessage.story.timer.getElapsed() - self.getLastUtterance().lastUpdateTime - + class Condition(object): """ A condition, basic conditions are built in, custom condition can be given by @@ -269,7 +270,7 @@ class Condition(object): condition = conditionClass(data['@id']) condition.type = data['type'] condition.originalJsonString = json.dumps(data) - + # TODO: should Condition be subclassed? if data['type'] == "replyContains": condition.method = condition._hasMetReplyContains @@ -279,7 +280,7 @@ class Condition(object): condition.method = condition._hasVariable if data['type'] == "diversion": condition.method = condition._hasDiverged - + if 'vars' in data: condition.vars = data['vars'] @@ -296,7 +297,7 @@ class Condition(object): # check if the message already finished playing if not story.lastMsgFinishTime: return False - + if 'onlyIfNoReply' in self.vars and self.vars['onlyIfNoReply']: if story.currentReply and story.currentReply is not None and story.currentReply.hasUtterances(): story.logger.log(LOG_BS, f'Only if no reply has text! {story.currentReply.getText()}') @@ -304,29 +305,29 @@ class Condition(object): return False # else: # story.logger.debug('Only if no reply has no text yet!') - + hasMetTimeout = now - story.lastMsgFinishTime >= float(self.vars['seconds']) if not hasMetTimeout: return False - + # update stats: story.stats['timeouts'] +=1 if 'needsReply' in self.vars and self.vars['needsReply'] is True: if story.currentReply is None or not story.currentReply.hasUtterances(): story.stats['silentTimeouts'] +=1 story.stats['consecutiveSilentTimeouts'] += 1 - + self.logInfo = "{}s".format(self.vars['seconds']) return True - + def _hasVariable(self, story) -> bool: if not story.lastMsgFinishTime: return False - + r = story.hasVariableSet(self.vars['variable']) if r: story.logger.debug(f"Variable {self.vars['variable']} is set.") - + if 'notSet' in self.vars and self.vars['notSet']: # inverse: r = not r @@ -335,28 +336,28 @@ class Condition(object): self.vars['variable'] ) return r - + def _hasDiverged(self, story) -> bool: if not story.lastMsgFinishTime: return False - + d = story.get(self.vars['diversionId']) if not d: story.logger.critical(f"Condition on non-existing diversion: {self.vars['diversionId']}") - + r = d.hasHit if r: story.logger.debug(f"Diversion {self.vars['diversionId']} has been hit.") - + if 'inverseMatch' in self.vars and self.vars['inverseMatch']: # inverse: r = not r - + self.logInfo = "Has {} diverged to {}".format( 'not' if 'inverseMatch' in self.vars and self.vars['inverseMatch'] else '', self.vars['diversionId'] ) - + return r def _hasMetReplyContains(self, story) -> bool: @@ -372,12 +373,12 @@ class Condition(object): return False capturedVariables = None - + if 'regex' in self.vars and len(self.vars['regex']): if 'regexCompiled' not in self.vars: # Compile once, as we probably run it more than once self.vars['regexCompiled'] = re.compile(self.vars['regex']) - + t = r.getText().lower() story.logger.log(LOG_BS, 'attempt regex: {} on {}'.format(self.vars['regex'], t)) result = self.vars['regexCompiled'].search(t) @@ -385,14 +386,14 @@ class Condition(object): #if there is something to match, but not found, it's never ok return False story.logger.debug('Got match on {}'.format(self.vars['regex'])) - + capturedVariables = result.groupdict() - + if ('instantMatch' in self.vars and self.vars['instantMatch']) or not r.isSpeaking(): # try to avoid setting variables for intermediate strings for captureGroup in capturedVariables: story.setVariableValue(captureGroup, capturedVariables[captureGroup]) - + if 'instantMatch' in self.vars and self.vars['instantMatch']: story.logger.info(f"Instant match on {self.vars['regex']}, {self.vars}") self.logInfo = "Instant match of {}, captured {}".format( @@ -401,8 +402,8 @@ class Condition(object): ) return True # TODO: implement 'instant match' -> don't wait for isFinished() - - + + # print(self.vars) # either there's a match, or nothing to match at all if 'delays' in self.vars: @@ -424,7 +425,7 @@ class Condition(object): self.logInfo = "Match of {}, captured {} after, {}".format( self.vars['regex'], capturedVariables, - timeSinceReply + timeSinceReply ) self.usedContainsDuration = float(delay['waitTime']) return True @@ -432,16 +433,16 @@ class Condition(object): # wait for delay to match story.logger.log(LOG_BS, "Wait for it...") return False - + # If there is a delay, it takes precedence of isSpeaking, since google does not always give an is_finished on short utterances (eg. "hello" or "no") if r.isSpeaking(): story.logger.log(LOG_BS, f"is speaking: {r.getLastUtterance().text} - {r.getLastUtterance().startTime}") return False - + # There is a match and no delay say, person finished speaking. Go ahead sir! self.logInfo = "Match" return True - + def getLogSummary(self): return { 'id': self.id @@ -463,7 +464,7 @@ class Direction(object): def addCondition(self, condition: Condition): self.conditions.append(condition) - + def setMetCondition(self, condition: Condition): self.conditionMet = condition @@ -477,11 +478,11 @@ class Direction(object): c = story.get(conditionId) direction.addCondition(c) return direction - + def getLogSummary(self): return { 'id': self.id, - 'condition': self.conditionMet.id if self.conditionMet else None + 'condition': self.conditionMet.id if self.conditionMet else None } @@ -510,7 +511,7 @@ class Diversion(object): self.regex = re.compile(self.params['regex']) else: self.regex = None - + if type == 'timeout': self.method = self._divergeIfTimeout self.finaliseMethod = self._returnAfterTimeout @@ -519,28 +520,28 @@ class Diversion(object): self.regex = re.compile(self.params['regex']) if type == 'interrupt': self.method = self._divergeIfInterrupted - + if not self.method: raise Exception("No valid type given for diversion") @classmethod def initFromJson(diversionClass, data, story): diversion = diversionClass(data['@id'], data['type'], data['params']) - + return diversion - - + + def getLogSummary(self): return { 'id': self.id, } - + async def divergeIfNeeded(self, story, direction = None): """ Validate if condition is met for the current story state Returns True when diverging """ - + # For all diversion except repeat (which simply doesn't have the variable) if 'notAfterMsgId' in self.params and self.params['notAfterMsgId']: msg = story.get(self.params['notAfterMsgId']) @@ -550,7 +551,7 @@ class Diversion(object): # story.logger.warn(f"Block diversion {self.id} because of hit message {self.params['notAfterMsgId']}") self.disabled = True # never run it and allow following timeouts/no_responses to run return False - + r = await self.method(story, direction.msgFrom if direction else None, direction.msgTo if direction else None, @@ -559,16 +560,16 @@ class Diversion(object): if self.type != 'repeat' and self.type !='interrupt': # repeat diversion should be usable infinte times self.hasHit = True - + story.addToLog(self) return r - + def createReturnDirectionsTo(self, story, startMsg, returnMsg, originalDirection = None, inheritTiming = True, timeoutDuration = .5, replyContainsDurations = None): """ The finishes of this diversion's strand should point to the return message with the right timeout/timing. If hit, this direction should also notify this diversion. - + replyContainsDurations: list formatted as in JSON [{ "minReplyDuration": "0", @@ -576,7 +577,7 @@ class Diversion(object): }] """ self.counter +=1 - + finishMessageIds = story.getFinishesForMsg(startMsg) finalTimeoutDuration = timeoutDuration finalContainsDurations = replyContainsDurations @@ -593,7 +594,7 @@ class Diversion(object): finalTimeoutDuration = float(condition.vars['seconds']) if condition.type == 'replyContains': finalContainsDurations = json.loads(condition.originalJsonString)['vars']['delays'] - + i = 0 for msgId in finishMessageIds: # Some very ugly hack to add a direction & condition @@ -601,7 +602,7 @@ class Diversion(object): msg = story.get(msgId) if not msg: continue - + direction = Direction(f"{self.id}-{i}-{self.counter}", msg, returnMsg) data = json.loads(f""" {{ @@ -615,11 +616,11 @@ class Diversion(object): }} """) data['vars']['onlyIfNoReply'] = finalContainsDurations is not None - + # TODO: also at replycontains if it exists, with the same timings condition = Condition.initFromJson(data, story) direction.addCondition(condition) - + if finalContainsDurations is not None: data2 = json.loads(f""" {{ @@ -636,14 +637,14 @@ class Diversion(object): condition2 = Condition.initFromJson(data2, story) direction.addCondition(condition2) story.add(condition2) - + direction.isDiversionReturn = True # will clear the currentDiversion on story story.logger.info(f"Created direction: {direction.id} {condition.id} with timeout {finalTimeoutDuration}s") story.add(condition) story.add(direction) - - - + + + async def finalise(self, story): """" Only used if the Diversion sets the story.currentDiversion @@ -653,11 +654,11 @@ class Diversion(object): story.logger.info(f"No finalisation for diversion {self.id}") story.currentDiversion = None return False - + await self.finaliseMethod(story) story.currentDiversion = None return True - + async def _divergeIfNoResponse(self, story, msgFrom, msgTo, direction): """ Participant doesn't speak for x consecutive replies (has had timeout) @@ -665,33 +666,33 @@ class Diversion(object): ':type story: Story' if story.currentDiversion or not msgFrom or not msgTo: return False - + if story.stats['consecutiveSilentTimeouts'] >= int(self.params['consecutiveSilences']): story.stats['diversions']['no_response'] += 1 msg = story.get(self.params['msgId']) if msg is None: story.logger.critical(f"Not a valid message id for diversion: {self.params['msgId']}") return - + story.logger.info(f"Diverge: No response {self.id} {story.stats}") - + self.returnMessage = msgTo - + if self.params['returnAfterStrand']: self.createReturnDirectionsTo(story, msg, msgTo, direction) - + await story.setCurrentMessage(msg) story.currentDiversion = self return True - + return - + async def _returnAfterNoResponse(self, story): story.logger.info(f"Finalise diversion: {self.id}") story.stats['consecutiveSilentTimeouts'] = 0 # reset counter after diverging # if self.params['returnAfterStrand']: # await story.setCurrentMessage(self.returnMessage) - + async def _divergeIfReplyContains(self, story, msgFrom, msgTo, _): """ Participant doesn't speak for x consecutive replies (has had timeout) @@ -701,83 +702,83 @@ class Diversion(object): if story.currentDiversion: # or not msgFrom or not msgTo: # don't do nested diversions return False - + if self.hasHit: # don't match twice return - + if story.currentReply is None or not self.regex: return - + direction = story.getDefaultDirectionForMsg(story.currentMessage) if not direction: # ignore the direction argument, and only check if the current message has a valid default - return - + return + msgTo = direction.msgTo - + if not direction: return - + waitTime = 1.8 if 'waitTime' not in self.params else float(self.params['waitTime']) timeSince = story.currentReply.getTimeSinceLastUtterance() if timeSince < waitTime: story.logger.log(LOG_BS, f"Waiting for replyContains: {timeSince} (needs {waitTime})") return - + r = self.regex.search(story.currentReply.getText()) if r is None: return - + if 'notForColor' in self.params and self.params['notForColor'] and story.currentMessage.color: if story.currentMessage.color.lower() == self.params['notForColor'].lower(): story.logger.debug(f"Skip diversion {self.id} because of section color") return - + story.logger.info(f"Diverge: reply contains {self.id}") story.stats['diversions']['reply_contains'] += 1 - + msg = story.get(self.params['msgId']) if msg is None: story.logger.critical(f"Not a valid message id for diversion: {self.params['msgId']}") return - + # TODO: pick the direction with timeout as next Message. self.returnMessage = msgTo - + if self.params['returnAfterStrand']: self.createReturnDirectionsTo(story, msg, msgTo, direction) - + await story.setCurrentMessage(msg) story.currentDiversion = self return True - + async def _returnAfterReplyContains(self, story): story.logger.info(f"Finalise diversion: {self.id}") # if self.params['returnAfterStrand']: # await story.setCurrentMessage(self.returnMessage) - + async def _divergeIfRepeatRequest(self, story, msgFrom, msgTo, direction): """ Participant asks if message can be repeated. """ # if not msgFrom or not msgTo: # return - + # TODO: how to handle this now we sometimes use different timings. # Perhaps set isFinished when matching condition. if story.currentReply is None or story.currentReply.getTimeSinceLastUtterance() > 1: return - + r = self.regex.search(story.currentReply.getText()) if r is None: return - + logger.info(f"Diverge: request repeat {self.id}") story.stats['diversions']['repeat'] += 1 await story.setCurrentMessage(story.currentMessage) return True - + async def _divergeIfTimeout(self, story, msgFrom, msgTo, direction): """ (1) last spoken at all @@ -785,15 +786,15 @@ class Diversion(object): """ if story.currentDiversion: return - + if msgFrom or msgTo: # not applicable a direction has been chosen return - + if not story.lastMsgFinishTime: # not during play back return - + # not applicable when timeout is set directions = story.getCurrentDirections() for direction in directions: @@ -805,78 +806,78 @@ class Diversion(object): if now - story.lastMsgFinishTime < float(self.params['minTimeAfterMessage']): # not less than x sec after it return - - + + interval = float(self.params['interval']) if not self.params['fromLastMessage']: # (1) last spoken at all - + timeSince = story.timer.getElapsed('last_speech') if story.timer.hasMark('last_speech') else story.timer.getElapsed('start') if story.timer.hasMark('last_diversion_timeout') and story.timer.getElapsed('last_diversion_timeout') > timeSince: timeSince = story.timer.getElapsed('last_diversion_timeout') if timeSince < interval: return - + story.stats['diversions']['timeout_total'] += 1 else: if story.currentMessage is None: return - + # if story.currentMessage.timeoutDiversionCount + 1 - + if story.currentReply is not None: # still playing back # or somebody has spoken already (timeout only works on silences) return - + if now - story.lastMsgFinishTime < interval: return - + story.currentMessage.timeoutDiversionCount += 1 story.stats['diversions']['timeout_last'] += 1 - + # if we're still here, there's a match! story.logger.info(f"Diverge: Timeout {self.id} of {self.params['interval']}") story.stats['diversions']['timeout'] += 1 - + msg = story.get(self.params['msgId']) if msg is None: story.logger.critical(f"Not a valid message id for diversion: {self.params['msgId']}") return - + # fall back to the currentMessage to return on. # TODO: maybe, if not chapter is found, the diversion should be # blocked alltogether? self.returnMessage = story.getNextChapterForMsg(story.currentMessage, False) or story.currentMessage - - + + if self.params['returnAfterStrand']: # no direction is here, as this diversion triggers before a direction is taken self.createReturnDirectionsTo(story, msg, self.returnMessage, inheritTiming=False) - + await story.setCurrentMessage(msg, allowReplyInterrupt=True) story.currentDiversion = self story.timer.setMark('last_diversion_timeout') return True - + async def _returnAfterTimeout(self, story): story.logger.info(f"Finalise diversion: {self.id}") - + async def _divergeIfInterrupted(self, story, msgFrom, msgTo, direction): """ This is here as a placeholder for the interruption diversion. These will however be triggered differently """ - + if story.currentDiversion or story.allowReplyInterrupt: return False - + msg = story.get(self.params['msgId']) if msg is None: story.logger.critical(f"Not a valid message id for diversion: {self.params['msgId']}") return - + self.returnMessage = story.currentMessage # no direction is here, as this diversion triggers before a direction is taken self.createReturnDirectionsTo(story, msg, self.returnMessage, inheritTiming=False, timeoutDuration=3, replyContainsDurations = [{ @@ -885,7 +886,7 @@ class Diversion(object): }]) await story.setCurrentMessage(msg) story.currentDiversion = self - + return True @@ -909,37 +910,37 @@ class Stopwatch(object): t = time.time() if self.paused_at != 0: pause_duration = t - self.paused_at - else: + else: pause_duration = 0 return t - self.marks[since_mark] - pause_duration - + def pause(self): self.paused_at = time.time() self.isRunning.clear() - + def resume(self): if self.paused_at == 0: return - + pause_duration = time.time() - self.paused_at for m in self.marks: self.marks[m] += pause_duration - + self.paused_at = 0 self.isRunning.set() - + def reset(self): self.marks = {} self.setMark('start') self.paused_at = 0 self.isRunning.set() - + def setMark(self, name): self.marks[name] = time.time() - + def hasMark(self, name): return name in self.marks - + def clearMark(self, name): if name in self.marks: self.marks.pop(name) @@ -968,15 +969,15 @@ class Story(object): self.diversions = [] self.interruptionDiversions = [] self.variables = {} - + def pause(self): self.logger.debug('pause hugvey') self.timer.pause() - + def resume(self): self.logger.debug('resume hugvey') self.timer.resume() - + def getLogSummary(self): summary = { # e[0]: the entity, e[1]: the logged time @@ -986,33 +987,33 @@ class Story(object): } # print(self.log) return summary - + def getLogCounts(self): return { 'messages': len([1 for e in self.log if isinstance(e[0], Message)]), 'diversions': len([1 for e in self.log if isinstance(e[0], Diversion)]), } - + def registerVariable(self, variableName, message): if variableName not in self.variables: self.variables[variableName] = [message] else: self.variables[variableName].append(message) - + def setVariableValue(self, name, value): if name not in self.variables: self.logger.warn(f"Set variable that is not needed in the story: {name}") else: self.logger.debug(f"Set variable {name} to {value}") - + self.variableValues[name] = value - + if name not in self.variables: return - + for message in self.variables[name]: message.setVariable(name, value) - + def hasVariableSet(self, name) -> bool: return name in self.variableValues and self.variableValues is not None @@ -1046,7 +1047,7 @@ class Story(object): # self.logger.debug(self.elements) # self.logger.debug(self.directionsPerMsg) - + self.diversions = [el for el in self.elements.values() if type(el) == Diversion] self.interruptionDiversions = [el for el in self.elements.values() if type(el) == Diversion and el.type == 'interrupt'] @@ -1058,17 +1059,17 @@ class Story(object): else: self.logger.warn( "Could not reinstatiate current message. Starting over") - + # Register variables for msg in self.getMessages(): # print(msg.id, msg.hasVariables()) if not msg.hasVariables(): continue - + for var in msg.variables: self.registerVariable(var, msg) - - + + self.logger.info(f'has variables: {self.variables}') self.logger.info(f'has {len(self.strands)} strands: {self.strands}') self.calculateFinishesForStrands() @@ -1084,13 +1085,13 @@ class Story(object): self.lastSpeechEndTime = None self.variableValues = {} # captured variables from replies self.finish_time = False - + self.events = [] # queue of received events self.commands = [] # queue of commands to send self.log = [] # all nodes/elements that are triggered self.msgLog = [] self.currentReply = None - + self.stats = { 'timeouts': 0, 'silentTimeouts': 0, @@ -1104,7 +1105,7 @@ class Story(object): 'timeout_last': 0 } } - + for msg in self.getMessages(): pass @@ -1124,7 +1125,7 @@ class Story(object): self.startMessage = obj if obj.isStrandStart: self.strands[obj.id] = [] - + if type(obj) == Direction: if obj.msgFrom.id not in self.directionsPerMsg: self.directionsPerMsg[obj.msgFrom.id] = [] @@ -1137,15 +1138,15 @@ class Story(object): if id in self.elements: return self.elements[id] return None - + def getMessages(self): return [el for el in self.elements.values() if type(el) == Message] - + def stop(self): self.logger.info("Stop Story") if self.isRunning: self.isRunning = False - + def shutdown(self): self.stop() self.hugvey = None @@ -1163,25 +1164,25 @@ class Story(object): # that is, until we have a 'reset' or 'start' event. # reinitiate current message await self.setCurrentMessage(self.currentMessage) - + if e['event'] == "playbackStart": if e['msgId'] != self.currentMessage.id: continue self.lastMsgStartTime = self.timer.getElapsed() self.logger.debug("Start playback") - - + + if e['event'] == "playbackFinish": if e['msgId'] == self.currentMessage.id: #TODO: migrate value to Messagage instead of Story self.lastMsgFinishTime = self.timer.getElapsed() self.hugvey.eventLogger.info(f"message: {self.currentMessage.id} {self.currentMessage.uuid} done") - + # 2019-02-22 temporary disable listening while playing audio: # if self.hugvey.google is not None: # self.logger.warn("Temporary 'fix' -> resume recording?") # self.hugvey.google.resume() - + if self.currentMessage.id not in self.directionsPerMsg: # print(self.currentDiversion) # if self.currentDiversion is not None: @@ -1194,7 +1195,7 @@ class Story(object): if e['event'] == 'speech': # participants speaks, reset counter self.stats['consecutiveSilentTimeouts'] = 0 - + # if self.currentMessage and not self.lastMsgStartTime: if self.currentMessage and not self.lastMsgFinishTime: # Ignore incoming speech events until we receive a 'playbackStart' event. @@ -1212,27 +1213,27 @@ class Story(object): # else: self.logger.info("ignore speech during playing message") continue - - + + # log if somebody starts speaking if self.currentReply is None: self.logger.info("Start speaking") self.currentReply= Reply(self.currentMessage) - + now = self.timer.getElapsed() utterance = self.currentReply.getActiveUtterance(now) utterance.setText(e['transcript'], now) self.hugvey.eventLogger.info("speaking: content {} \"{}\"".format(id(utterance), e['transcript'])) self.timer.setMark('last_speech') - + if e['is_final']: utterance.setFinished(self.timer.getElapsed()) self.hugvey.eventLogger.info("speaking: stop {}".format(id(utterance))) - + if self.hugvey.recorder: self.hugvey.recorder.updateTranscription(self.currentReply.getText()) - - + + async def _processDirections(self, directions): ':type directions: list(Direction)' chosenDirection = None @@ -1250,9 +1251,9 @@ class Story(object): self.addToLog(direction) self.currentMessage.setFinished(self.timer.getElapsed()) chosenDirection = direction - + isDiverging = await self._processDiversions(chosenDirection) - + allowReplyInterrupt = False # in some cases, conditions should be allowed to interrupt the reply if metCondition: @@ -1260,15 +1261,15 @@ class Story(object): allowReplyInterrupt = True if metCondition.usedContainsDuration is not None and metCondition.usedContainsDuration < 0.1: allowReplyInterrupt = True - + if not isDiverging and chosenDirection: if chosenDirection.isDiversionReturn and self.currentDiversion: await self.currentDiversion.finalise(self) - + await self.setCurrentMessage(chosenDirection.msgTo, allowReplyInterrupt=allowReplyInterrupt) - + return chosenDirection - + async def _processDiversions(self, direction: None) -> bool: """ Process the diversions on stack. If diverging, return True, else False @@ -1276,7 +1277,7 @@ class Story(object): Else, they are None """ diverge = False - + activeDiversions = [] activeTimeoutDiv = None activeTimeoutLastDiv = None @@ -1286,7 +1287,7 @@ class Story(object): if diversion.disabled or diversion.hasHit or diversion.type == 'interrupt': # interruptions are triggered somewhere else. continue - + if diversion.type == 'timeout': if diversion.params['timesOccured'] > 0: if not diversion.params['fromLastMessage']: @@ -1297,35 +1298,35 @@ class Story(object): if not activeTimeoutLastDiv or activeTimeoutLastDiv.params['timesOccured'] > diversion.params['timesOccured']: activeTimeoutLastDiv = diversion continue - + if diversion.type == 'no_response': if diversion.params['timesOccured'] > 0: if not activeNoResponseDiv or activeNoResponseDiv.params['timesOccured'] > diversion.params['timesOccured']: activeNoResponseDiv = diversion continue - + activeDiversions.append(diversion) - + if activeTimeoutDiv: activeDiversions.append(activeTimeoutDiv) if activeTimeoutLastDiv: activeDiversions.append(activeTimeoutLastDiv) if activeNoResponseDiv: activeDiversions.append(activeNoResponseDiv) - + for diversion in activeDiversions: # TODO: collect diversions and order by times + timesOccured (for timeout & no_response) d = await diversion.divergeIfNeeded(self, direction) if d: diverge = True return diverge - + def addToLog(self, node): self.log.append((node, self.timer.getElapsed())) - + if isinstance(node, Message): self.msgLog.append(node) - + if self.hugvey.recorder: if isinstance(node, Message): self.hugvey.recorder.log('hugvey', node.text, node.id) @@ -1333,7 +1334,7 @@ class Story(object): self.hugvey.recorder.log('diversion',node.id) if isinstance(node, Condition): self.hugvey.recorder.log('condition',node.logInfo, node.id) - + def logHasMsg(self, node): return node in self.msgLog @@ -1347,13 +1348,13 @@ class Story(object): while self.isRunning: if self.isRunning is False: break - + # pause on timer paused await self.timer.isRunning.wait() # wait for un-pause for i in range(len(self.events)): await self._processPendingEvents() - + # Test stability of Central Command with deliberate crash # if self.timer.getElapsed() > 5: # raise Exception('test') @@ -1362,7 +1363,7 @@ class Story(object): directions = self.getCurrentDirections() await self._processDirections(directions) - + # TODO create timer event # self.commands.append({'msg':'TEST!'}) @@ -1385,25 +1386,25 @@ class Story(object): 'action': 'stop', 'id': self.currentMessage.id, }) - - + + message.uuid = shortuuid.uuid() self.currentMessage = message self.lastMsgTime = time.time() self.lastMsgFinishTime = None # to be filled in by the event self.lastMsgStartTime = None # to be filled in by the event self.allowReplyInterrupt = allowReplyInterrupt - + # if not reset: self.previousReply = self.currentReply # we can use this for interrptions self.currentReply = useReply #self.currentMessage.reply - + # send command to already mute mic self.hugvey.sendCommand({ 'action': 'prepare', 'id': message.id }) - + # else: # # if we press 'save & play', it should not remember it's last reply to that msg # self.previousReply = self.currentReply # we can use this for interrptions @@ -1413,16 +1414,21 @@ class Story(object): message.id, message.getText())) self.addToLog(message) self.hugvey.eventLogger.info(f"message: {message.id} {message.uuid} start \"{message.getText()}\"") - + # TODO: prep events & timer etc. fn = await message.getAudioFilePath() - + # get duration of audio file, so the client can detect a hang of 'play' - with wave.open(fn,'r') as fp: - frames = fp.getnframes() - rate = fp.getframerate() - duration = frames/float(rate) - + + #old: with wave + # with wave.open(fn,'r') as fp: + # frames = fp.getnframes() + # rate = fp.getframerate() + # duration = frames/float(rate) + + # get duration of audio file, new: with sox + duration = sox.file_info.duration(fn) + # self.hugvey.google.pause() # pause STT to avoid text events while decision is made self.hugvey.sendCommand({ 'action': 'play', @@ -1431,7 +1437,7 @@ class Story(object): 'params': message.getParams(), 'duration': duration }) - + # 2019-02-22 temporary disable listening while playing audio: # if self.hugvey.google is not None: # self.logger.warn("Temporary 'fix' -> stop recording") @@ -1442,7 +1448,7 @@ class Story(object): for direction in self.getCurrentDirections(): conditions = [c.id for c in direction.conditions] logmsg += "\n- {0} -> {1} (when: {2}) ".format(direction.msgFrom.id, direction.msgTo.id, conditions) - + self.logger.log(LOG_BS,logmsg) def getCurrentDirections(self): @@ -1450,19 +1456,19 @@ class Story(object): return [] else: return self.directionsPerMsg[self.currentMessage.id] - + def getNextChapterForMsg(self, msg, canIncludeSelf = True, depth = 0): if canIncludeSelf and msg.chapterStart: self.logger.info(f"Next chapter: {msg.id}") return msg - + if depth >= 70: # protection against infinite loop? return None - + if msg.id not in self.directionsPerMsg: return None - + for direction in self.directionsPerMsg[msg.id]: r = self.getNextChapterForMsg(direction.msgTo, True, depth+1) if r: @@ -1481,13 +1487,13 @@ class Story(object): startMsg = self.startMessage await self.setCurrentMessage(startMsg) await self._renderer() - + def isFinished(self): if hasattr(self, 'finish_time') and self.finish_time: return time.time() - self.finish_time - + return False - + def _finish(self): """ Finish story and set hugvey to the right state @@ -1495,7 +1501,7 @@ class Story(object): self.finish() #stop google etc: self.hugvey.available() - + def finish(self): """ Finish only the story @@ -1505,50 +1511,50 @@ class Story(object): self.stop() self.finish_time = time.time() self.timer.pause() - + def calculateFinishesForMsg(self, msgId, depth = 0): if not msgId in self.directionsPerMsg or len(self.directionsPerMsg[msgId]) < 1: # is finish return [msgId] - + if depth > 40: return [] - + finishes = [] for d in self.directionsPerMsg[msgId]: if d.msgTo.id == msgId: continue finishes.extend(self.calculateFinishesForMsg(d.msgTo.id, depth+1)) - + # de-duplicate before returning return list(set(finishes)) - + def calculateFinishesForStrands(self): for startMsgId in self.strands: msg = self.get(startMsgId) #: :type msg: Message if msg.isStart: # ignore for the beginning continue - + self.logger.log(LOG_BS, f"Get finishes for {startMsgId}") self.strands[startMsgId] = self.calculateFinishesForMsg(startMsgId) - + self.logger.log(LOG_BS, f"Finishes: {self.strands}") - + def getFinishesForMsg(self, msg): """ Find the end of strands - + Most often they will be 'start's so to speed up these are pre-calculated Others can be calculated on the spot - + returns message ids """ if msg.id in self.strands: return self.strands[msg.id] - + return self.calculateFinishesForMsg(msg.id) - + def getDefaultDirectionForMsg(self, msg): """ There is only a default direction (for reply contains diversion) if it has @@ -1557,10 +1563,10 @@ class Story(object): if not msg.id in self.directionsPerMsg: # is finish return None - + if len(self.directionsPerMsg[msg.id]) > 1: return None - + # TODO: should the direction have at least a timeout condition set, or not perse? - - return self.directionsPerMsg[msg.id][0] \ No newline at end of file + + return self.directionsPerMsg[msg.id][0]