Add ALMA logic

app.py

@@ -1,7 +1,56 @@
 from fastapi import FastAPI
+from huggingface_hub import snapshot_download
+from huggingface_hub import hf_hub_download
+import os
+from pydantic import BaseModel
+from fastapi.responses import JSONResponse
 
 app = FastAPI()
 
-@app.get("/")
-def greet_json():
-    return {"Hello": "World!"}
+def download_file_from_hf(repo_id, filename):
+    """
+    Downloads a single file from a Hugging Face repo into ~/.sinatools
+
+    Args:
+        repo_id (str): Hugging Face repo id, e.g. "SinaLab/ArabGlossBERT"
+        filename (str): Path of the file inside the repo, e.g. "config.json"
+
+    Returns:
+        str: Absolute path to the downloaded file
+    """
+    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")
+
+from sinatools.morphology.morph_analyzer import analyze
+
+
+class ALMARequest(BaseModel):
+    text: str
+
+@app.post("/predict")
+def predict(request: ALMARequest):
+    text = request.text  
+    
+    alma_output = analyze(text)
+    content = {
+        "resp": alma_output,
+        "statusText": "OK",
+        "statusCode": 0,
+    }
+
+    return JSONResponse(
+        content=content,
+        media_type="application/json",
+        status_code=200,
+    )