Update convert.py

convert.py

@@ -12,13 +12,90 @@ def convert_to_braille_unicode(str_input: str, path: str = "./braille_map.json")
     return str_output
 
 
+def braille_to_text(braille_unicode: str) -> str:
+    """
+    Convert braille unicode characters to readable text
+    Based on standard English Braille Grade 1 alphabet
+    """
+    # Braille to text mapping for basic alphabet
+    braille_to_text_map = {
+        '⠁': 'a', '⠃': 'b', '⠉': 'c', '⠙': 'd', '⠑': 'e',
+        '⠋': 'f', '⠛': 'g', '⠓': 'h', '⠊': 'i', '⠚': 'j',
+        '⠅': 'k', '⠇': 'l', '⠍': 'm', '⠝': 'n', '⠕': 'o',
+        '⠏': 'p', '⠟': 'q', '⠗': 'r', '⠎': 's', '⠞': 't',
+        '⠥': 'u', '⠧': 'v', '⠺': 'w', '⠭': 'x', '⠽': 'y',
+        '⠵': 'z',
+        
+        # Numbers (preceded by number sign ⠼)
+        '⠁': '1', '⠃': '2', '⠉': '3', '⠙': '4', '⠑': '5',
+        '⠋': '6', '⠛': '7', '⠓': '8', '⠊': '9', '⠚': '0',
+        
+        # Common punctuation
+        '⠀': ' ',  # space
+        '⠲': '.', '⠂': ',', '⠖': ';', '⠒': ':', '⠦': '?',
+        '⠖': '!', '⠄': "'", '⠤': '-', '⠌': '/', '⠐⠂': '"',
+        '⠣': '(', '⠜': ')', 
+        
+        # Special indicators
+        '⠼': '#',  # number indicator
+        '⠠': '',   # capital indicator (we'll handle this specially)
+    }
+    
+    result = ""
+    i = 0
+    number_mode = False
+    capitalize_next = False
+    
+    while i < len(braille_unicode):
+        char = braille_unicode[i]
+        
+        # Handle number indicator
+        if char == '⠼':
+            number_mode = True
+            i += 1
+            continue
+            
+        # Handle capital indicator
+        if char == '⠠':
+            capitalize_next = True
+            i += 1
+            continue
+        
+        # Convert current character
+        if char in braille_to_text_map:
+            if number_mode and char in '⠁⠃⠉⠙⠑⠋⠛⠓⠊⠚':
+                # Convert to numbers when in number mode
+                number_map = {
+                    '⠁': '1', '⠃': '2', '⠉': '3', '⠙': '4', '⠑': '5',
+                    '⠋': '6', '⠛': '7', '⠓': '8', '⠊': '9', '⠚': '0'
+                }
+                result += number_map[char]
+            else:
+                text_char = braille_to_text_map[char]
+                if capitalize_next and text_char.isalpha():
+                    result += text_char.upper()
+                    capitalize_next = False
+                else:
+                    result += text_char
+                    
+                # Reset number mode after non-number character
+                if char == '⠀' or not char in '⠁⠃⠉⠙⠑⠋⠛⠓⠊⠚':
+                    number_mode = False
+        else:
+            # Unknown braille character, keep as is or use placeholder
+            result += char
+            
+        i += 1
+    
+    return result
+
+
 def parse_xywh_and_class(boxes: torch.Tensor) -> list:
     """
     boxes input tensor
         boxes (torch.Tensor) or (numpy.ndarray): A tensor or numpy array containing the detection boxes,
             with shape (num_boxes, 6).
         orig_shape (torch.Tensor) or (numpy.ndarray): Original image size, in the format (height, width).
-
     Properties:
         xyxy (torch.Tensor) or (numpy.ndarray): The boxes in xyxy format.
         conf (torch.Tensor) or (numpy.ndarray): The confidence values of the boxes.
@@ -50,4 +127,4 @@ def parse_xywh_and_class(boxes: torch.Tensor) -> list:
         cluster = cluster[cluster[:, 0].argsort()]
         boxes_return.append(cluster)
 
-    return boxes_return
+    return boxes_return