return pictogram id list along with image

main.py

@@ -5,7 +5,7 @@ from transformers import MBartForConditionalGeneration, MBartTokenizerFast
 import torch
 import requests, json, base64
 from io import BytesIO
-from typing import Optional
+from typing import List, Optional
 
 
 # Initialize FastAPI app with versioning
@@ -25,6 +25,7 @@ class TranslationResponse(BaseModel):
     language: str
     src: str
     tgt: Optional[str] = None
+    pictogram_ids: List[Optional[int]] = None
     image_base64: Optional[str] = None
 
 
@@ -87,6 +88,7 @@ def translate(request: TranslationRequest):
             language=request.language,
             src=request.src,
             tgt=None,
+            pictogram_ids=[],
             image_base64=None,
         )
 
@@ -111,6 +113,7 @@ def translate(request: TranslationRequest):
                 language=request.language,
                 src=request.src,
                 tgt=tgt_sentence,
+                pictogram_ids=pictogram_ids,
                 image_base64=encoded_image,
             )
 
@@ -118,5 +121,6 @@ def translate(request: TranslationRequest):
         language=request.language,
         src=request.src,
         tgt=tgt_sentence,
+        pictogram_ids=pictogram_ids,
         image_base64=None,
     )