Update app.py

app.py

@@ -60,13 +60,20 @@ LANGUAGES = {
 # Function to get the appropriate translation model and tokenizer
 def get_translation_model(source_lang, target_lang):
     model_name = f"Helsinki-NLP/opus-mt-{source_lang}-{target_lang}"
-    model = MarianMTModel.from_pretrained(model_name)
-    tokenizer = MarianTokenizer.from_pretrained(model_name)
-    return model, tokenizer
+    try:
+        model = MarianMTModel.from_pretrained(model_name)
+        tokenizer = MarianTokenizer.from_pretrained(model_name)
+        print(f"Loaded translation model for {source_lang} to {target_lang}")
+        return model, tokenizer
+    except Exception as e:
+        print(f"Error loading translation model for {source_lang} to {target_lang}: {e}")
+        return None, None
 
 # Function to translate text
 def translate_text(text, source_lang, target_lang):
     model, tokenizer = get_translation_model(source_lang, target_lang)
+    if model is None or tokenizer is None:
+        return "Translation model error."
     inputs = tokenizer([text], return_tensors="pt", truncation=True)
     translated_ids = model.generate(inputs['input_ids'], max_length=1024)
     translated_text = tokenizer.decode(translated_ids[0], skip_special_tokens=True)
@@ -85,12 +92,14 @@ def summarize_text(text, target_language="English"):
         early_stopping=True
     )
     summary = multilingual_summarization_tokenizer.decode(summary_ids[0], skip_special_tokens=True)
+    print(f"Generated summary in English: {summary}")
     
-    target_lang_code = LANGUAGES[target_language]
+    target_lang_code = LANGUAGES.get(target_language, "en_XX")
     
     # Translate summary to the target language if needed
     if target_lang_code != "en_XX":
         summary = translate_text(summary, "en_XX", target_lang_code)
+        print(f"Translated summary to {target_language}: {summary}")
     
     return summary