add feedback prompt

app_web.py

@@ -65,6 +65,37 @@ def init_session():
         st.session_state.stata_file = None
     if "chat_id" not in st.session_state:
         st.session_state.chat_id = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    if "feddback_data" not in st.session_state:
+        st.session_state.feddback_data = None
+
+@st.dialog("Provide Your Feedback")
+def feedback_prompt():
+    st.write("How would you rate your WEAIBot experience?")
+    
+    # Built-in Streamlit feedback widget (thumbs or stars)
+    rating = st.feedback("faces")  # or st.feedback("stars")
+    
+    # Optional text details
+    comments = st.text_area("Tell us more (optional):")
+    feedback_data =  {
+            "rating": rating,
+            "comments": comments,
+        }
+    col1, col2 = st.columns(2)
+
+    with col1:
+        if st.button("Submit Feedback", type="primary", use_container_width=True):
+            # Save feedback to session state or database
+            st.session_state["feedback_data"] = feedback_data
+            st.toast("Thank you for your feedback!", icon="🎉")
+
+            with open(f"{DATA_DIR}/logs/{st.session_state.chat_id}/feedback.json", "w", encoding="utf-8") as f:
+                json.dump(feedback_data, f, indent=4)
+
+            st.rerun()
+    with col2:
+        if st.button("Cancel", use_container_width=True):
+            st.rerun()
 
 
 # ── Sidebar ──────────────────────────────────────────────────────────────────
@@ -290,7 +321,7 @@ def process_contact_request(original_query:str, qcat_result:Dict, uploaded_files
     #save ticket
     with open(f"{DATA_DIR}/tickets/{ticket}/contact_data.json", "w", encoding="utf-8") as f:
         json.dump(contact_data, f, indent=4)
-        
+
     if qcat_result.get("action") == "upload":
         if uploaded_files:
             # Specify the path where the zip file will be saved
@@ -382,6 +413,7 @@ def render_chat():
     """Render the chat interface with message history and input."""
     cfg = st.session_state.cfg
 
+    queries = 0
     # Display chat history
     for i, msg in enumerate(st.session_state.messages):
         with st.chat_message(msg["role"]):
@@ -396,8 +428,11 @@ def render_chat():
                     # Pre-populate if feedback already exists in history
                     default=msg.get("feedback", None),
                 )
+            if msg["role"] == "user":
+                queries += 1
 
-
+    if len(queries)>4 and st.session_state.feedback_data is None:
+        feedback_prompt()
     # Chat input
     user_input = st.chat_input("Ask a question about your knowledge base...",
                             accept_file="multiple",