sentence-wise translator

Signed-off-by: Ilham Syahid S <ilhamsyahids@gmail.com>

app.py

@@ -40,20 +40,30 @@ def translation(model_name, source, target, text):
         src_lang=source,
         tgt_lang=target,
     )
-    output = translator(text, max_length=400)
+
+    # sentence-wise translation
+    sentences = text.split("\n")
+    translated_sentences = []
+    for sentence in sentences:
+        translated_sentence = translator(sentence, max_length=400)[0][
+            "translation_text"
+        ]
+        translated_sentences.append(translated_sentence)
+    output = "\n".join(translated_sentences)
 
     end_time = time.time()
 
-    full_output = output
-    output = output[0]["translation_text"]
+    # output = translator(text, max_length=400)
+    # full_output = output
+    # output = output[0]["translation_text"]
     result = {
         "inference_time": end_time - start_time,
         "source": source,
         "target": target,
         "result": output,
-        "full_output": full_output,
+        # "full_output": full_output,
     }
-    return result
+    return result, output
 
 
 if __name__ == "__main__":
@@ -71,7 +81,7 @@ if __name__ == "__main__":
                 "nllb-distilled-600M",
                 "nllb-distilled-1.3B",
                 # "nllb-1.3B",
-                #  "nllb-3.3B"
+                # "nllb-3.3B"
             ],
             label="NLLB Model",
             default="nllb-distilled-1.3B",
@@ -81,15 +91,23 @@ if __name__ == "__main__":
         gr.inputs.Textbox(lines=5, label="Input text"),
     ]
 
-    outputs = gr.outputs.JSON()
+    outputs = [
+        gr.outputs.JSON(label="Metadata"),
+        gr.outputs.Textbox(label="Output text"),
+    ]
 
     title = "NLLB (No Language Left Behind) demo"
 
     demo_status = "Demo is running on CPU"
-    description = f"Using NLLB model, details: https://github.com/facebookresearch/fairseq/tree/nllb. {demo_status}"
+    description = f"Using NLLB model, details: https://github.com/facebookresearch/fairseq/tree/nllb.\n{demo_status}"
     examples = [
         ["nllb-distilled-1.3B", "Najdi Arabic", "English", "جلست اطفال"],
-        ["nllb-distilled-600M", "Najdi Arabic", "English", "شد للبيع طابقين مع شرع له نظيف حق غمارتين"]
+        [
+            "nllb-distilled-600M",
+            "Najdi Arabic",
+            "English",
+            "شد للبيع طابقين مع شرع له نظيف حق غمارتين",
+        ],
     ]
 
     gr.Interface(