make gpt more robust

zyddnys · Jun 17, 2024 · 371f5aa · 371f5aa
1 parent ad5171f
commit 371f5aa
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/manga_translator/translators/chatgpt.py b/manga_translator/translators/chatgpt.py
@@ -174,10 +174,10 @@ async def _translate(self, from_lang: str, to_lang: str, queries: List[str]) ->
                 # Try splitting by newlines instead
                 new_translations = re.split(r'\n', response)
 
-            if len(new_translations) != query_size:
-                # super method will repeat translation as per self._INVALID_REPEAT_COUNT
-                translations = []
-                break
+            if len(new_translations) > query_size:
+                new_translations = new_translations[: len(query_size)]
+            elif len(new_translations) < query_size :
+                new_translations = new_translations + [''] * (query_size - len(new_translations))
 
             translations.extend([t.strip() for t in new_translations])