Update app.py

app.py

@@ -89,7 +89,8 @@ except Exception as e:
 LANGUAGE_CODES = {
     "kapampangan": "pam",
     "filipino": "fil",  # Replaced tagalog with filipino
-    "english": "eng"
+    "english": "eng",
+    "tagalog": "tgl",
 }
 
 # TTS Models (Kapampangan, Tagalog, English)
@@ -126,8 +127,7 @@ TRANSLATION_MODELS = {
     "pam-eng": "Coco-18/opus-mt-pam-en",
     "eng-pam": "Coco-18/opus-mt-en-pam",
     "tgl-eng": "Helsinki-NLP/opus-mt-tl-en",
-    "eng-tgl": "Helsinki-NLP/opus-mt-en-tl"
-     # Special model for pam-fil translations in both directions
+    "eng-tgl": "Helsinki-NLP/opus-mt-en-tl",
     "phi": "Coco-18/opus-mt-phi"  
 }
 
@@ -175,17 +175,18 @@ def home():
 @app.route("/health", methods=["GET"])
 def health_check():
     # Initialize direct language pair statuses based on loaded models
-    translation_status = {
-        "pam-eng": "loaded" if "pam-eng" in translation_models and translation_models["pam-eng"] is not None else "failed",
-        "eng-pam": "loaded" if "eng-pam" in translation_models and translation_models["eng-pam"] is not None else "failed",
-        "fil-eng": "loaded" if "fil-eng" in translation_models and translation_models["fil-eng"] is not None else "failed",
-        "eng-fil": "loaded" if "eng-fil" in translation_models and translation_models["eng-fil"] is not None else "failed",
-    }
+    translation_status = {}
+    
+    # Add status for direct model pairs
+    for lang_pair in ["pam-eng", "eng-pam", "tgl-eng", "eng-tgl"]:
+        translation_status[lang_pair] = "loaded" if lang_pair in translation_models and translation_models[lang_pair] is not None else "failed"
     
-    # Add special phi model status for pam-fil translations
+    # Add special phi model status
     phi_status = "loaded" if "phi" in translation_models and translation_models["phi"] is not None else "failed"
     translation_status["pam-fil"] = phi_status
     translation_status["fil-pam"] = phi_status
+    translation_status["pam-tgl"] = phi_status  # Using phi model but replacing tgl with fil
+    translation_status["tgl-pam"] = phi_status  # Using phi model but replacing tgl with fil
     
     health_status = {
         "api_status": "online",
@@ -396,8 +397,22 @@ def translate_text():
         
         logger.info(f"🔄 Translating from {source_language} to {target_language}: '{source_text}'")
         
-        # Special handling for pam-fil and fil-pam using the single phi model
+        # Special handling for pam-fil, fil-pam, pam-tgl and tgl-pam using the phi model
+        use_phi_model = False
+        actual_source_code = source_code
+        actual_target_code = target_code
+        
+        # Check if we need to use the phi model with fil replacement
         if (source_code == "pam" and target_code == "fil") or (source_code == "fil" and target_code == "pam"):
+            use_phi_model = True
+        elif (source_code == "pam" and target_code == "tgl"):
+            use_phi_model = True
+            actual_target_code = "fil"  # Replace tgl with fil for the phi model
+        elif (source_code == "tgl" and target_code == "pam"):
+            use_phi_model = True
+            actual_source_code = "fil"  # Replace tgl with fil for the phi model
+            
+        if use_phi_model:
             model_key = "phi"
             
             # Check if we have the phi model
@@ -411,7 +426,9 @@ def translate_text():
                 tokenizer = translation_tokenizers[model_key]
                 
                 # Prepend target language token to input
-                input_text = f">>{target_code}<< {source_text}"
+                input_text = f">>{actual_target_code}<< {source_text}"
+                
+                logger.info(f"🔄 Using phi model with input: '{input_text}'")
                 
                 # Tokenize the text
                 tokenized = tokenizer(input_text, return_tensors="pt", padding=True)