add debug

app.py

@@ -126,22 +126,32 @@ def opus_trans(article, target_language):
 
 
 def nllb_trans(article, target_language):
+    result_lang = detect_lang(article)
+
     tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
     model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
     inputs = tokenizer(article, return_tensors="pt")
 
     if target_language == "English":
         target_lang = "eng_Latn"
+        target_language = "en"
     elif target_language == "Chinese":
         target_lang = "zho_Hans"
+        target_language = "zh"
 
-    translated_tokens = model.generate(
-        **inputs,
-        forced_bos_token_id=tokenizer.lang_code_to_id[target_lang],
-        max_length=30,
-    )
+    if result_lang != target_language:
+        translated_tokens = model.generate(
+            **inputs,
+            forced_bos_token_id=tokenizer.lang_code_to_id[target_lang],
+            max_length=30,
+        )
+        translated = tokenizer.batch_decode(
+            translated_tokens, skip_special_tokens=True
+        )[0]
+    else:
+        translated = "Error: You chose the same language as the article detected language. Please reselect language and try again."
 
-    return tokenizer.batch_decode(translated_tokens, skip_special_tokens=True)[0]
+    return translated
 
 
 def translate(article, toolkit, target_language):