Update app.py
Browse files
app.py
CHANGED
@@ -94,7 +94,7 @@ def launch_bot():
|
|
94 |
"## Temporary test demo only\n"
|
95 |
)
|
96 |
|
97 |
-
st.markdown(f"<center> <h2> Header Demo Test: {cfg.title} </h2> </center>", unsafe_allow_html=True)
|
98 |
|
99 |
if "messages" not in st.session_state.keys():
|
100 |
reset()
|
@@ -121,35 +121,11 @@ def launch_bot():
|
|
121 |
st.write(prompt)
|
122 |
st.session_state.ex_prompt = None
|
123 |
|
124 |
-
# Generate a new response if last message is not from assistant
|
125 |
-
if st.session_state.messages[-1]["role"] != "assistant":
|
126 |
-
with st.chat_message("assistant", avatar="🤖"):
|
127 |
-
if cfg.streaming:
|
128 |
-
stream = generate_streaming_response(prompt)
|
129 |
-
response = st.write_stream(stream)
|
130 |
-
else:
|
131 |
-
with st.spinner("Thinking..."):
|
132 |
-
response = generate_response(prompt)
|
133 |
-
st.write(response)
|
134 |
-
|
135 |
-
response = escape_dollars_outside_latex(response)
|
136 |
-
message = {"role": "assistant", "content": response, "avatar": '🤖'}
|
137 |
-
st.session_state.messages.append(message)
|
138 |
-
|
139 |
-
# Send query and response to Amplitude Analytics
|
140 |
-
send_amplitude_data(
|
141 |
-
user_query=st.session_state.messages[-2]["content"],
|
142 |
-
chat_response=st.session_state.messages[-1]["content"],
|
143 |
-
demo_name=cfg.title,
|
144 |
-
language=st.session_state.language
|
145 |
-
)
|
146 |
-
st.rerun()
|
147 |
-
|
148 |
if (st.session_state.messages[-1]["role"] == "assistant") & (st.session_state.messages[-1]["content"] != "How may I help you?"):
|
149 |
streamlit_feedback(feedback_type="thumbs", on_submit = thumbs_feedback, key = st.session_state.feedback_key,
|
150 |
kwargs = {"user_query": st.session_state.messages[-2]["content"],
|
151 |
"chat_response": st.session_state.messages[-1]["content"],
|
152 |
-
"demo_name": cfg.title,
|
153 |
"response_language": st.session_state.language})
|
154 |
|
155 |
|
|
|
94 |
"## Temporary test demo only\n"
|
95 |
)
|
96 |
|
97 |
+
st.markdown(f"<center> <h2> Header Demo Test: {cfg.title()} </h2> </center>", unsafe_allow_html=True)
|
98 |
|
99 |
if "messages" not in st.session_state.keys():
|
100 |
reset()
|
|
|
121 |
st.write(prompt)
|
122 |
st.session_state.ex_prompt = None
|
123 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
124 |
if (st.session_state.messages[-1]["role"] == "assistant") & (st.session_state.messages[-1]["content"] != "How may I help you?"):
|
125 |
streamlit_feedback(feedback_type="thumbs", on_submit = thumbs_feedback, key = st.session_state.feedback_key,
|
126 |
kwargs = {"user_query": st.session_state.messages[-2]["content"],
|
127 |
"chat_response": st.session_state.messages[-1]["content"],
|
128 |
+
"demo_name": cfg.title(),
|
129 |
"response_language": st.session_state.language})
|
130 |
|
131 |
|