first implementation of csv agent

src/chatbot_csv.py

@@ -1,11 +1,14 @@
 import os
 import streamlit as st
 from dotenv import load_dotenv
+from io import BytesIO
 
 from modules.history import ChatHistory
 from modules.layout import Layout
 from modules.utils import Utilities
 from modules.sidebar import Sidebar
+from langchain.agents import create_csv_agent
+from langchain.chat_models import ChatOpenAI
 
 def init():
     load_dotenv()
@@ -28,6 +31,8 @@ def main():
             history = ChatHistory()
             sidebar.show_options()
 
+            uploaded_file_content = BytesIO(uploaded_file.getvalue())
+
 
 
             try:
@@ -53,6 +58,11 @@ def main():
 
                     history.generate_messages(response_container)
 
+                if st.session_state["show_csv_agent"]:
+                    query = st.text_input(label="Use CSV agent for precise information about the CSV file itself")
+                    if query != "":
+                        agent = create_csv_agent(ChatOpenAI(temperature=0), uploaded_file_content, verbose=True)
+                        st.write(agent.run(query))
             except Exception as e:
                 st.error(f"Error: {str(e)}")
 

src/modules/sidebar.py

@@ -1,6 +1,5 @@
 import streamlit as st
-from langchain.agents import create_csv_agent
-from langchain.chat_models import ChatOpenAI
+
 
 class Sidebar:
     MODEL_OPTIONS = ["gpt-3.5-turbo", "gpt-4"]
@@ -44,9 +43,7 @@ class Sidebar:
     def csv_agent_button(self):
         st.session_state.setdefault("show_csv_agent", False)
         if st.sidebar.button("CSV Agent"):
-            st.session_state["show_csv_agent"] = True
-        else:
-            st.session_state["show_csv_agent"] = False
+            st.session_state["show_csv_agent"] = not st.session_state["show_csv_agent"]
 
     def show_options(self):
         with st.sidebar.expander("🛠️ Tools", expanded=False):

src/modules/utils.py

@@ -31,12 +31,14 @@ class Utilities:
         """
         uploaded_file = st.sidebar.file_uploader("upload", type="csv", label_visibility="collapsed")
         if uploaded_file is not None:
+            
 
             def show_user_file(uploaded_file):
                 file_container = st.expander("Your CSV file :")
                 shows = pd.read_csv(uploaded_file)
                 uploaded_file.seek(0)
                 file_container.write(shows)
+             
 
             show_user_file(uploaded_file)
         else: