Spaces:
Running
Running
File size: 1,995 Bytes
bf12aca |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 |
import os
import streamlit as st
from streamlit_chat import message
class ChatHistory:
def __init__(self):
self.history = st.session_state.get("history", [])
st.session_state["history"] = self.history
def default_greeting(self):
return "Hey Robby ! 👋"
def default_prompt(self, topic):
return f"Hello ! Ask me anything about {topic} 🤗"
def initialize_user_history(self):
st.session_state["user"] = [self.default_greeting()]
def initialize_assistant_history(self, uploaded_file):
st.session_state["assistant"] = [self.default_prompt(uploaded_file.name)]
def initialize(self, uploaded_file):
if "assistant" not in st.session_state:
self.initialize_assistant_history(uploaded_file)
if "user" not in st.session_state:
self.initialize_user_history()
def reset(self, uploaded_file):
st.session_state["history"] = []
self.initialize_user_history()
self.initialize_assistant_history(uploaded_file)
st.session_state["reset_chat"] = False
def append(self, mode, message):
st.session_state[mode].append(message)
def generate_messages(self, container):
if st.session_state["assistant"]:
with container:
for i in range(len(st.session_state["assistant"])):
message(
st.session_state["user"][i],
is_user=True,
key=f"history_{i}_user",
avatar_style="big-smile",
)
message(st.session_state["assistant"][i], key=str(i), avatar_style="thumbs")
def load(self):
if os.path.exists(self.history_file):
with open(self.history_file, "r") as f:
self.history = f.read().splitlines()
def save(self):
with open(self.history_file, "w") as f:
f.write("\n".join(self.history))
|