Update app.py

app.py

@@ -4,12 +4,13 @@ import torch
 from transformers import (
     MarianMTModel, MarianTokenizer,
     T5Tokenizer, T5ForConditionalGeneration,
-    pipeline
+    pipeline,
+    AutoModelForSeq2SeqLM,  # Changed from AutoModelForSeq2SeqLegacy
+    AutoTokenizer
 )
 import fitz  # PyMuPDF
 import docx2txt
 from fpdf import FPDF
-from transformers import AutoModelForSeq2SeqLegacy, AutoTokenizer
 import spacy
 import re
 
@@ -83,15 +84,20 @@ def save_as_pdf(text, output_path):
         pdf = FPDF()
         pdf.add_page()
         pdf.set_auto_page_break(auto=True, margin=15)
-        pdf.set_font("Arial", size=12)
+        pdf.add_font('DejaVu', '', '/usr/share/fonts/truetype/dejavu/DejaVuSansCondensed.ttf', uni=True)
+        pdf.set_font('DejaVu', size=12)
         
         # Split text into paragraphs
         paragraphs = text.split('\n\n')
         
         for para in paragraphs:
             # Add paragraph with spacing
-            pdf.multi_cell(0, 10, para.strip())
-            pdf.ln(5)  # Add some space between paragraphs
+            try:
+                pdf.multi_cell(0, 10, para.strip())
+                pdf.ln(5)  # Add some space between paragraphs
+            except Exception as e:
+                print(f"Error writing paragraph: {str(e)}")
+                continue
             
         pdf.output(output_path)
         return output_path