Refactored text chunk splitting logic in translate function

app.py

@@ -42,20 +42,21 @@ def _translate(text: str, src_lang: str, tgt_lang: str):
 
 def translate(text: str, src_lang: str, tgt_lang: str):
     # split the input text into smaller chunks
-    # split first on newlines
     outputs = ""
     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(".")
+            # split on full stops, question marks, and exclamation marks
+            sentence_chunks = re.split(r"(?<=[.!?])\s+", chunk)
             for sentence in sentence_chunks:
-                outputs += f"{_translate(sentence, src_lang, tgt_lang)}. "
+                if sentence.strip():  # check if the sentence is not empty
+                    outputs += f"{_translate(sentence, src_lang, tgt_lang)} "
+            outputs += "\n\n"
         else:
             outputs += _translate(chunk, src_lang, tgt_lang) + "\n\n"
 
-    return outputs
+    return outputs.strip()
 
 
 description = """