minor refactor

utils/server_utils.py

@@ -1,7 +1,25 @@
 import asyncio
 from functools import partial
+import contextlib
+from threading import Lock
+from typing import ContextManager, Generic, TypeVar
 
 def run_in_executor(func, *args, executor=None, **kwargs):
     callback = partial(func, *args, **kwargs)
     loop = asyncio.get_event_loop()
     return asyncio.get_event_loop().run_in_executor(executor, callback)
+
+
+T = TypeVar("T")
+class Mutex(Generic[T]):
+    def __init__(self, value: T):
+        self.__value = value
+        self.__lock = Lock()
+
+    @contextlib.contextmanager
+    def lock(self) -> ContextManager[T]:
+        self.__lock.acquire()
+        try:
+            yield self.__value
+        finally:
+            self.__lock.release()

utils/text_utilities.py

@@ -19,6 +19,9 @@ def preprocess_sentence(sentence):
     return ' '.join(tokens)
 
 def compute_similarity(sent1, sent2):
+    """
+    Compute the similarity
+    """
     tfidf_vectorizer = TfidfVectorizer()
     if sent1 is not None and sent2 is not None:
         tfidf_matrix = tfidf_vectorizer.fit_transform([sent1, sent2])

utils/viz_utilities.py

@@ -19,6 +19,7 @@ spacy_stopwords = en.Defaults.stop_words
 
 STOPWORDS = set(STOPWORDS).union(set(stopwords.words("english"))).union(set(spacy_stopwords))
 
+
 def create_wordcloud(timestamp, real_time=False):
     """
     Create a basic word cloud visualization of transcribed text