.

app.py

@@ -110,16 +110,26 @@ if uploaded_file:
     # Scrub + Sentiment
     # -----------------------------
     cleaner = scrubadub.Scrubber()
-    convo_df["redacted"] = convo_df["content"].apply(lambda x: cleaner.clean(str(x)))
-
     analyzer = SentimentIntensityAnalyzer()
-    convo_df["sentiment"] = convo_df["content"].apply(lambda x: analyzer.polarity_scores(str(x))["compound"])
+
+    redacted_rows = []
+    for i, row in convo_df.iterrows():
+        original_text = str(row["content"])
+        redacted_text = cleaner.clean(original_text)
+        sentiment_score = analyzer.polarity_scores(original_text)["compound"]
+        redacted_rows.append({
+            **row,
+            "redacted": redacted_text,
+            "sentiment": sentiment_score
+        })
+
+    convo_df = pd.DataFrame(redacted_rows)
 
     # -----------------------------
     # Inline PII Editing
     # -----------------------------
     st.subheader(f"💬 Conversation: {selected_title}")
-    editable_rows = []
+    edited_rows = []
     for i, row in convo_df.iterrows():
         st.markdown(f"**{row['role'].capitalize()} ({row['create_time']}):**")
         edited_text = st.text_area(
@@ -127,12 +137,12 @@ if uploaded_file:
             value=row["redacted"], 
             key=f"edit_{i}"
         )
-        editable_rows.append({
+        edited_rows.append({
             **row,
             "redacted": edited_text
         })
 
-    convo_df = pd.DataFrame(editable_rows)
+    convo_df = pd.DataFrame(edited_rows)
     st.dataframe(convo_df[["role", "redacted", "sentiment", "create_time"]])
 
     # -----------------------------