jaccard, temp, etc.

utterance/osc.py

@@ -37,6 +37,11 @@ class OscBroadcaster:
 		osc_send(msg, self.name)
 		update()
 
+	def temperature(self, temp: float, channel: str):
+		msg = oscbuildparse.OSCMessage(channel, None, [temp])
+		osc_send(msg, self.name)
+		update()
+
 class OscReceiver:
 
 	def __init__(self, name: str, host: str, port: str, callback_fn=None):

utterance/utils.py

@@ -16,7 +16,7 @@ def clean(text: str) -> str:
 
 def format(text: str) -> str:
 
-	return text.replace('\r\n', '\n').replace('\n\n', '\n')
+	return text.replace('\r\n', '\n').replace('\n\n', '\n').replace('‘', "'").replace("’", "'")
 
 def fragments(utterance: str):
 	frags = []

utterance/voice.py

@@ -1,6 +1,9 @@
 from aitextgen import aitextgen
 import utterance.utils
-import regex, string, time
+import gensim, regex, string, time
+
+UTTERANCE_MEMORY_LEN = 15
+UTTERANCE_MEMORY_MIN_DIST = 0.2
 
 class Voice:
 
@@ -9,6 +12,7 @@ class Voice:
 		self.temp = temp
 		self.lenght = lenght
 		self.v = aitextgen(model_folder=model, tokenizer_file=tokenizer)
+		self.utterances = []
 
 	def utter_n(self, n: int, temp: float = None, lenght: int = None):
 		t = self.temp if temp == None else temp
@@ -20,12 +24,49 @@ class Voice:
 		l = self.lenght if lenght == None else lenght		
 		return self.utter_n(n=1, temp=temp, lenght=lenght)[0]
 
+	def select(self, utterance: str) -> bool:
+
+		# fuction making sure to not say the same thing
+
+		print("select;")
+
+		toks = set(gensim.utils.tokenize(utterance))
+		
+		i = 0
+		for u in self.utterances:
+			d = gensim.matutils.jaccard_distance(toks, u)
+			print(f"{i} - d: {d}")
+			if d < UTTERANCE_MEMORY_MIN_DIST:
+				return False
+
+		self.utterances.append(toks)
+
+		print(len(self.utterances))
+
+		if len(self.utterances) > UTTERANCE_MEMORY_LEN:
+			self.utterances.pop(0)
+
+		return True
+
+
 	def set_channel(self, root: str, endpoint: str):
 		self.channel = root + endpoint
 
 	def channel(self):
 		return self.channel
 
+	def set_calculate(self, root: str, endpoint: str):
+		self.calculate = root + endpoint
+
+	def calculate(self):
+		return self.calculate
+
+	def set_temperature(self, root: str, endpoint: str):
+		self.temperature = root + endpoint
+
+	def calculate(self):
+		return self.temperature
+
 	def fragments(self, utt: str):
 		self.utterance = utt
 		self.utterance_fragments = utterance.utils.fragments(utt)