Update translator.py

translator.py

@@ -15,24 +15,26 @@ class TextTranslator(BaseModel):
 output_parser = PydanticOutputParser(pydantic_object=TextTranslator)
 format_instructions = output_parser.get_format_instructions()
 
-def text_translator(input_text : str, language : str) -> str:
-    human_template = """Enter the text that you want to translate: 
-                {input_text}, and enter the language that you want it to translate to {language}. {format_instructions}"""
-    human_message_prompt = HumanMessagePromptTemplate.from_template(human_template)
-    
-    chat_prompt = ChatPromptTemplate.from_messages([human_message_prompt])
-    
-    prompt = chat_prompt.format_prompt(input_text = input_text, language = language, format_instructions = format_instructions)
-    
-    messages = prompt.to_messages()
-    
-    response = chat(messages = messages)
-    
-    output = output_parser.parse(response.content)
-    
-    output_text = output.output
-    
-    return output_text
+def text_translator(input_text: str, language: str) -> str:
+    try:
+        human_template = """Enter the text that you want to translate: 
+                    {input_text}, and enter the language that you want it to translate to {language}. {format_instructions}"""
+        human_message_prompt = HumanMessagePromptTemplate.from_template(human_template)
+
+        chat_prompt = ChatPromptTemplate.from_messages([human_message_prompt])
+        prompt = chat_prompt.format_prompt(
+            input_text=input_text,
+            language=language,
+            format_instructions=format_instructions
+        )
+        messages = prompt.to_messages()
+
+        response = chat(messages=messages)
+        output = output_parser.parse(response.content)
+        return output.output
+    except Exception as e:
+        return f"❌ Error: {str(e)}"
+
 
 def text_translator_ui():
     with gr.Column() as translator_ui: