Update models/braille_translator.py

models/braille_translator.py

@@ -53,10 +53,88 @@ def text_to_grade1_braille(text):
     
     return braille_text
 
+
 def text_to_braille(text, use_context=True):
     """
     Convert text to Braille, with optional context enhancement.
     
+    Args:
+        text: Text to convert to Braille
+        use_context: Whether to use AI to enhance context understanding
+        
+    Returns:
+        Dictionary with Braille text and metadata
+    """
+    try:
+        # Basic Braille translation
+        braille_text = text_to_grade1_braille(text)
+        
+        # Create an ASCII representation for PDF
+        ascii_braille = unicode_braille_to_ascii(braille_text)
+        
+        # If context enhancement is enabled
+        context_summary = None
+        if use_context and len(text) > 200:  # Only for longer texts
+            summarizer = get_summarizer()
+            if summarizer:
+                try:
+                    # Generate a summary to understand context
+                    summary_result = summarizer(text)
+                    if summary_result and len(summary_result) > 0:
+                        context_summary = summary_result[0]['summary_text']
+                except Exception as e:
+                    print(f"Summarization error: {str(e)}")
+        
+        # Format the Braille text for better readability
+        formatted_braille = format_braille_text(braille_text)
+        formatted_ascii = format_braille_text(ascii_braille)
+        
+        return {
+            'braille_text': braille_text,
+            'formatted_braille': formatted_braille,
+            'ascii_braille': ascii_braille,
+            'formatted_ascii': formatted_ascii,
+            'context_summary': context_summary,
+            'success': True
+        }
+    except Exception as e:
+        return {
+            'braille_text': '',
+            'error': str(e),
+            'success': False
+        }
+
+def unicode_braille_to_ascii(braille_text):
+    """
+    Convert Unicode Braille to ASCII representation.
+    
+    Args:
+        braille_text: Text with Unicode Braille characters
+        
+    Returns:
+        ASCII representation of Braille
+    """
+    # Map each Braille character to a descriptive ASCII representation
+    result = ""
+    for char in braille_text:
+        if char in BRAILLE_MAP.values():
+            # For Braille characters, use a letter representation
+            for letter, braille in BRAILLE_MAP.items():
+                if braille == char and len(letter) == 1:
+                    result += f"[{letter.upper()}]"
+                    break
+            else:
+                result += "[?]"
+        else:
+            # For non-Braille characters, keep them as is
+            result += char
+    
+    return result
+
+def text_to_braille1(text, use_context=True):
+    """
+    Convert text to Braille, with optional context enhancement.
+    
     Args:
         text: Text to convert to Braille
         use_context: Whether to use AI to enhance context understanding