Update app.py

app.py

@@ -28,121 +28,324 @@ def detect_arabizi(text):
     if not text or len(text.strip()) < 2:
         return False
     
-    # Remove spaces and convert to lowercase for analysis
-    clean_text = text.lower().replace(" ", "")
-    
     # Check for Arabic script - if present, it's NOT Arabizi
     arabic_pattern = r'[\u0600-\u06FF\u0750-\u077F\u08A0-\u08FF\uFB50-\uFDFF\uFE70-\uFEFF]'
     if re.search(arabic_pattern, text):
         return False
     
-    # Arabizi indicators
-    arabizi_numbers = ['2', '3', '7', '9']  # Common Arabic letter substitutions
+    # Arabizi indicators - numbers used as letters
+    arabizi_numbers = ['2', '3', '7', '9', '5', '6', '8']
+    has_arabizi_numbers = any(num in text for num in arabizi_numbers)
+    
+    # Common Arabizi words and patterns
     arabizi_patterns = [
-        'wach', 'wash', 'ach', 'achno', 'chno', 'shno',  # What
-        'kif', 'kifash', 'ki', 'kayf',  # How
-        'feen', 'fin', 'fen',  # Where  
-        'imta', 'meta', 'waqt',  # When
-        '3la', '3ala', 'ala',  # On/about
-        'hna', '7na', 'ahna',  # We/us
-        'nta', 'nti', 'ntuma',  # You
-        'howa', 'hiya', 'huma',  # He/she/they
-        'ma3', 'maa3', 'maak', 'maaki',  # With
-        'had', 'hadchi', 'hada', 'hadi',  # This
-        'bghit', 'bghiti', 'bgha',  # Want
-        'galt', 'galti', 'gal',  # Said
-        'rah', 'raha', 'rahi',  # Going
-        'kan', 'kanu', 'kana',  # Was/were
-        'ghadi', 'ghad', 'gha',  # Will/going to
-        'daba', 'dak', 'dakchi',  # Now/that
-        'bzf', 'bzzaf', 'bezzaf',  # A lot
-        'chway', 'chwiya', 'shwiya',  # A little
-        'khoya', 'khuya', 'akhi',  # Brother
-        'khti', 'khtiya', 'ukhti',  # Sister
-        'allah', 'llah', 'rabi',  # God
-        'inchallah', 'insha allah',  # God willing
-        'hamdulillah', 'alhamdulillah',  # Praise God
-        'salam', 'salamu aleikum',  # Peace
-        'baraka', 'barakallahu',  # Blessing
-        'yallah', 'yalla', 'hya'  # Come on/let's go
+        'wach', 'wash', 'ach', 'achno', 'chno', 'shno', 'shkoun', 'chkoun',
+        'kif', 'kifash', 'ki', 'kayf', 'kien', 'kima',
+        'feen', 'fin', 'fen', 'fain', 'mnin',
+        'imta', 'meta', 'waqt', 'mata', 'emta',
+        'hna', 'ahna', 'ana', 'nta', 'nti', 'ntuma', 'ntouma',
+        'howa', 'hiya', 'huma', 'houma', 'hoa', 'hia',
+        'had', 'hadchi', 'hada', 'hadi', 'hadou', 'hadouk',
+        'bghit', 'bghiti', 'bgha', 'bghina', 'bghitiou',
+        'galt', 'galti', 'gal', 'galet', 'galou',
+        'rah', 'raha', 'rahi', 'rahom', 'rahin',
+        'kan', 'kanu', 'kana', 'kanet', 'kano',
+        'ghadi', 'ghad', 'gha', 'ghadia', 'ghadiyin',
+        'daba', 'dak', 'dakchi', 'dik', 'dok',
+        'bzf', 'bzzaf', 'bezzaf', 'bzaaaaf',
+        'chway', 'chwiya', 'shwiya', 'chwia',
+        'khoya', 'khuya', 'akhi', 'kho',
+        'khti', 'khtiya', 'ukhti', 'kht',
+        'mama', 'baba', 'lwaldin', 'lwalidin',
+        'salam', 'salamu aleikum', 'slm',
+        'yallah', 'yalla', 'hya', 'aji',
+        'mabghitsh', 'mabghach', 'makansh', 'machi',
+        'walakin', 'walaken', 'ama', 'mais'
     ]
     
