Update app.py

app.py

@@ -110,26 +110,32 @@ def reverse_text_for_rtl(text):
     return ' '.join([word[::-1] for word in text.split()])
 
 # Helper function to write PDF documents
-def write_pdf(transcription, output_file, tokenizer=None, translation_model=None):
+def write_pdf(transcription, output_file, tokenizer=None, translation_model=None, target_language=None):
     # Create PDF with A4 page size
     c = canvas.Canvas(output_file, pagesize=A4)
+    
     # Get the directory where app.py is located
     app_dir = os.path.dirname(os.path.abspath(__file__))
 
-    # Construct the full path to the font file
-    font_path = os.path.join(app_dir, 'B-NAZANIN.TTF')
+    # Define font paths for different languages
+    fonts = {
+        'fa': os.path.join(app_dir, 'B-NAZANIN.TTF'),  # Persian Font
+        'ar': os.path.join(app_dir, 'Amiri-Regular.ttf'),  # Arabic Font
+        'default': 'Arial'  # Default font for other languages
+    }
 
+    # Register and set the appropriate font
+    font_path = fonts.get(target_language, fonts['default'])
+    
     if os.path.exists(font_path):
         try:
-            pdfmetrics.registerFont(TTFont('B-Nazanin', font_path))
-        except Exception as e:  # Changed to general exception to avoid missing imports
+            pdfmetrics.registerFont(TTFont('custom_font', font_path))
+            c.setFont('custom_font', 12)
+        except Exception as e:
             raise RuntimeError(f"Error registering font: {e}.")
     else:
         raise FileNotFoundError(f"Font file not found at {font_path}. Please ensure it is available.")
 
-    # Set font and size
-    c.setFont('B-Nazanin', 12)
-
     # Initialize y position from top of page
     y_position = A4[1] - 50  # Start 50 points from top
     line_height = 20
@@ -145,18 +151,24 @@ def write_pdf(transcription, output_file, tokenizer=None, translation_model=None
         # Format the line with segment number
         line = f"{i + 1}. {text.strip()}"
 
-        # Reshape and reorder the text for correct RTL display
-        reshaped_text = arabic_reshaper.reshape(line)
-        bidi_text = get_display(reshaped_text)
+        # For RTL languages like Persian and Arabic, reshape and reorder text
+        if target_language in ['fa', 'ar']:
+            reshaped_text = arabic_reshaper.reshape(line)
+            bidi_text = get_display(reshaped_text)
+        else:
+            bidi_text = line  # For LTR languages, no reshaping needed
 
         # Add new page if needed
         if y_position < 50:  # Leave 50 points margin at bottom
             c.showPage()
-            c.setFont('B-Nazanin', 12)
+            c.setFont('custom_font', 12)
             y_position = A4[1] - 50
 
-        # Draw the text right-aligned
-        c.drawRightString(A4[0] - 50, y_position, bidi_text)  # 50 points margin from right
+        # Draw the text right-aligned for RTL languages, otherwise left-aligned
+        if target_language in ['fa', 'ar']:
+            c.drawRightString(A4[0] - 50, y_position, bidi_text)  # Right align for RTL
+        else:
+            c.drawString(50, y_position, bidi_text)  # Left align for LTR
 
         # Update y position for next line
         y_position -= line_height
@@ -166,6 +178,7 @@ def write_pdf(transcription, output_file, tokenizer=None, translation_model=None
     return output_file
 
 
+
 # Helper function to write PowerPoint slides
 def write_ppt(transcription, output_file, tokenizer=None, translation_model=None):
     ppt = Presentation()