Add custom dictionary for accurate UI term translations

app.py

@@ -8,6 +8,7 @@ from flask_cors import CORS
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 from pathlib import Path
 import os
+import json
 
 # Initialize Flask app
 app = Flask(__name__)
@@ -16,6 +17,25 @@ CORS(app)  # Enable CORS for all routes
 # Global variables for model
 translator = None
 MODEL_DIR = Path("models/en-mr-marianmt")
+translations_dict = None
+
+def load_translations_dict():
+    """Load the custom translations dictionary"""
+    global translations_dict
+    
+    if translations_dict is not None:
+        return translations_dict
+    
+    dict_path = Path("translations_dict.json")
+    if dict_path.exists():
+        with open(dict_path, 'r', encoding='utf-8') as f:
+            translations_dict = json.load(f)
+        print("✓ Translations dictionary loaded")
+    else:
+        translations_dict = {"en_to_mr": {}, "mr_to_en": {}}
+        print("⚠ Translations dictionary not found, using model only")
+    
+    return translations_dict
 
 def load_model():
     """Load the MarianMT translation model"""
@@ -39,6 +59,32 @@ def load_model():
         print(f"✗ Failed to load model: {e}")
         raise
 
+def translate_with_dict(text, source, target):
+    """
+    Translate using dictionary first, fallback to model
+    Returns (translation, used_dict)
+    """
+    dict_data = load_translations_dict()
+    
+    # Normalize text for dictionary lookup
+    text_lower = text.lower().strip()
+    
+    # Check dictionary first
+    if source == "en" and target == "mr":
+        if text_lower in dict_data["en_to_mr"]:
+            return dict_data["en_to_mr"][text_lower], True
+    elif source == "mr" and target == "en":
+        if text in dict_data["mr_to_en"]:
+            return dict_data["mr_to_en"][text], True
+    
+    # Fallback to model
+    if translator is None:
+        load_model()
+    
+    result = translator(text, max_length=512)
+    translated = result[0]['translation_text'] if result else text
+    return translated, False
+
 @app.route('/')
 def home():
     """Home endpoint with API information"""
@@ -122,9 +168,10 @@ def translate():
         if source == target:
             return jsonify({"error": "Source and target languages must be different"}), 400
         
-        # Load model if not loaded
+        # Load model and dictionary if not loaded
         if translator is None:
             load_model()
+        load_translations_dict()
         
         # Handle batch translation
         is_batch = isinstance(text, list)
@@ -137,8 +184,7 @@ def translate():
                 translations.append("")
                 continue
             
-            result = translator(t, max_length=512)
-            translated = result[0]['translation_text'] if result else t
+            translated, from_dict = translate_with_dict(t, source, target)
             translations.append(translated)
         
         # Return response
@@ -170,6 +216,7 @@ if __name__ == '__main__':
     print("=" * 60)
     
     try:
+        load_translations_dict()
         load_model()
         print("\nStarting server...")
         

translations_dict.json