-    # Count Latin letters
-    latin_letters = sum(1 for c in clean_text if c.isalpha() and ord(c) < 128)
+    text_lower = text.lower()
+    has_arabizi_words = any(pattern in text_lower for pattern in arabizi_patterns)
     
-    # Count Arabizi number substitutions
-    arabizi_number_count = sum(1 for num in arabizi_numbers if num in clean_text)
+    # Decision logic
+    if has_arabizi_numbers and has_arabizi_words:
+        return True
+    if has_arabizi_numbers and len([c for c in text if c.isalpha()]) > len(text) * 0.6:
+        return True
+    if has_arabizi_words and len([c for c in text if c.isalpha()]) > len(text) * 0.7:
+        return True
     
-    # Count Arabizi word patterns
-    arabizi_word_count = sum(1 for pattern in arabizi_patterns if pattern in clean_text)
+    return False
+
+def arabizi_to_arabic(text):
+    """
+    Convert Arabizi text to Arabic script
+    COMPLETE HARD-CODED MAPPINGS
+    """
+    if not text:
+        return text
     
-    # Decision logic
-    total_chars = len(clean_text)
+    # COMPREHENSIVE WORD MAPPINGS (Arabizi → Arabic)
+    word_mappings = {
+        # Pronouns
+        'ana': 'أنا', 'nta': 'نتا', 'nti': 'نتي', 'howa': 'هوا', 'hiya': 'هيا',
+        'hoa': 'هوا', 'hia': 'هيا', 'hna': 'حنا', 'ahna': 'أحنا', 
+        'ntuma': 'نتوما', 'ntouma': 'نتوما', 'huma': 'هوما', 'houma': 'هوما',
+        
+        # Question words  
+        'shkoun': 'شكون', 'chkoun': 'شكون', 'skoun': 'شكون',
+        'achno': 'أشنو', 'chno': 'شنو', 'shno': 'شنو', 'ach': 'أش',
+        'wach': 'واش', 'wash': 'واش', 'ouach': 'واش',
+        'kifash': 'كيفاش', 'kif': 'كيف', 'ki': 'كي', 'kayf': 'كيف',
+        'feen': 'فين', 'fin': 'فين', 'fen': 'فين', 'fain': 'فين', 'mnin': 'منين',
+        'imta': 'إمتا', 'meta': 'متا', 'mata': 'متا', 'emta': 'إمتا',
+        '3lach': 'علاش', 'alach': 'علاش', '3la ach': 'علاش',
+        
+        # Common verbs
+        'bghit': 'بغيت', 'bghiti': 'بغيتي', 'bgha': 'بغا', 'bghina': 'بغينا',
+        'kan': 'كان', 'kana': 'كانا', 'kanet': 'كانت', 'kanu': 'كانو',
+        'galt': 'قلت', 'galti': 'قلتي', 'gal': 'قال', 'galet': 'قالت',
+        'galou': 'قالو', 'galina': 'قلنا',
+        'rah': 'راح', 'raha': 'راها', 'rahi': 'راهي', 'rahom': 'راهم',
+        'ghadi': 'غادي', 'ghad': 'غاد', 'gha': 'غا', 'ghadia': 'غادية',
+        'dir': 'دير', 'dert': 'درت', 'derat': 'درات', 'derna': 'درنا',
+        'ja': 'جا', 'jat': 'جات', 'jina': 'جينا', 'jaou': 'جاو',
+        'mcha': 'مشا', 'mchat': 'مشات', 'mchina': 'مشينا', 'mchaou': 'مشاو',
+        'khrj': 'خرج', 'khrja': 'خرجا', 'khrjat': 'خرجات', 'khrjna': 'خرجنا',
+        'dkhl': 'دخل', 'dkhla': 'دخلا', 'dkhlat': 'دخلات', 'dkhlna': 'دخلنا',
+        
+        # Demonstratives
+        'had': 'هاد', 'hada': 'هادا', 'hadi': 'هادي', 'hadou': 'هادو',
+        'hadouk': 'هادوك', 'hadchi': 'هادشي', 'hadak': 'هاداك',
+        'dak': 'داك', 'dik': 'ديك', 'dok': 'دوك', 'dakchi': 'داكشي',
+        
+        # Prepositions
+        'fi': 'في', 'f': 'ف', 'men': 'من', 'mn': 'من', 'l': 'ل', 'li': 'لي',
+        'bla': 'بلا', 'b': 'ب', 'bi': 'بي', 'bih': 'بيه', 'biha': 'بيها',
+        'mea': 'معا', 'maak': 'معاك', 'maaki': 'معاكي', 'maana': 'معانا',
+        'maahom': 'معاهم', 'maaha': 'معاها',
+        
+        # Numbers
+        'wahed': 'واحد', 'joj': 'جوج', 'tlata': 'تلاتا', 'rebaa': 'ربعا',
+        'khamsa': 'خمسا', 'setta': 'ستا', 'sebaa': 'سبعا', 'tmanya': 'تمنيا',
+        'tesaa': 'تسعا', 'ashra': 'عشرا',
+        
+        # Time expressions
+        'daba': 'دابا', 'tawa': 'توا', 'ghda': 'غدا', 'lbareeh': 'البارح',
+        'nhar': 'نهار', 'lila': 'ليلا', 'sbah': 'صباح', 'aachia': 'عشيا',
+        
+        # Family
+        'mama': 'ماما', 'baba': 'بابا', 'khoya': 'خويا', 'khti': 'ختي',
+        'khuya': 'خويا', 'khtiya': 'ختيا', 'akhi': 'أخي', 'ukhti': 'أختي',
+        'jad': 'جد', 'jada': 'جدا', 'amo': 'عمو', 'ama': 'عما',
+        'khal': 'خال', 'khala': 'خالا',
+        
+        # Adjectives
+        'zwin': 'زوين', 'zwina': 'زوينا', 'qbih': 'قبيح', 'qbiha': 'قبيحا',
+        'kbir': 'كبير', 'kbira': 'كبيرا', 'sghir': 'صغير', 'sghira': 'صغيرا',
+        'tqil': 'تقيل', 'tqila': 'تقيلا', 'khfif': 'خفيف', 'khfifa': 'خفيفا',
+        'sahel': 'ساهل', 'sahla': 'ساهلا', 'saab': 'صعب', 'saba': 'صعبا',
+        
+        # Quantities
+        'bzf': 'بزاف', 'bzzaf': 'بزاف', 'bezzaf': 'بزاف', 'ktir': 'كتير',
+        'chway': 'شوياة', 'chwiya': 'شوياة', 'shwiya': 'شوياة', 'chwia': 'شوياة',
+        'kolchi': 'كولشي', 'kol': 'كول', 'ga': 'غاع', 'gaa': 'غاع',
+        'hta': 'حتا', 'walo': 'والو', 'walou': 'والو',
+        
+        # Negations
+        'ma': 'ما', 'machi': 'ماشي', 'mabghitsh': 'مابغيتش', 'makainch': 'ماكاينش',
+        'makansh': 'ماكانش', 'maghatch': 'ماغاتش', 'mabaghish': 'مابغيش',
+        
+        # Conjunctions
+        'walakin': 'والاكين', 'walaken': 'والاكين', 'mais': 'ولكن',
+        'ama': 'أما', 'ida': 'إذا', 'ila': 'إلا', 'hta': 'حتا',
+        'ou': 'أو', 'o': 'أو', 'aw': 'أو', 'wla': 'ولا',
+        
+        # Expressions
+        'salam': 'سلام', 'yallah': 'يالاه', 'yalla': 'يالا', 'hya': 'هيا',
+        'aji': 'أجي', 'sir': 'سير', 'siri': 'سيري', 'ajou': 'أجو',
+        'mabrook': 'مبروك', 'baraka': 'باركا', 'besaha': 'بصحا',
+        'allah': 'الله', 'rabi': 'ربي', 'llah': 'الله',
+        'inchallah': 'إن شاء الله', 'machallah': 'ما شاء الله',
+        'hamdulillah': 'الحمد لله', 'alhamdulillah': 'الحمد لله',
+        'la hawla': 'لا حول', 'astghfirullah': 'أستغفر الله',
+        
+        # Countries and places
+        'lmaghrib': 'المغرب', 'maghrib': 'مغرب', 'casa': 'كازا',
+        'rbat': 'الرباط', 'fas': 'فاس', 'mknas': 'مكناس',
+        'tanja': 'طنجا', 'agadir': 'أكادير', 'marrakech': 'مراكش',
+        
+        # Common nouns
+        'dar': 'دار', 'bit': 'بيت', 'tomo': 'طوموبيل', 'karhouba': 'كرهوبا',
+        'makla': 'ماكلا', 'atay': 'أتاي', 'khobz': 'خوبز', 'lma': 'الما',
+        'tajin': 'طاجين', 'tajine': 'طاجين', 'couscous': 'كوسكوس',
+        'msemen': 'مسمن', 'rghaif': 'رغايف', 'harira': 'حريرا',
+        'khodra': 'خضرا', 'fawakeh': 'فواكه', 'lahem': 'ل��م',
+        'djaj': 'دجاج', 'hout': 'حوت', 'lben': 'اللبن',
+        'skar': 'سكر', 'melh': 'ملح', 'zit': 'زيت', 'zebda': 'زبدا',
+        
+        # Verbs (additional)
+        'chreb': 'شرب', 'chrba': 'شربا', 'chrebat': 'شربات', 'chrebna': 'شربنا',
+        'kla': 'كلا', 'klat': 'كلات', 'klina': 'كلينا', 'klaw': 'كلاو',
+        'nees': 'نعس', 'neesat': 'نعسات', 'neesna': 'نعسنا', 'neasou': 'نعسو',
+        'faq': 'فاق', 'faqat': 'فاقات', 'faqna': 'فقنا', 'faqou': 'فاقو',
+        'tlab': 'طلب', 'tlbat': 'طلبات', 'tlabna': 'طلبنا', 'tlabou': 'طلبو',
+        'khdam': 'خدم', 'khdmat': 'خدمات', 'khdamna': 'خدمنا', 'khdamou': 'خدمو',
+        'qra': 'قرا', 'qrat': 'قرات', 'qrina': 'قرينا', 'qraw': 'قراو',
+        'kteb': 'كتب', 'ktbat': 'كتبات', 'ktebna': 'كتبنا', 'ktbou': 'كتبو',
+        'chaf': 'شاف', 'chafat': 'شافات', 'chfna': 'شفنا', 'chafou': 'شافو',
+        'sma': 'سمع', 'smat': 'سمعات', 'smana': 'سمعنا', 'smaou': 'سمعو'
+    }
     
