Update app.py

app.py

@@ -15,6 +15,21 @@ from pydub import AudioSegment
 import openai
 from reportlab.lib.pagesizes import A4
 from reportlab.pdfgen import canvas
+# ---------------- Translation Model ----------------
+from transformers import MarianTokenizer, MarianMTModel
+import torch
+
+trans_model_name = "Helsinki-NLP/opus-mt-ur-en"
+tokenizer_trans = MarianTokenizer.from_pretrained(trans_model_name)
+model_trans = MarianMTModel.from_pretrained(trans_model_name)
+
+def translate_ur_to_en(ur_text):
+    if not ur_text.strip():
+        return ""
+    batch = tokenizer_trans(ur_text, return_tensors="pt", padding=True)
+    generated_ids = model_trans.generate(**batch)
+    en_text = tokenizer_trans.batch_decode(generated_ids, skip_special_tokens=True)[0]
+    return en_text
 
 # ---------------- Config ----------------
 OPENAI_KEY = os.environ.get("OPENAI_API_KEY")
@@ -99,6 +114,8 @@ def build_system_prompt():
 
 def generate_letter_with_llm(raw_text: str, picked_authority: str, user_info: dict):
     raw_text = safe_text(raw_text)
+    # --- New Step: Translate Urdu text to English ---
+    raw_text_en = translate_ur_to_en(raw_text)
     if OPENAI_KEY:
         try:
             system_prompt = build_system_prompt()