Update app.py

app.py

@@ -9,6 +9,7 @@ from data_process import *
 from langchain.tools.base import StructuredTool
 from langchain.agents import initialize_agent
 from qa_txt import llm
+from trans import trans
 import gradio as gr
 from langchain.agents import (
     create_react_agent,
@@ -28,13 +29,13 @@ def faq(query: str) -> str:
 qa_faq = StructuredTool.from_function(
     func = faq ,
     description="""
-    Respond to general questions about the website like the documentation, contact, utility, support... Don't use it when the user request data about a subject (economie, justice, water, or any type of public dataset) only for contact or useful links data. Used also for translating other languages in french.
+    Respond to general questions about the website like the documentation, contact, utility, support... Don't use it when the user request data about a subject (economie, justice, water, or any type of public dataset) only for contact or useful links data.
 
     Parameters :
     - query (string) : the same input as the user input no more no less and dont translate it even if it is in another language.
 
     Returns :
-    - string : the output as returned from the function in the language of the input if it is in french respond in french, if it is in arabic respond in arabic... just for general questions but for the translation the output must be in french .
+    - string : the output as returned from the function in the language of the input if it is in french respond in french, if it is in arabic respond in arabic...
     """
 )
 
@@ -62,9 +63,26 @@ fetch_data = StructuredTool.from_function(
     """,
 )
 
+def translate(query: str) -> str:
+    translated = trans.invoke({"input": query})['text']
+    return translated
+
+translate_text = StructuredTool.from_function(
+    func=translate,
+    description= """
+    Translate from any language to french.
+
+    Parameters :
+    - query (string) : the same input as the user input no more no less.
+    Returns :
+    - string : isolate just the translated text in french with no other useless words.
+    """,
+)
+
 tools_add = [
     qa_faq,
     fetch_data,
+    translate_text,
 ]
 
 agent = create_react_agent(llm=llm, tools=tools_add, prompt=prompt)