Update app.py

Tambah china tradisional dan vietnam

app.py

@@ -3,6 +3,7 @@ from transformers import pipeline
 import langdetect
 import logging
 import os
+import opencc  # Untuk konversi Mandarin Sederhana/Tradisional
 
 # Atur direktori cache
 os.environ["HF_HOME"] = "/app/cache"
@@ -20,7 +21,9 @@ MODEL_MAP = {
     "fr": "Helsinki-NLP/opus-mt-fr-en",  # Prancis ke Inggris
     "es": "Helsinki-NLP/opus-mt-es-en",  # Spanyol ke Inggris
     "ja": "Helsinki-NLP/opus-mt-ja-en",  # Jepang ke Inggris
-    "zh": "Helsinki-NLP/opus-mt-zh-en",  # Mandarin ke Inggris
+    "zh-CN": "Helsinki-NLP/opus-mt-zh-en",  # Mandarin Sederhana ke Inggris
+    "zh-TW": "Helsinki-NLP/opus-mt-zh-en",  # Mandarin Tradisional ke Inggris
+    "vi": "Helsinki-NLP/opus-mt-vi-en",  # Vietnam ke Inggris
 }
 
 # Inisialisasi pipeline untuk setiap model
@@ -34,10 +37,21 @@ except Exception as e:
     logger.error(f"Model initialization failed: {str(e)}")
     raise Exception(f"Model initialization failed: {str(e)}")
 
+# Inisialisasi konverter OpenCC untuk Mandarin Tradisional ke Sederhana
+converter = opencc.OpenCC('t2s')  # Tradisional ke Sederhana
+
+def is_traditional_chinese(text: str) -> bool:
+    """Cek apakah teks menggunakan karakter Mandarin Tradisional."""
+    traditional_chars = set('繁體字')  # Contoh karakter Tradisional
+    return any(char in traditional_chars for char in text)
+
 def detect_language(text: str) -> str:
     """Deteksi bahasa dari teks menggunakan langdetect."""
     try:
         lang = langdetect.detect(text)
+        # Jika terdeteksi sebagai Mandarin (zh), cek apakah Tradisional atau Sederhana
+        if lang == "zh":
+            return "zh-TW" if is_traditional_chinese(text) else "zh-CN"
         return lang if lang in MODEL_MAP else "en"  # Default ke Inggris jika tidak didukung
     except Exception as e:
         logger.warning(f"Language detection failed: {str(e)}, defaulting to English")
@@ -54,11 +68,16 @@ async def translate(text: str):
         # Jika sudah Inggris atau bahasa tidak didukung, kembalikan teks asli
         if source_lang == "en":
             return {"translated_text": text}
+        # Jika Mandarin Tradisional, konversi ke Sederhana
+        input_text = text
+        if source_lang == "zh-TW":
+            input_text = converter.convert(text)
+            logger.info("Converted Traditional Chinese to Simplified Chinese")
         # Terjemahkan ke Inggris
         translator = translators.get(source_lang)
         if not translator:
             raise HTTPException(status_code=400, detail=f"Translation not supported for language: {source_lang}")
-        result = translator(text)
+        result = translator(input_text)
         return {"translated_text": result[0]["translation_text"]}
     except Exception as e:
         logger.error(f"Processing failed: {str(e)}")