Add Feedback Button that uses the st.feedback module

app.py

@@ -70,6 +70,27 @@ def clear_chat_history():
 
 st.button('Clear Chat', on_click=clear_chat_history)
 
+def feedback():
+    with st.container():
+        sentiment_mapping = [":material/thumb_down:", ":material/thumb_up:"]
+        rating = st.feedback()
+        # feedback = ""
+        if rating is not None:
+            st.markdown(f"You selected: {sentiment_mapping[rating]}")
+
+        feedback = st.text_input("Please detail your rationale for choosing {sentiment_mapping[selected]}: ", "")
+        # rating = st.radio("Rate your experience:", ["👍", "Neutral", "👎"])
+        review = {"rating": {rating}, "feedback": {feedback}}
+        st.markdown(review)
+        #     # Save the feedback data
+        # if st.button("Submit"):
+        #     with open("feedback.json", "a") as f:
+        #         f.write()
+        #     st.write("Thank you for your feedback!")
+
+
+st.button('Give Feedback on Last Response', on_click=feedback)
+
 # build our chain outside the working body so that its only instantiated once - simply pass it the chat history for chat completion
 chain = ChainBuilder().build_chain()
 
@@ -211,16 +232,6 @@ def handle_user_input(user_input):
         
         st.session_state["messages"].append({"role": "assistant", "content": response, "warning": stream_warning, "error": stream_error})
 
-# def feedback():
-#     feedback = st.text_input("Please enter your feedback:")
-#     rating = st.radio("Rate your experience:", ["👍", "Neutral", "👎"])
-#     review = {"rating": {rating}, "feedback": {feedback}}
-#         # Save the feedback data
-#     if st.button("Submit"):
-#         with open("feedback.json", "a") as f:
-#             f.write()
-#         st.write("Thank you for your feedback!")
-
 main = st.container()
 with main:
     history = st.container(height=400)
@@ -235,7 +246,6 @@ with main:
                     # receive feedback on AI outputs if the user feels inclined to give it
                     # rating = st.radio("Rate your experience:", ["Very satisfied", "Somewhat satisfied", "Neutral", "Somewhat dissatisfied", "Very dissatisfied"])
                     # st.button("Provide Feedback", on_click=feedback)
-                    st.button("Provide Feedback", key="feedback")
                 if message["error"] is not None:
                     st.error(message["error"],icon="🚨")
                 if message["warning"] is not None: