add

app.py

@@ -1,13 +1,22 @@
+import pandas as pd
 import streamlit as st
 import os
 import json
 from openai import AzureOpenAI
 from model import create_models, configure_settings, load_documents_and_create_index, \
     create_chat_prompt_template, execute_query
+from datasets import Dataset, DatasetDict, load_dataset, concatenate_datasets
 
 client = AzureOpenAI(azure_endpoint="https://personality-service.openai.azure.com/",
                      api_key=os.getenv("AZURE_OPENAI_KEY"), api_version="2024-02-15-preview")
 
+TOKEN = os.getenv('hf_token')
+def store_feedback(user_input, response, feedback, rating,repo):
+    dataset = load_dataset(repo, use_auth_token=TOKEN, download_mode="force_redownload", ignore_verifications=True)
+    new_entry = pd.DataFrame({"user_input": [user_input], "response": [response], "feedback": [feedback], "rating": [rating]})
+    new_dataset = Dataset.from_pandas(new_entry)
+    updated_dataset = concatenate_datasets([dataset["train"], new_dataset])
+    updated_dataset.push_to_hub(repo, private=False, use_auth_token=TOKEN)
 
 # Function to generate a completion using OpenAI API
 def generate_one_completion(message, temperature):
@@ -171,6 +180,15 @@ def main_app():
 
             st.session_state['analysis'] = response
             st.session_state['show_chat'] = True
+
+            # Ask for feedback
+            feedback = st.text_input("Provide your feedback on the response:")
+            rating = st.slider("Rate the response:", 0, 10, 5)
+
+            if st.sidebar.button('Submit Feedback'):
+                store_feedback(str(prompt), response, feedback, rating, "wu981526092/feedback_report")
+                st.sidebar.success("Feedback submitted successfully!")
+
             st.rerun()
 
         # display the response
@@ -261,8 +279,8 @@ def sidebar_components():
             #     st.session_state['profile'] = json.load(uploaded_file)
         else:
             st.sidebar.title("Chat with Our Career Advisor")
-            st.sidebar.markdown(
-                "Hello, we hope you learned something about yourself in this report. This chat is here so you can ask any questions you have about your report!  It’s also a great tool to get ideas about how you can use the information in your report for your personal development and achieving your current goals.")
+            #st.sidebar.markdown(
+                #"Hello, we hope you learned something about yourself in this report. This chat is here so you can ask any questions you have about your report!  It’s also a great tool to get ideas about how you can use the information in your report for your personal development and achieving your current goals.")
 
             # Name to be included in the questions
             # name = st.session_state['profile']['bio_information'].get('Name', 'the individual')
@@ -306,8 +324,19 @@ def sidebar_components():
                         chat_prompt_template = create_chat_prompt_template(st.session_state['analysis'],st.session_state['definition'])
                     response = execute_query(index, chat_prompt_template, user_input)
 
+                    # Display the response
                     st.sidebar.markdown(response)
 
+                    # Ask for feedback
+                    feedback = st.sidebar.text_input("Provide your feedback on the response:")
+                    rating = st.sidebar.slider("Rate the response:", 0, 10, 5)
+
+                    if st.sidebar.button('Submit Feedback'):
+                        store_feedback(user_input, response, feedback, rating, "wu981526092/feedback_report")
+                        st.sidebar.success("Feedback submitted successfully!")
+
+
+
 # Initialize session state variables with default values if not already set
 session_defaults = {
     'show_chat': None,