Fix convert handling

1 files changed, 11 insertions, 0 deletions

diff --git a/engine/romaji.py b/engine/romaji.py
index 90437f1..34a1dde 100644
--- a/engine/romaji.py
+++ b/engine/romaji.py
@@ -56,6 +56,12 @@ class RomajiSegment(segment.Segment):
             self._jachars = jachars
             return [RomajiSegment(c)]
 
+        jachars, c = romaji_correction_rule.get(text, (None, None))
+        if jachars:
+            self._enchars = text[0]
+            self._jachars = jachars
+            return [RomajiSegment(c)]
+
         for i in range(-min(4, len(text)), 0):
             enchars = text[i:]
 
@@ -107,6 +113,11 @@ class RomajiSegment(segment.Segment):
             self._enchars = c
             return [RomajiSegment(text[0], jachars)]
 
+        jachars, c = romaji_correction_rule.get(text, (None, None))
+        if jachars:
+            self._enchars = c
+            return [RomajiSegment(text[0], jachars)]
+
         for i in range(min(4, len(text)), 0, -1):
             enchars = text[:i]