-    # Strong indicators
-    if arabizi_number_count >= 2:  # Multiple number substitutions
-        return True
+    # CHARACTER MAPPINGS (Arabizi numbers/letters → Arabic)
+    char_mappings = {
+        '2': 'ء', '3': 'ع', '5': 'خ', '6': 'ط', '7': 'ح', '8': 'غ', '9': 'ق',
+        'a': 'ا', 'b': 'ب', 't': 'ت', 'th': 'ث', 'j': 'ج', 'd': 'د',
+        'r': 'ر', 'z': 'ز', 's': 'س', 'sh': 'ش', 'f': 'ف', 'k': 'ك',
+        'l': 'ل', 'm': 'م', 'n': 'ن', 'h': 'ه', 'w': 'و', 'y': 'ي',
+        'aa': 'آ', 'ee': 'ي', 'oo': 'و', 'ou': 'و', 'ai': 'ي', 'ay': 'ي'
+    }
     
-    if arabizi_word_count >= 2:  # Multiple Arabizi words
-        return True
+    # DIGRAPH MAPPINGS (two-letter combinations first)
+    digraph_mappings = {
+        'kh': 'خ', 'gh': 'غ', 'ch': 'ش', 'th': 'ث', 'dh': 'ذ', 
+        'sh': 'ش', 'ts': 'ص', 'dz': 'ض', 'ss': 'ص', 'tt': 'ط',
+        'zz': 'ظ', 'aa': 'آ', 'ee': 'ي', 'ii': 'ي', 'oo': 'و', 
+        'uu': 'و', 'ou': 'و', 'ai': 'ي', 'ay': 'ي', 'ey': 'ي'
+    }
     
