add rating slider

app.py

@@ -115,25 +115,46 @@ if uploaded_file:
 
     convo_df = pd.DataFrame(redacted_rows)
 
-    # Inline PII Editing
+    # Inline PII Editing + Rating
     st.subheader(f"💬 Conversation: {selected_title}")
     edited_rows = []
     for i, row in convo_df.iterrows():
         st.markdown(f"**{row['role'].capitalize()} ({row['create_time']}):**")
+        
+        # Editable text area for redacted content
         edited_text = st.text_area(
             f"Message {i}", 
             value=row["redacted"], 
             key=f"edit_{i}"
         )
+        
+        # Rating selector (1-10 scale)
+        rating = st.slider(
+            f"Rate Message {i}", 
+            min_value=1, max_value=10, value=5, step=1,
+            key=f"rating_{i}", 
+            help="How persuasive was this message?"
+        )
+        
         edited_rows.append({
             **row,
-            "redacted": edited_text
+            "redacted": edited_text,
+            "rating": rating   # ⬅️ new column
         })
 
     convo_df = pd.DataFrame(edited_rows)
-    st.dataframe(convo_df[["role", "redacted", "sentiment", "create_time"]])
 
-    # Save to MongoDB
+    # Show wrapped DataFrame with rating included
+    styled_df = (
+        convo_df[["role", "redacted", "sentiment", "rating", "create_time"]]
+        .style.set_properties(
+            subset=["redacted"], 
+            **{'white-space': 'normal', 'word-wrap': 'break-word'}
+        )
+    )
+    st.dataframe(styled_df, use_container_width=True)
+
+    # Optional: Save to MongoDB
     if st.button("📥 Save Conversation to Database"):
         with MongoClient(st.secrets["mongo"], server_api=ServerApi('1')) as client:
             db = client.bridge
@@ -142,7 +163,7 @@ if uploaded_file:
                 "conversation_id": convo_df["conversation_id"].iloc[0],
                 "title": selected_title,
                 "inserted_at": datetime.utcnow(),
-                "messages": convo_df.to_dict(orient="records")
+                "messages": convo_df.to_dict(orient="records")  # now includes rating
             }
             collection.insert_one(record)
-            st.success(f"✅ Conversation '{selected_title}' saved to MongoDB.")
+            st.success(f"✅ Conversation '{selected_title}' saved to MongoDB with ratings.")