Add evaluate_synonyms function

app.py

@@ -29,29 +29,36 @@ download_file_from_hf("SinaLab/ArabGlossBERT","four_grams.pickle")
 download_file_from_hf("SinaLab/ArabGlossBERT","five_grams.pickle")
 
 from sinatools.synonyms.synonyms_generator import extend_synonyms
+from sinatools.synonyms.synonyms_generator import evaluate_synonyms
 from sinatools.morphology.morph_analyzer import analyze
 
 class synonymsRequest(BaseModel):
     synset:  str
     useALMA: str
     level: str
+    serviceType: str
 
 @app.post("/predict")
 def predict(request: synonymsRequest):
     synset  = request.synset  
     useALMA = request.useALMA
     level   = int(request.level)
+    serviceType = request.serviceType
 
-    final_synset = [] 
-    if useALMA == True:
-        synonyms = synset.split("|")
-        for synonym in synonyms: 
-            final_synset.append(analyze(synonym.strip(), language = "MSA", task = "lemmatization", flag = "1")[0]["lemma"])
-        list_of_synon_with_fuzzy_value = extend_synonyms(" | ".join(final_synset), level)
-    else:
-        list_of_synon_with_fuzzy_value = extend_synonyms(synset, level)
-    
-    content = {"resp": list_of_synon_with_fuzzy_value, "statusText": "OK","statusCode" : 0}
+    if serviceType == "evaluate": 
+        list_of_synon_with_fuzzy_value = evaluate_synonyms(synset, level)
+        content = {"resp": list_of_synon_with_fuzzy_value, "statusText": "OK","statusCode" : 0}
+    else: 
+        final_synset = [] 
+        if useALMA == True:
+            synonyms = synset.split("|")
+            for synonym in synonyms: 
+                final_synset.append(analyze(synonym.strip(), language = "MSA", task = "lemmatization", flag = "1")[0]["lemma"])
+            list_of_synon_with_fuzzy_value = extend_synonyms(" | ".join(final_synset), level)
+        else:
+            list_of_synon_with_fuzzy_value = extend_synonyms(synset, level)
+        
+        content = {"resp": list_of_synon_with_fuzzy_value, "statusText": "OK","statusCode" : 0}
 
     return JSONResponse(
         content=content,