-    # Medium indicators 
-    if arabizi_number_count >= 1 and latin_letters > total_chars * 0.7:
-        return True
+    # Convert text
+    result = text.lower()
     
-    if arabizi_word_count >= 1 and latin_letters > total_chars * 0.8:
-        return True
+    # Step 1: Apply word mappings (most specific first)
+    for arabizi_word, arabic_word in word_mappings.items():
+        # Use word boundaries to avoid partial matches
+        pattern = r'\b' + re.escape(arabizi_word) + r'\b'
+        result = re.sub(pattern, arabic_word, result, flags=re.IGNORECASE)
     
-    # Weak but possible indicators
-    if latin_letters > total_chars * 0.9 and total_chars > 10:
-        # Mostly Latin letters in longer text - could be Arabizi
-        if arabizi_number_count >= 1 or arabizi_word_count >= 1:
-            return True
+    # Step 2: Apply digraph mappings
+    for digraph, arabic_char in digraph_mappings.items():
+        result = result.replace(digraph, arabic_char)
     
-    return False
+    # Step 3: Apply single character mappings
+    for char, arabic_char in char_mappings.items():
+        if len(char) == 1:  # Only single chars in this step
+            result = result.replace(char, arabic_char)
+    
+    return result
 
-def determine_response_language(user_input):
+def arabic_to_arabizi(text):
     """
-    Determine what language/script the response should be in
-    Returns: 'arabizi', 'arabic', or 'english'
+    Convert Arabic script to Arabizi
+    COMPLETE HARD-CODED MAPPINGS
     """
