Update modules/translator.py

modules/translator.py

@@ -2,7 +2,7 @@ import os
 import gradio as gr
 from pydantic import BaseModel, Field
 from langchain.prompts import HumanMessagePromptTemplate, ChatPromptTemplate
-from langchain.output_parsers import StructuredOutputParser
+from langchain.output_parsers import PydanticOutputParser
 from langchain_openai import ChatOpenAI
 
 chat = ChatOpenAI()
@@ -11,8 +11,8 @@ chat = ChatOpenAI()
 class TextTranslator(BaseModel):
     output: str = Field(description="Python string containing the output text translated in the desired language")
 
-# Updated: Use StructuredOutputParser with Pydantic v2
-output_parser = StructuredOutputParser(pydantic_object=TextTranslator)
+# Use PydanticOutputParser (no need for response_schemas)
+output_parser = PydanticOutputParser(pydantic_object=TextTranslator)
 
 def text_translator(input_text: str, language: str) -> str:
     human_template = """Enter the text that you want to translate: 
@@ -23,7 +23,7 @@ def text_translator(input_text: str, language: str) -> str:
     messages = prompt.to_messages()
     response = chat(messages=messages)
     
-    # Use model_dump() for Pydantic v2
+    # Use output_parser to parse the response
     output = output_parser.parse(response.content)
     return output.output