Update app.py

app.py

@@ -1,21 +1,37 @@
-from googletrans import Translator
+import logging
+from transformers import MarianMTModel, MarianTokenizer
 
-# Initialize the translator
-translator = Translator()
+# Setup logging
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+
+# Initialize logger
+logger = logging.getLogger(__name__)
 
 # Define a function to translate text from Russian to English
 def translate_to_english(text):
-    # Translate the text
-    translation = translator.translate(text, src='ru', dest='en')
-    # Return the translated text
-    return translation.text
+    logger.info("Loading model and tokenizer")
+    model_name = 'Helsinki-NLP/opus-mt-ru-en'
+    tokenizer = MarianTokenizer.from_pretrained(model_name)
+    model = MarianMTModel.from_pretrained(model_name)
+
+    logger.info("Tokenizing text")
+    tokenized_text = tokenizer.prepare_seq2seq_batch([text], return_tensors='pt')
+
+    logger.info("Performing translation")
+    translation = model.generate(**tokenized_text)
+
+    logger.info("Decoding translation")
+    translated_text = tokenizer.decode(translation[0], skip_special_tokens=True)
+    return translated_text
 
 # Sample Russian text
 russian_text = "Привет, как дела?"
 
+logger.info("Starting translation")
 # Translate the text
 english_text = translate_to_english(russian_text)
 
+logger.info("Translation complete")
 # Print the translated text
 print(f"Original Russian text: {russian_text}")
 print(f"Translated English text: {english_text}")