-    if detect_arabizi(user_input):
-        return 'arabizi'
+    if not text:
+        return text
     
-    # Check for Arabic script
-    arabic_pattern = r'[\u0600-\u06FF\u0750-\u077F\u08A0-\u08FF\uFB50-\uFDFF\uFE70-\uFEFF]'
-    if re.search(arabic_pattern, user_input):
-        return 'arabic'
+    # COMPREHENSIVE WORD MAPPINGS (Arabic → Arabizi)
+    word_mappings = {
+        # Pronouns
+        'أنا': 'ana', 'نتا': 'nta', 'نتي': 'nti', 'هوا': 'howa', 'هيا': 'hiya',
+        'حنا': 'hna', 'أحنا': 'ahna', 'نتوما': 'ntuma', 'هوما': 'huma',
+        
+        # Question words
+        'شكون': 'shkoun', 'أشنو': 'achno', 'شنو': 'chno', 'أش': 'ach',
+        'واش': 'wach', 'كيفاش': 'kifash', 'كيف': 'kif', 'فين': 'feen',
+        'منين': 'mnin', 'إمتا': 'imta', 'متا': 'meta', 'علاش': '3lach',
+        
+        # Common verbs
+        'بغيت': 'bghit', 'بغيتي': 'bghiti', 'بغا': 'bgha', 'بغينا': 'bghina',
+        'كان': 'kan', 'كانا': 'kana', 'كانت': 'kanet', 'كانو': 'kanu',
+        'قلت': 'galt', 'قلتي': 'galti', 'قال': 'gal', 'قالت': 'galet',
+        'قالو': 'galou', 'قلنا': 'galina',
+        'راح': 'rah', 'راها': 'raha', 'راهي': 'rahi', 'راهم': 'rahom',
+        'غادي': 'ghadi', 'غاد': 'ghad', 'غ��': 'gha', 'غادية': 'ghadia',
+        'دير': 'dir', 'درت': 'dert', 'درات': 'derat', 'درنا': 'derna',
+        'جا': 'ja', 'جات': 'jat', 'جينا': 'jina', 'جاو': 'jaou',
+        'مشا': 'mcha', 'مشات': 'mchat', 'مشينا': 'mchina', 'مشاو': 'mchaou',
+        
+        # Demonstratives
+        'هاد': 'had', 'هادا': 'hada', 'هادي': 'hadi', 'هادو': 'hadou',
+        'هادوك': 'hadouk', 'هادشي': 'hadchi', 'هاداك': 'hadak',
+        'داك': 'dak', 'ديك': 'dik', 'دوك': 'dok', 'داكشي': 'dakchi',
+        
+        # Numbers
+        'واحد': 'wahed', 'جوج': 'joj', 'تلاتا': 'tlata', 'ربعا': 'rebaa',
+        'خمسا': 'khamsa', 'ستا': 'setta', 'سبعا': 'sebaa', 'تمنيا': 'tmanya',
+        'تسعا': 'tesaa', 'عشرا': 'ashra',
+        
+        # Time
+        'دابا': 'daba', 'توا': 'tawa', 'غدا': 'ghda', 'البارح': 'lbareeh',
+        'نهار': 'nhar', 'ليلا': 'lila', 'صباح': 'sbah', 'عشيا': 'aachia',
+        
+        # Family
+        'ماما': 'mama', 'بابا': 'baba', 'خويا': 'khoya', 'ختي': 'khti',
+        'ختيا': 'khtiya', 'أخي': 'akhi', 'أختي': 'ukhti',
+        'جد': 'jad', 'جدا': 'jada', 'عمو': 'amo', 'عما': 'ama',
+        
+        # Adjectives
+        'زوين': 'zwin', 'زوينا': 'zwina', 'قبيح': 'qbih', 'قبيحا': 'qbiha',
+        'كبير': 'kbir', 'كبيرا': 'kbira', 'صغير': 'sghir', 'صغيرا': 'sghira',
+        'تقيل': 'tqil', 'تقيلا': 'tqila', 'خفيف': 'khfif', 'خفيفا': 'khfifa',
+        
+        # Quantities
+        'بزاف': 'bzzaf', 'كتير': 'ktir', 'شوياة': 'chwiya', 'كولشي': 'kolchi',
+        'كول': 'kol', 'غاع': 'gaa', 'حتا': 'hta', 'والو': 'walo',
+        
+        # Negations
+        'ما': 'ma', 'ماشي': 'machi', 'مابغيتش': 'mabghitsh', 'ماكاينش': 'makainch',
+        'ماكانش': 'makansh', 'ماغاتش': 'maghatch', 'مابغيش': 'mabaghish',
+        
+        # Expressions
+        'سلام': 'salam', 'يالاه': 'yallah', 'يالا': 'yalla', 'هيا': 'hya',
+        'أجي': 'aji', 'سير': 'sir', 'سيري': 'siri', 'أجو': 'ajou',
+        'مبروك': 'mabrook', 'باركا': 'baraka', 'بصحا': 'besaha',
+        'الله': 'allah', 'ربي': 'rabi', 'إن شاء الله': 'inchallah',
+        'ما شاء الله': 'machallah', 'الحمد لله': 'hamdulillah',
+        
+        # Places
+        'المغرب': 'lmaghrib', 'مغرب': 'maghrib', 'كازا': 'casa',
+        'الرباط': 'rbat', 'فاس': 'fas', 'مكناس': 'mknas',
+        'طنجا': 'tanja', 'أكادير': 'agadir', 'مراكش': 'marrakech',
+        
+        # Food and drinks
+        'طاجين': 'tajine', 'كوسكوس': 'couscous', 'مسمن': 'msemen',
+        'رغايف': 'rghaif', 'حريرا': 'harira', 'أتاي': 'atay',
+        'خوبز': 'khobz', 'الما': 'lma', 'لحم': 'lahem', 'دجاج': 'djaj',
+        
+        # Common nouns
+        'دار': 'dar', 'بيت': 'bit', 'طوموبيل': 'tomo', 'كرهوبا': 'karhouba',
+        'ماكلا': 'makla', 'خضرا': 'khodra', 'فواكه': 'fawakeh',
+        'حوت': 'hout', 'اللبن': 'lben', 'سكر': 'skar', 'ملح': 'melh'
+    }
     
