/
hr_agent_frontend.py
41 lines (29 loc) · 1.25 KB
/
hr_agent_frontend.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
import streamlit as st
import random
from streamlit_chat import message
#from hr_agent_backend_azure import get_response
from hr_agent_backend_local import get_response
def process_input(user_input):
response = get_response(user_input)
return response
st.header("HR Chatbot")
st.markdown("Ask your HR-related questions here.")
if "past" not in st.session_state:
st.session_state["past"] = []
if "generated" not in st.session_state:
st.session_state["generated"] = []
if "input_message_key" not in st.session_state:
st.session_state["input_message_key"] = str(random.random())
chat_container = st.container()
user_input = st.text_input("Type your message and press Enter to send.", key=st.session_state["input_message_key"])
if st.button("Send"):
response = process_input(user_input)
st.session_state["past"].append(user_input)
st.session_state["generated"].append(response)
st.session_state["input_message_key"] = str(random.random())
st.experimental_rerun()
if st.session_state["generated"]:
with chat_container:
for i in range(len(st.session_state["generated"])):
message(st.session_state["past"][i], is_user=True, key=str(i) + "_user")
message(st.session_state["generated"][i], key=str(i))