start implementing csv agent

.gitignore

@@ -154,3 +154,4 @@ cython_debug/
 #venv
 
 *.pkl
+*.csv

src/chatbot_csv.py

@@ -3,12 +3,11 @@ import streamlit as st
 import asyncio
 from dotenv import load_dotenv
 
-
 from modules.history import ChatHistory
 from modules.layout import Layout
 from modules.utils import Utilities
 from modules.sidebar import Sidebar
-
+from csv_agent import extra_chatbot_page
 
 def init():
     load_dotenv()
@@ -16,6 +15,7 @@ def init():
 
 
 async def main():
+
     init()
     layout, sidebar, utils = Layout(), Sidebar(), Utilities()
     layout.show_header()
@@ -31,6 +31,8 @@ async def main():
             history = ChatHistory()
             sidebar.show_options()
 
+
+
             try:
                 chatbot = await utils.setup_chatbot(
                     uploaded_file, st.session_state["model"], st.session_state["temperature"]
@@ -56,9 +58,14 @@ async def main():
 
             except Exception as e:
                 st.error(f"Error: {str(e)}")
+        # Ajout du bouton pour naviguer vers la page du chatbot supplémentaire
+    if st.sidebar.button("Aller au chatbot supplémentaire"):
+        st.session_state["extra_chatbot"] = True
+        extra_chatbot_page()
 
     sidebar.about()
 
-
 if __name__ == "__main__":
+    if "page" not in st.session_state:
+        st.session_state.page = "main"
     asyncio.run(main())

src/csv_agent.py

@@ -0,0 +1,19 @@
+from langchain.agents import create_csv_agent
+import streamlit as st
+
+from langchain.llms import OpenAI
+
+
+def extra_chatbot_page():
+
+    query = st.text_input(label="Use CSV agent for precise informations about the csv file itself")
+
+    if query :
+        agent = create_csv_agent(OpenAI(temperature=0), 'poto-associations-sample.csv', verbose=True)
+        if agent :
+            st.write(agent.run(query))
+
+if __name__ == "__main__":
+    extra_chatbot_page()
+
+

src/modules/chatbot.py

@@ -5,8 +5,7 @@ from langchain.prompts.prompt import PromptTemplate
 
 
 class Chatbot:
-    _template = """Given the following conversation and a follow-up question, rephrase the follow-up question to be a stand-alone question.
-    You can assume that the question is about the information in a CSV file.
+    _template = """Given the following conversation and a follow-up question, rephrase the follow-up question to be a standalone question.
     Chat History:
     {chat_history}
     Follow-up entry: {question}
@@ -14,13 +13,12 @@ class Chatbot:
 
     CONDENSE_QUESTION_PROMPT = PromptTemplate.from_template(_template)
 
-    qa_template = """"You are an AI conversational assistant to answer questions based on information from a csv file.
+    qa_template = """"You are an AI conversational assistant to answer questions based on a context.
     You are given data from a csv file and a question, you must help the user find the information they need. 
-    Only give responses for information you know about. Don't try to make up an answer.
-    Your answers should be short,friendly, in the same language.
+    Your answers should be friendly, in the same language.
     question: {question}
     =========
-    {context}
+    context: {context}
     =======
     """
 
@@ -35,7 +33,6 @@ class Chatbot:
         """
         Starts a conversational chat with a model via Langchain
         """
-
         chain = ConversationalRetrievalChain.from_llm(
             llm=ChatOpenAI(model_name=self.model_name, temperature=self.temperature),
             condense_question_prompt=self.CONDENSE_QUESTION_PROMPT,

src/modules/layout.py

@@ -2,13 +2,14 @@ import streamlit as st
 
 
 class Layout:
+    
     def show_header(self):
         """
         Displays the header of the app
         """
         st.markdown(
             """
-            <h1 style='text-align: center;'>ChatBot-CSV, Talk with your  csv-data ! 💬</h1>
+            <h1 style='text-align: center;'>ChatBot-CSV, Talk with the content of your csv data! 💬</h1>
             """,
             unsafe_allow_html=True,
         )
@@ -38,5 +39,7 @@ class Layout:
                 label_visibility="collapsed",
             )
             submit_button = st.form_submit_button(label="Send")
+            
             is_ready = submit_button and user_input
         return is_ready, user_input
+    

src/modules/sidebar.py

@@ -2,7 +2,7 @@ import streamlit as st
 
 
 class Sidebar:
-    MODEL_OPTIONS = ["gpt-3.5-turbo"]
+    MODEL_OPTIONS = ["gpt-3.5-turbo", "gpt-4"]
     TEMPERATURE_MIN_VALUE = 0.0
     TEMPERATURE_MAX_VALUE = 1.0
     TEMPERATURE_DEFAULT_VALUE = 0.0
@@ -39,11 +39,14 @@ class Sidebar:
             step=self.TEMPERATURE_STEP,
         )
         st.session_state["temperature"] = temperature
-
+    
     def show_options(self):
         with st.sidebar.expander("🛠️ Settings", expanded=False):
+
             self.reset_chat_button()
             self.model_selector()
             self.temperature_slider()
             st.session_state.setdefault("model", self.MODEL_OPTIONS[0])
             st.session_state.setdefault("temperature", self.TEMPERATURE_DEFAULT_VALUE)
+    
+