@@ -0,0 +1,153 @@
+{
+  "en_to_mr": {
+    "home": "घर",
+    "profile": "प्रोफाइल",
+    "settings": "सेटिंग्स",
+    "events": "कार्यक्रम",
+    "driver": "चालक",
+    "host": "यजमान",
+    "login": "लॉगिन",
+    "logout": "लॉगआउट",
+    "welcome": "स्वागत",
+    "search": "शोधा",
+    "save": "जतन करा",
+    "cancel": "रद्द करा",
+    "delete": "हटवा",
+    "edit": "संपादित करा",
+    "notifications": "सूचना",
+    "password": "पासवर्ड",
+    "email": "ईमेल",
+    "username": "वापरकर्तानाव",
+    "dashboard": "डॅशबोर्ड",
+    "menu": "मेनू",
+    "help": "मदत",
+    "about": "बद्दल",
+    "contact": "संपर्क",
+    "submit": "सबमिट करा",
+    "back": "मागे",
+    "next": "पुढे",
+    "previous": "मागील",
+    "continue": "सुरू ठेवा",
+    "confirm": "पुष्टी करा",
+    "close": "बंद करा",
+    "open": "उघडा",
+    "new": "नवीन",
+    "create": "तयार करा",
+    "update": "अपडेट करा",
+    "upload": "अपलोड करा",
+    "download": "डाउनलोड करा",
+    "share": "शेअर करा",
+    "send": "पाठवा",
+    "receive": "प्राप्त करा",
+    "accept": "स्वीकारा",
+    "reject": "नाकारा",
+    "approve": "मंजूर करा",
+    "decline": "नकार द्या",
+    "yes": "होय",
+    "no": "नाही",
+    "ok": "ठीक आहे",
+    "error": "त्रुटी",
+    "success": "यशस्वी",
+    "warning": "चेतावणी",
+    "info": "माहिती",
+    "loading": "लोड होत आहे",
+    "please wait": "कृपया प्रतीक्षा करा",
+    "try again": "पुन्हा प्रयत्न करा",
+    "refresh": "रिफ्रेश करा",
+    "reload": "पुन्हा लोड करा",
+    "view": "पहा",
+    "details": "तपशील",
+    "description": "वर्णन",
+    "name": "नाव",
+    "date": "तारीख",
+    "time": "वेळ",
+    "location": "स्थान",
+    "address": "पत्ता",
+    "phone": "फोन",
+    "message": "संदेश",
+    "comment": "टिप्पणी",
+    "reply": "उत्तर द्या",
+    "like": "आवडले",
+    "dislike": "आवडले नाही",
+    "favorite": "आवडते",
+    "bookmark": "बुकमार्क",
+    "follow": "फॉलो करा",
+    "unfollow": "अनफॉलो करा",
+    "block": "ब्लॉक करा",
+    "report": "तक्रार करा",
+    "filter": "फिल्टर",
+    "sort": "क्रमवारी लावा",
+    "category": "श्रेणी",
+    "tag": "टॅग",
+    "status": "स्थिती",
+    "active": "सक्रिय",
+    "inactive": "निष्क्रिय",
+    "online": "ऑनलाइन",
+    "offline": "ऑफलाइन",
+    "available": "उपलब्ध",
+    "unavailable": "अनुपलब्ध",
+    "busy": "व्यस्त",
+    "away": "दूर",
+    "account": "खाते",
+    "user": "वापरकर्ता",
+    "admin": "प्रशासक",
+    "guest": "अतिथी",
+    "member": "सदस्य",
+    "premium": "प्रीमियम",
+    "free": "मोफत",
+    "paid": "सशुल्क",
+    "subscription": "सदस्यता",
+    "plan": "योजना",
+    "price": "किंमत",
+    "total": "एकूण",
+    "subtotal": "उपएकूण",
+    "discount": "सवलत",
+    "tax": "कर",
+    "payment": "पेमेंट",
+    "checkout": "चेकआउट",
+    "cart": "कार्ट",
+    "wishlist": "इच्छासूची",
+    "order": "ऑर्डर",
+    "shipping": "शिपिंग",
+    "delivery": "डिलिव्हरी",
+    "track": "ट्रॅक करा",
+    "invoice": "बीजक",
+    "receipt": "पावती",
+    "refund": "परतावा",
+    "return": "परत करा",
+    "exchange": "बदला",
+    "support": "समर्थन",
+    "feedback": "अभिप्राय",
+    "rating": "रेटिंग",
+    "review": "पुनरावलोकन",
+    "terms": "अटी",
+    "privacy": "गोपनीयता",
+    "policy": "धोरण",
+    "legal": "कायदेशीर",
+    "copyright": "कॉपीराइट",
+    "version": "आवृत्ती",
+    "language": "भाषा",
+    "theme": "थीम",
+    "dark mode": "डार्क मोड",
+    "light mode": "लाइट मोड"
+  },
+  "mr_to_en": {
+    "घर": "home",
+    "प्रो���ाइल": "profile",
+    "सेटिंग्स": "settings",
+    "कार्यक्रम": "events",
+    "चालक": "driver",
+    "यजमान": "host",
+    "लॉगिन": "login",
+    "लॉगआउट": "logout",
+    "स्वागत": "welcome",
+    "शोधा": "search",
+    "जतन करा": "save",
+    "रद्द करा": "cancel",
+    "हटवा": "delete",
+    "संपादित करा": "edit",
+    "सूचना": "notifications",
+    "पासवर्ड": "password",
+    "ईमेल": "email"
+  }
+}