add

app.py

@@ -190,7 +190,19 @@ def main_app():
             # Ask for feedback
             st.markdown(st.session_state['analysis'])
 
-            st.markdown("Provide feedback on the report:")
+            # Ask the user to choose the type of feedback
+            feedback_type = st.selectbox(
+                "Select the type of feedback:",
+                ["Report", "Coach"]
+            )
+
+            # Set the dataset identifier based on feedback type
+            if feedback_type == "Report":
+                dataset_id = "wu981526092/feedback_report"
+            else:
+                dataset_id = "wu981526092/feedback_coach"
+
+            st.markdown(f"Provide feedback on the {feedback_type.lower()}:")
 
             criteria = {
                 "Faithfulness": "Are all claims made in the answer inferred from the given context, i.e., not hallucinated?",
@@ -203,12 +215,17 @@ def main_app():
 
             ratings = {}
             for criterion, description in criteria.items():
-                ratings[criterion] = st.slider(f"{criterion}: {description}", 0, 10, 5,key=f"coach {criterion}")
+                ratings[criterion] = st.slider(f"{criterion}: {description}", 0, 10, 5,key=f"{feedback_type} {criterion}")
 
-            feedback = st.text_input("Provide additional feedback on the response:",key=f"coach feedback")
+            feedback = st.text_input("Provide additional feedback on the response:",key=f"{feedback_type} feedback")
 
             if st.button('Submit Report Feedback'):
-                store_feedback(str(generate_prompt_from_profile(profile, version=st.session_state['version'])), st.session_state['analysis'], feedback, str(ratings), "wu981526092/feedback_report")
+                if feedback_type == "Report":
+                    store_feedback(str(generate_prompt_from_profile(profile, version=st.session_state['version'])), st.session_state['analysis'], feedback, str(ratings), dataset_id)
+                else:
+                    store_feedback(st.session_state['coach_query'], st.session_state['coach_response'], feedback, str(ratings), dataset_id)
+
+
                 st.success("Feedback submitted successfully!")
 
 
@@ -333,37 +350,13 @@ def sidebar_components():
 
             if st.sidebar.button('Submit'):
                 if user_input:
+                    st.session_state['coach_query'] = user_input
                     if st.session_state['chat_context'] == "profile":
                         chat_prompt_template = create_chat_prompt_template(get_profile_str(st.session_state['profile']),st.session_state['definition'])
                     else:
                         chat_prompt_template = create_chat_prompt_template(st.session_state['analysis'],st.session_state['definition'])
-                    response = execute_query(index, chat_prompt_template, user_input)
-
-                    st.sidebar.markdown(response)
-
-                    st.markdown("Provide feedback on the response:")
-
-                    # Feedback criteria
-                    st.sidebar.markdown("## Feedback Criteria")
-                    criteria = {
-                        "Faithfulness": "Are all claims made in the answer inferred from the given context, i.e., not hallucinated?",
-                        "Answer Relevancy": "Is the answer relevant to the question?",
-                        "Context Relevancy": "Is the context relevant to the question?",
-                        "Correctness": "Is the answer factually correct, based on the context?",
-                        "Clarity": "Is the answer explained clearly without the extensive jargon of the original document?",
-                        "Completeness": "Is the question answered fully, with all parts and subquestions being addressed?",
-                    }
-                    ratings = {}
-                    for criterion, description in criteria.items():
-                        ratings[criterion] = st.sidebar.slider(f"{criterion}: {description}", 0, 10, 5,key=f"report {criterion}")
-
-                    # Ask for feedback
-                    feedback = st.sidebar.text_input("Provide additional feedback on the response:",key=f"report feedback")
-
-                    if st.sidebar.button('Submit Coach Feedback'):
-                        store_feedback(user_input, response, feedback, str(ratings), "wu981526092/feedback_coach")
-                        time.sleep(10)
-                        st.sidebar.success("Feedback submitted successfully!")
+                    st.session_state['coach_response'] = execute_query(index, chat_prompt_template, user_input)
+                    st.sidebar.markdown(st.session_state['coach_response'])
 
 
 
@@ -378,7 +371,9 @@ session_defaults = {
     'version': "",
     'username': '',
     'password': '',
-    'authenticated': False
+    'authenticated': False,
+    'coach_response':"",
+    'coach_query':""
 }
 
 for key, default in session_defaults.items():
@@ -390,38 +385,3 @@ if st.session_state['authenticated']:
     main_app()
 else:
     login_page()
-
-# # Display the sidebar components based on the state
-# if 'show_chat' not in st.session_state:
-#     st.session_state['show_chat'] = None
-#
-# if 'definition' not in st.session_state:
-#     st.session_state['definition'] = 1
-#
-# if 'chat_context' not in st.session_state:
-#     st.session_state['chat_context'] = "analysis"
-#
-# if 'profile' not in st.session_state:
-#     st.session_state['profile'] = None
-#
-# if 'analysis' not in st.session_state:
-#     st.session_state['analysis'] = None
-#
-# if 'temperature' not in st.session_state:
-#     st.session_state['temperature'] = 0
-#
-# if 'version' not in st.session_state:
-#     st.session_state['version'] = ""
-#
-# # Initialize session state for username, password, and authentication
-# if 'username' not in st.session_state:
-#     st.session_state['username'] = ''
-# if 'password' not in st.session_state:
-#     st.session_state['password'] = ''
-# if 'authenticated' not in st.session_state:
-#     st.session_state['authenticated'] = False
-# # Show login or main app based on authentication
-# if st.session_state['authenticated']:
-#     main_app()
-# else:
-#     login_page()