Refactor translation logic to handle long chunks

app.py

@@ -40,11 +40,17 @@ def translate(text: str, src_lang: str, tgt_lang: str):
     # split the input text into smaller chunks
     # split first on newlines
     outputs = ""
-    chunks = text.split("\n")
-    for chunk in chunks:
-        # run the translation on each chunk
-        output = _translate(chunk, src_lang, tgt_lang)
-        outputs += output + "\n"
+    paragraph_chunks = text.split("\n")
+    for chunk in paragraph_chunks:
+        # check if the chunk is too long
+        if len(chunk) > 500:
+            # split on full stops
+            sentence_chunks = chunk.split(".")
+            for sentence in sentence_chunks:
+                outputs += f"{_translate(sentence, src_lang, tgt_lang)}. "
+        else:
+            outputs += _translate(chunk, src_lang, tgt_lang) + "\n"
+
     return outputs