Update translit.py

translit.py

@@ -94,12 +94,12 @@ arabic_to_english = {
     "ُ": "u",
     "ِ": "i",
     "،": ",",
-    "ֹ": "o",  # holam
-    "ַ": "a",  # patah
-    "ִ": "i",  # hiriq
+    "ֹ": "ᵒ",  # holam
+    "ַ": "ᵃ",  # patah
+    "ִ": "ⁱ",  # hiriq
     "ְ": "",   # shva
-    "ֻ": "u",  # kubutz
-    'ֵ': "e",
+    "ֻ": "ᵘ",  # kubutz
+    'ֵ': "ᵉ"   # tzere
     "ّ": "SHADDA"  # shadda
 }
 
@@ -155,6 +155,7 @@ def reverse_holam_shadda_vav(input_string):
     
     return result
 
+
 def to_taatik(arabic):
     taatik = []
     for index, letter in enumerate(arabic):
@@ -174,7 +175,23 @@ def to_taatik(arabic):
     return reordered
 
 
-
+def postprocess_arabic_transliteration(text):
+    # Step 1: Replace long vowels at the end of words
+    text = re.sub(r'([āīēūō])(\W*$|\W+)', lambda m: m.group(1).translate(str.maketrans('āīēūō', 'aieuo')) + m.group(2), text)
+    
+    # Step 2: Convert 'iy' to 'ī', but keep 'iyy'
+    text = re.sub(r'iy(?!y)', 'ī', text)
+    
+    # Step 3: Convert 'uw' to 'ū', but keep 'uww'
+    text = re.sub(r'uw(?!w)', 'ū', text)
+    
+    # Step 4: Convert 'ay' to 'ē', but keep 'ayy'
+    text = re.sub(r'ay(?!y)', 'ē', text)
+    
+    # Step 5: Convert 'aw' to 'ō', but keep 'aww'
+    text = re.sub(r'aw(?!w)', 'ō', text)
+    
+    return text
 
 def to_translit(arabic):
     translit = []
@@ -184,15 +201,16 @@ def to_translit(arabic):
         else:
             if arabic_to_english[letter] == "SHADDA":
                 if translit[-1][0] in vowels:
-                    translit[-2][1] = translit[-2][1].upper()
+                    #translit[-2][1] = translit[-2][1].upper()
+                    translit[-2][1] = translit[-2][1] + translit[-2][1]
                 else:
-                    translit[-1][1] = translit[-1][1].upper()
-    
+                    #translit[-1][1] = translit[-1][1].upper()
+                    translit[-1][1] = translit[-1][1] + translit[-1][1]
+
             else:
                 translit.append([letter, arabic_to_english[letter]])
             
-    return "".join([x[1] for x in translit])
-
+    return postprocess_arabic_transliteration("".join([x[1] for x in translit]))
 
 # %%