chatCSV / src /modules /sidebar.py
niro
txt gpt
62c16a7
Raw
History Blame Contribute Delete
3.06 kB
import streamlit as st
class Sidebar:
MODEL_OPTIONS = ["gpt-4", "gpt-3.5-turbo"]
TEMPERATURE_MIN_VALUE = 0.0
TEMPERATURE_MAX_VALUE = 1.0
TEMPERATURE_DEFAULT_VALUE = 0.0
TEMPERATURE_STEP = 0.01
@staticmethod
def show_logo(path):
st.sidebar.image(path)
@staticmethod
def about():
about = st.sidebar.expander("About 🤖")
sections = [
"#### ChatBot-CSV is an AI chatbot featuring conversational memory, designed to enable users to discuss their CSV data in a more intuitive manner. 📄",
"#### He employs large language models to provide users with seamless, context-aware natural language interactions for a better understanding of their CSV data. 🌐",
"#### For more information on the differences and specifics between ChatBot and CSVagent, check out the read-me on GitHub 🙌 ",
# "#### Powered by [Langchain](https://github.com/hwchase17/langchain), [OpenAI](https://platform.openai.com/docs/models/gpt-3-5) and [Streamlit](https://github.com/streamlit/streamlit) ⚡",
# "#### Source code : [yvann-hub/ChatBot-CSV](https://github.com/yvann-hub/ChatBot-CSV)",
]
for section in sections:
about.write(section)
@staticmethod
def reset_chat_button():
if st.button("Reset chat"):
st.session_state["reset_chat"] = True
st.session_state.setdefault("reset_chat", False)
def model_selector(self):
model = st.selectbox(label="Model", options=self.MODEL_OPTIONS)
st.session_state["model"] = model
def temperature_slider(self):
temperature = st.slider(
label="Temperature",
min_value=self.TEMPERATURE_MIN_VALUE,
max_value=self.TEMPERATURE_MAX_VALUE,
value=self.TEMPERATURE_DEFAULT_VALUE,
step=self.TEMPERATURE_STEP,
)
st.session_state["temperature"] = temperature
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"] = not st.session_state["show_csv_agent"]
def show_options(self):
with st.sidebar.expander("🛠️ Tools", expanded=False):
self.reset_chat_button()
self.csv_agent_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)
""" def csv_agent(self, ):
# Ajout du bouton pour naviguer vers la page du chatbot supplémentaire
if csv_agent_button = st.sidebar.button("CSV Agent"):
query = st.text_input(label="Use CSV agent for precise informations about the csv file itself")
if query != "" :
agent = create_csv_agent(ChatOpenAI(temperature=0), 'poto-associations-sample.csv', verbose=True)
st.write(agent.run(query))"""