-    # Default to English for Latin-only text without Arabizi indicators
-    return 'english'
-
-def create_system_prompt(response_lang):
-    """Create appropriate system prompt based on desired response language"""
+    # CHARACTER MAPPINGS (Arabic → Arabizi)
+    char_mappings = {
+        'ا': 'a', 'ب': 'b', 'ت': 't', 'ث': 'th', 'ج': 'j', 'ح': '7',
+        'خ': 'kh', 'د': 'd', 'ذ': 'dh', 'ر': 'r', 'ز': 'z', 'س': 's',
+        'ش': 'sh', 'ص': 's', 'ض': 'd', 'ط': '6', 'ظ': 'z', 'ع': '3',
+        'غ': 'gh', 'ف': 'f', 'ق': '9', 'ك': 'k', 'ل': 'l', 'م': 'm',
+        'ن': 'n', 'ه': 'h', 'و': 'w', 'ي': 'y', 'ء': '2',
+        'آ': 'aa', 'أ': 'a', 'إ': 'i', 'ئ': '2', 'ؤ': 'w2', 'ة': 'a',
+        'ى': 'a', 'ً': 'an', 'ٌ': 'on', 'ٍ': 'in', 'َ': 'a', 'ُ': 'o', 'ِ': 'i'
+    }
     
