update more logging

app.py

@@ -172,6 +172,10 @@ def nllb_trans(article, target_language):
 
 
 def mbart_trans(article, target_language):
+    logger.info(f"Article to translate : {article}")
+    logger.info(f"Chose which translation model: mbart model")
+    logger.info(f"Language selected: {target_language}")
+
     result_lang = detect_lang(article)
     if target_language == "English":
         target_lang = "en"
@@ -180,14 +184,27 @@ def mbart_trans(article, target_language):
 
     if result_lang != target_lang:
         if target_language == "English":
+            logger.info(
+                f'Translated Result: {mbart_m2en_model.translate(article, target_lang="en")}'
+            )
             return mbart_m2en_model.translate(article, target_lang="en")
         else:
+            logger.info(
+                f'Translated Result: {mbart_m2en_model.translate(article, target_lang="zh")}'
+            )
             return mbart_en2m_model.translate(article, target_lang="zh")
     else:
+        logger.warn(
+            "Error: You chose the same language as the article detected language. Please reselect language and try again."
+        )
         return "Error: You chose the same language as the article detected language. Please reselect language and try again."
 
 
 def m2m_trans(article, target_language):
+    logger.info(f"Article to translate : {article}")
+    logger.info(f"Chose which translation model: m2m model")
+    logger.info(f"Language selected: {target_language}")
+
     result_lang = detect_lang(article)
     if target_language == "English":
         target_lang = "en"
@@ -195,10 +212,19 @@ def m2m_trans(article, target_language):
         target_lang = "zh"
     if result_lang != target_lang:
         if target_language == "English":
+            logger.info(
+                f"Translation Result: {m2m_model.translate(article, target_lang)}"
+            )
             return m2m_model.translate(article, target_lang)
         elif target_language == "Chinese":
+            logger.info(
+                f"Translation Result: {m2m_model.translate(article, target_lang)}"
+            )
             return m2m_model.translate(article, target_lang)
     else:
+        logger.warn(
+            f"Error: You chose the same language as the article detected language. Please reselect language and try again."
+        )
         return "Error: You chose the same language as the article detected language. Please reselect language and try again."
 
 
@@ -228,9 +254,6 @@ with gr.Blocks(theme=myTheme) as demo:
     trans_btn.click(
         fn=translate, inputs=[article, toolkit_select, lang_select], outputs=result
     )
-    logger.info(f"Article to translate : {article}")
-    logger.info(f"Chose which translation model: {toolkit_select}")
-    logger.info(f"Language selected: {lang_select}")
-    logger.info(f"Translation Result: {result}")
+
 
 demo.launch()