Update app.py

app.py

@@ -18,7 +18,7 @@ translation_model = M2M100ForConditionalGeneration.from_pretrained(model_name)
 
 
 @tool
-def agent_translate(text: str) -> str:
+def agent_translate(text: str, detected_lang: str) -> str:
     """A tool that translate text with next logic:
         1) If text starts with 'PL:', translate the rest to Polish.
         2) If detected language == 'ru', translate to English.
@@ -26,13 +26,13 @@ def agent_translate(text: str) -> str:
         and return translated text
     Args:
         text: text in russian/english or polish languages.
+        detected_lang: detected language of the text
     """
     if text.startswith("PL:"):
         original = text[3:].strip()
         target_lang = "PL"
     else:
-        detected_lang = detect(text)
-        if detected_lang == "ru":
+        if detected_lang.lower() == "ru":
             target_lang = "EN"
             original = text
         else:
@@ -40,7 +40,6 @@ def agent_translate(text: str) -> str:
             original = text
 
     try:
-       
         inputs = tokenizer(original, return_tensors="pt")
 
         forced_bos_token_id = tokenizer.get_lang_id(target_lang.lower())
@@ -56,6 +55,7 @@ def agent_translate(text: str) -> str:
         return f"Error during translating: { str(e)}"
 
 
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.