-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
50 lines (37 loc) · 1.32 KB
/
app.py
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
import streamlit as st
from streamlit_chat import message
import os
import pickle
from langchain.schema import (
SystemMessage,
HumanMessage,
AIMessage
)
from module.chatbot import get_response
def main():
# initialize message history
if "messages" not in st.session_state:
st.session_state.messages = [
SystemMessage(content="You are a helpful assistant.")
]
st.header("Your own PlantEra bot 🤖")
# sidebar with user input
with st.sidebar:
user_input = st.text_input("Your message: ", key="user_input")
# handle user input
if user_input:
#print(HumanMessage(content=user_input).content)
st.session_state.messages.append(HumanMessage(content=user_input).content)
with st.spinner("Thinking..."):
response = get_response(st.session_state.messages[-1])
st.session_state.messages.append(AIMessage(content=response))
# display message history
messages = st.session_state.get('messages', [])
#print(messages)
for i, msg in enumerate(messages[1:]):
if i % 2 == 0:
message(msg, is_user=True, key=str(i) + '_user')
else:
message(msg.content, is_user=False, key=str(i) + '_ai')
if __name__ == '__main__':
main()