fix empty translation

GreenWizard2015 · Sep 29, 2023 · e36e6fd · e36e6fd
1 parent 2a7ba3d
commit e36e6fd
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/core/worker.py b/core/worker.py
@@ -91,7 +91,7 @@ def _fastTranslate(self, text, languageCode):
 
   def _fullTranslate(self, text, fastTranslation, language):
     if 0 == len(text):
-      yield "", False
+      yield None
       return
 
     translationProcess = self._assistant.translate(

diff --git a/main.py b/main.py
@@ -171,12 +171,15 @@ def fastTranslated(self, text):
 
   def fullTranslated(self, translationResult):
     self._lastAIResult = translationResult
-    self._refineBtn.config(state=tk.DISABLED if translationResult.pending else tk.NORMAL)
+    translation = '' if translationResult is None else translationResult.translation
+    pending = False if translationResult is None else translationResult.pending
+
+    self._refineBtn.config(state=tk.DISABLED if pending else tk.NORMAL)
 
     self._fullOutputText.delete("1.0", tk.END)
-    self._fullOutputText.insert(tk.END, translationResult.translation)
+    self._fullOutputText.insert(tk.END, translation)
 
-    if translationResult.pending:
+    if pending:
       notification = self._localization(
         "Translation is not accurate and will be updated soon."
       ).get()