Update app.py

app.py

@@ -15,6 +15,9 @@ def isTrue(x) -> bool:
         return x
     return x.strip().lower() == 'true'
 
+if "feedback_key" not in st.session_state:
+        st.session_state.feedback_key = 0
+
 def launch_bot():
     def generate_response(question):
         response = vq.submit_query(question)
@@ -78,21 +81,13 @@ def launch_bot():
     if "messages" not in st.session_state.keys():
         st.session_state.messages = [{"role": "assistant", "content": "How may I help you?"}]
 
-    if "feedback_key" not in st.session_state:
-        st.session_state.feedback_key = 0
+    
                 
     # Display chat messages
     for message in st.session_state.messages:
         with st.chat_message(message["role"]):
             st.write(message["content"])
 
-        if (message["role"] == "assistant") & (not st.session_state.first_turn):
-            feedback = streamlit_feedback(feedback_type="thumbs", key = st.session_state.feedback_key)
-            print(f"DEBUG: Feedback key: {st.session_state.feedback_key}")
-            if feedback:
-                st.write(feedback)
-            st.session_state.feedback_key += 1
-
     example_container = st.empty()
     with example_container:
         if show_example_questions():
@@ -123,6 +118,12 @@ def launch_bot():
             message = {"role": "assistant", "content": response}
             st.session_state.messages.append(message)
             st.rerun()
+
+    feedback = streamlit_feedback(feedback_type="thumbs", key = st.session_state.feedback_key)
+    print(f"DEBUG: Feedback key: {st.session_state.feedback_key}")
+    # if feedback:
+    #     st.write(feedback)
+    st.session_state.feedback_key += 1
     
 if __name__ == "__main__":
     launch_bot()