Update alma output

app.py

@@ -4,7 +4,7 @@ import os
 from pydantic import BaseModel
 from fastapi.responses import JSONResponse
 
-print("Version ---- 1")
+print("Version ---- 2")
 app = FastAPI()
 
 def download_file_from_hf(repo_id, filename):
@@ -27,7 +27,7 @@ download_file_from_hf("SinaLab/ArabGlossBERT","four_grams.pickle")
 download_file_from_hf("SinaLab/ArabGlossBERT","five_grams.pickle")
 
 from sinatools.morphology.morph_analyzer import analyze
-
+from sinatools.utils.tokenizer import sentence_tokenizer
 
 class ALMARequest(BaseModel):
     text: str
@@ -42,12 +42,22 @@ def predict(request: ALMARequest):
     task = request.task
     flag = request.flag
 
-    alma_output = analyze(text, language, task, str(flag))
-    content = {
-        "resp": alma_output,
-        "statusText": "OK",
-        "statusCode": 0,
-    }
+    sentences = sentence_tokenizer(text, dot=True, new_line=True, question_mark=True, exclamation_mark=True)
+    
+    results_with_sentences = []
+        
+    for i, sentence in enumerate(sentences):
+        sentence_id = i + 1
+        
+        result = analyze(sentence, language, task, str(flag))
+            
+        results_with_sentences.append({
+            "sentence_id": sentence_id,
+            "sentence": sentence,
+            "lemmatizer_results": result
+        })
+    
+    content = {"resp": results_with_sentences, "statusText": "OK", "statusCode": 0}
 
     return JSONResponse(
         content=content,