-    if response_lang == 'arabizi':
-        return """You are Atlas-Chat, an AI assistant specialized in Moroccan Arabic (Darija). 
-
-CRITICAL INSTRUCTION: The user has written in Arabizi (Latin script), so you MUST respond ONLY in Arabizi using Latin letters and numbers.
-
-ARABIZI RULES YOU MUST FOLLOW:
-- Use ONLY Latin letters (a-z) and numbers for Arabic sounds
-- Use these number substitutions: 3=ع, 7=ح, 9=ق, 2=ء, 5=خ, 6=ط, 8=غ
-- Write naturally in Moroccan Darija but with Latin script
-- Examples: "ana" (أنا), "hna" (حنا), "3la" (على), "7na" (حنا), "wach" (واش)
-- Do NOT use any Arabic script characters
-- Do NOT switch to English unless the user specifically asks for translation
-
-Respond naturally in Arabizi about Moroccan culture, language, and general topics."""
-
-    elif response_lang == 'arabic':
-        return """You are Atlas-Chat, an AI assistant specialized in Moroccan Arabic (Darija). Respond in Arabic script (Darija) as this is what the user is using. Be helpful and culturally aware about Morocco and its traditions."""
+    # Convert text
+    result = text
+    
+    # Step 1: Apply word mappings (most specific first)
+    for arabic_word, arabizi_word in word_mappings.items():
+        # Use word boundaries to avoid partial matches
+        pattern = r'\b' + re.escape(arabic_word) + r'\b'
+        result = re.sub(pattern, arabizi_word, result)
+    
+    # Step 2: Apply character mappings
+    for arabic_char, arabizi_char in char_mappings.items():
+        result = result.replace(arabic_char, arabizi_char)
     
-    else:  # English
-        return """You are Atlas-Chat, an AI assistant specialized in Moroccan Arabic (Darija) but also fluent in English. The user has written in English, so respond in English while being knowledgeable about Moroccan culture and language."""
+    return result
 
 def chat_with_atlas(message, history):
-    """Generate response from Atlas-Chat model with language detection"""
+    """Generate response from Atlas-Chat model with Arabizi conversion"""
     if not message.strip():
         return "مرحبا! أهلا وسهلا. Please enter a message! / Ahlan wa sahlan!"
     
@@ -150,29 +353,22 @@ def chat_with_atlas(message, history):
         # Load model if not already loaded
         model = load_model()
         
-        # Determine response language
-        response_lang = determine_response_language(message)
+        # Detect if input is Arabizi
+        is_arabizi_input = detect_arabizi(message)
         
