georad commited on
Commit
a56730d
·
verified ·
1 Parent(s): 4a784c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -26
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