Update utils/pdf_generator.py

utils/pdf_generator.py

@@ -31,10 +31,95 @@ try:
 except Exception as e:
     print(f"Error registering font: {str(e)}")
 
+
 def create_braille_pdf(original_text, braille_text, title="Menu in Braille"):
     """
     Create a PDF file with original text and its Braille translation.
     
+    Args:
+        original_text: Original text content
+        braille_text: Braille translation
+        title: PDF title
+        
+    Returns:
+        BytesIO object containing the PDF
+    """
+    try:
+        # Create a BytesIO object to store the PDF
+        buffer = io.BytesIO()
+        
+        # Create the PDF document
+        doc = SimpleDocTemplate(
+            buffer,
+            pagesize=letter,
+            rightMargin=72,
+            leftMargin=72,
+            topMargin=72,
+            bottomMargin=72
+        )
+        
+        # Define styles
+        styles = getSampleStyleSheet()
+        title_style = styles['Title']
+        heading_style = styles['Heading2']
+        normal_style = styles['Normal']
+        
+        # Create a custom style for Braille text
+        braille_style = ParagraphStyle(
+            'Braille',
+            parent=normal_style,
+            fontName='Helvetica',  # Use standard font to avoid issues
+            fontSize=14,
+            leading=18,
+            spaceAfter=12
+        )
+        
+        # Create the content
+        content = []
+        
+        # Add title
+        content.append(Paragraph(title, title_style))
+        content.append(Spacer(1, 12))
+        
+        # Add original text section
+        content.append(Paragraph("Original Text", heading_style))
+        content.append(Spacer(1, 6))
+        
+        # Split original text by lines and add each as a paragraph
+        for line in original_text.split('\n'):
+            if line.strip():
+                content.append(Paragraph(line.replace('&', '&amp;').replace('<', '&lt;').replace('>', '&gt;'), normal_style))
+            else:
+                content.append(Spacer(1, 12))
+        
+        content.append(Spacer(1, 24))
+        
+        # Add Braille section
+        content.append(Paragraph("Braille Translation", heading_style))
+        content.append(Spacer(1, 6))
+        
+        # Split Braille text by lines and add each as a paragraph
+        for line in braille_text.split('\n'):
+            if line.strip():
+                content.append(Paragraph(line.replace('&', '&amp;').replace('<', '&lt;').replace('>', '&gt;'), braille_style))
+            else:
+                content.append(Spacer(1, 12))
+        
+        # Build the PDF
+        doc.build(content)
+        
+        # Reset buffer position to the beginning
+        buffer.seek(0)
+        return buffer
+    except Exception as e:
+        print(f"Error in create_braille_pdf: {str(e)}")
+        raise
+
+
+def create_braille_pdf1(original_text, braille_text, title="Menu in Braille"):
+    """
+    Create a PDF file with original text and its Braille translation.
+    
     Args:
         original_text: Original text content
         braille_text: Braille translation