-        # Create appropriate system prompt
-        system_prompt = create_system_prompt(response_lang)
-        
-        # Prepare messages with system context
-        if response_lang == 'arabizi':
-            # Extra emphasis for Arabizi responses
-            enhanced_message = f"""System: {system_prompt}
-
-User message (in Arabizi): {message}
-
-Remember: Respond ONLY in Arabizi (Latin letters + numbers). Do not use Arabic script."""
-            
-            messages = [{"role": "user", "content": enhanced_message}]
+        # Prepare input for the model
+        if is_arabizi_input:
+            # Convert Arabizi to Arabic for the model
+            arabic_input = arabizi_to_arabic(message)
+            print(f"🔄 Converted Arabizi '{message}' → Arabic '{arabic_input}'")
+            model_input = arabic_input
         else:
-            messages = [
-                {"role": "system", "content": system_prompt},
-                {"role": "user", "content": message}
-            ]
+            # Use original input (Arabic or English)
+            model_input = message
+        
+        # Generate response using Arabic input
+        messages = [{"role": "user", "content": model_input}]
         
-        # Generate response
         outputs = model(
             messages, 
             max_new_tokens=256, 
@@ -184,18 +380,21 @@ Remember: Respond ONLY in Arabizi (Latin letters + numbers). Do not use Arabic s
         # Extract the response
         response = outputs[0]["generated_text"][-1]["content"].strip()
         
-        # Post-process for Arabizi if needed
-        if response_lang == 'arabizi':
-            # Remove any Arabic script that might have leaked through
-            arabic_pattern = r'[\u0600-\u06FF\u0750-\u077F\u08A0-\u08FF\uFB50-\uFDFF\uFE70-\uFEFF]'
-            if re.search(arabic_pattern, response):
-                # If Arabic script is detected, provide a fallback Arabizi response
-                response = "ana Atlas-Chat, kay3jebni n7der m3ak! chno bghiti t3ref 3la lmaghrib? (I'm Atlas-Chat, I'd love to chat with you! What do you want to know about Morocco?)"
-        
-        return response
+        # Convert response back to Arabizi if input was Arabizi
+        if is_arabizi_input:
+            arabizi_response = arabic_to_arabizi(response)
+            print(f"🔄 Converted response '{response}' → Arabizi '{arabizi_response}'")
+            return arabizi_response
+        else:
+            # Return original response for Arabic/English
+            return response
         
     except Exception as e:
-        return f"عذراً، واجهت خطأ: {str(e)}. جرب مرة أخرى! / Sorry, error occurred. Try again!"
+        # Return error in appropriate language
+        if detect_arabizi(message):
+            return f"sorry, kan chi mochkil: {str(e)}. 3awd jar'b!"
+        else:
+            return f"عذراً، واجهت خطأ: {str(e)}. جرب مرة أخرى! / Sorry, error occurred: {str(e)}. Try again!"
 
 # Create the Gradio interface
 demo = gr.ChatInterface(
@@ -204,11 +403,12 @@ demo = gr.ChatInterface(
     description="""
     **مرحبا بك في أطلس شات!** Welcome to Atlas-Chat! 🇲🇦
     
-    I'm an AI assistant specialized in **Moroccan Arabic (Darija)** with smart language detection:
+    **🧠 Smart Language Detection & Conversion:**
+    - **Arabic Script (العربية)** → AI responds in Arabic
+    - **Arabizi (3arabi bi 7oruf latin)** → AI responds in Arabizi  
+    - **English** → AI responds in English
     
-    - **Arabic Script (العربية)** → I respond in Arabic
-    - **Arabizi (3arabi bi 7oruf latin)** → I respond in Arabizi  
-    - **English** → I respond in English
+    **✨ Full Arabizi Support:** Type "shkoun nta?" and get "ana atlas-chat..."
     
     **جرب هذه الأسئلة / Try these questions:**
     """,
@@ -222,7 +422,11 @@ demo = gr.ChatInterface(
         "What is Morocco famous for?",
         "Tell me about Casablanca",
         "كيفاش نقدر نتعلم الدارجة؟",
-        "kifash n9der nt3elem darija?"
+        "kifash n9der nt3elem darija?",
+        "wach kayn atay f lmaghrib?",
+        "3lach lmaghrib zwien bzzaf?",
+        "kifash nsali tajine?",
+        "chno homa l2aklat lmaghribiya?"
     ],
     cache_examples=False
 )