Update utils.py

utils.py

@@ -10,6 +10,7 @@ import numpy as np
 import asyncio
 import nltk
 from sentence_transformers import SentenceTransformer, util
+import signal
 
 nltk.download('punkt')
 
@@ -189,21 +190,36 @@ async def parallel_scrap(urls):
     return results
 
 
+class TimeoutError(Exception):
+    pass
+
+
+def signal_handler(signum, frame):
+    raise TimeoutError("Function timed out")
+    
+
 def matchingScore(sentence, content):
-    if sentence in content:
-        return 1
-    sentence = removePunc(sentence)
-    content = removePunc(content)
-    if sentence in content:
-        return 1
-    else:
-        n = 5
-        ngrams = getQueries(sentence, n)
-        print("ngrams done.......")
-        if len(ngrams) == 0:
-            return 0
-        matched = [x for x in ngrams if " ".join(x) in content]
-        return len(matched) / len(ngrams)
+    signal.signal(signal.SIGALRM, signal_handler)
+    signal.alarm(10)  # Set alarm for 10 seconds
+
+    try:
+        if sentence in content:
+            return 1
+        sentence = removePunc(sentence)
+        content = removePunc(content)
+        if sentence in content:
+            return 1
+        else:
+            n = 5
+            ngrams = getQueries(sentence, n)
+            print("ngrams done.......")
+            if len(ngrams) == 0:
+                return 0
+            matched = [x for x in ngrams if " ".join(x) in content]
+            signal.alarm(0)  # Cancel the alarm if calculation completes before timeout
+            return len(matched) / len(ngrams)
+    except TimeoutError:
+        return 0
 
 
 async def matchingScoreAsync(sentences, content, content_idx, ScoreArray):