Use sinatools synonyms generator

app.py

@@ -1,7 +1,56 @@
 from fastapi import FastAPI
+from huggingface_hub import hf_hub_download
+import os
+from pydantic import BaseModel
+from fastapi.responses import JSONResponse
 
+print("Version ---- 1")
 app = FastAPI()
 
-@app.get("/")
-def greet_json():
-    return {"Hello": "World!"}
+def download_file_from_hf(repo_id, filename):
+    target_dir = os.path.expanduser("~/.sinatools")
+    os.makedirs(target_dir, exist_ok=True)
+
+    file_path = hf_hub_download(
+        repo_id=repo_id,
+        filename=filename,
+        local_dir=target_dir,
+        local_dir_use_symlinks=False
+    )
+
+    return file_path
+
+download_file_from_hf("SinaLab/ALMA","lemmas_dic.pickle")
+download_file_from_hf("SinaLab/synonyms_models","graph_l2.pkl")
+download_file_from_hf("SinaLab/synonyms_models","graph_l3.pkl")
+
+from sinatools.synonyms.synonyms_generator import extend_synonyms
+from sinatools.morphology.morph_analyzer import analyze
+
+class synonymsRequest(BaseModel):
+    synset:  str
+    useALMA: str
+    level: str
+
+@app.post("/predict")
+def predict(request: synonymsRequest):
+    synset  = request.synset  
+    useALMA = request.useALMA
+    level   = int(request.level)
+
+    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,
+        media_type="application/json",
+        status_code=200,
+    )

requirements.txt

@@ -1,2 +1,3 @@
 fastapi
-uvicorn[standard]
+uvicorn[standard]
+SinaTools