Update main.py

main.py

@@ -21,6 +21,7 @@ from deep_translator import GoogleTranslator
 from IPython.display import display
 
 from surveycaa import surveyCaa
+from csvanswer import LLMAnswer
 
 @cl.password_auth_callback
 def auth_callback(username: str, password: str):
@@ -39,27 +40,6 @@ def auth_callback(username: str, password: str):
             identifier=ident + " : 🧑‍🎓 User Datapcc", metadata={"role": "user", "provider": "credentials"}
         )
         
-def create_agent(filename: str):
-    """
-    Create an agent that can access and use a large language model (LLM).
-
-    Args:
-        filename: The path to the CSV file that contains the data.
-
-    Returns:
-        An agent that can access and use the LLM.
-    """
-
-    # Create an OpenAI object.
-    os.environ['OPENAI_API_KEY'] = os.environ['OPENAI_API_KEY']
-    llm = ChatOpenAI(temperature=0, model="gpt-4o-2024-05-13")
-
-    # Read the CSV file into a Pandas DataFrame.
-    df = pd.read_csv(filename)
-
-    # Create a Pandas DataFrame agent.
-    return create_csv_agent(llm, filename, verbose=False, allow_dangerous_code=True, handle_parsing_errors=True, agent_type=AgentType.OPENAI_FUNCTIONS)
-
 def query_agent(agent, query):
     """
     Query an agent and return the response as a string.
@@ -173,26 +153,4 @@ async def on_chat_start():
 @cl.on_message
 async def on_message(message: cl.Message):
     await cl.Message(f"> SURVEYIA").send()
-    agent = create_agent("./public/surveyia.csv")
-    cb = cl.AsyncLangchainCallbackHandler()
-    try:
-        res = await agent.acall("Réponds en langue française à la question suivante : " + message.content, callbacks=[cb])
-        await cl.Message(author="COPILOT",content=GoogleTranslator(source='auto', target='fr').translate(res['output'])).send()
-    except ValueError as e:
-        res = str(e)
-        resArray = res.split(":")
-        ans = ''
-        if str(res).find('parsing') != -1:
-            for i in range(2,len(resArray)):
-                ans += resArray[i]
-            await cl.Message(author="COPILOT",content=ans.replace("`","")).send()
-        else:
-            await cl.Message(author="COPILOT",content="Reformulez votre requête, s'il vous plait 😃").send()
-    # Query the agent.
-    #response = query_agent(agent=agent, query=message.content)
-    # Decode the response.
-    #decoded_response = decode_response(response)
-
-    # Write the response to the Streamlit app.
-    #result = write_response(decoded_response)   
-    #await cl.Message(author="COPILOT",content=GoogleTranslator(source='auto', target='fr').translate(result)).send()
+    await LLMAnswer()