edit pii.

app.py

@@ -9,10 +9,11 @@ from datetime import datetime
 from uuid import uuid4
 
 # -----------------------------
-# Consent Setup
+# Page Config & Consent
 # -----------------------------
 st.set_page_config(page_title="ChatGPT Log Analyzer", page_icon="🤖")
 
+# Consent
 if "consent" not in st.session_state:
     st.session_state.consent = ""
 
@@ -40,11 +41,21 @@ if st.session_state.consent == "Yes, I consent":
         st.session_state.id = datetime.now().strftime('%Y%m-%d%H-%M-') + str(uuid4())
     st.success("✅ You consented to participate.")
     st.info(f"Your anonymized ID is: **{st.session_state.id}**. Keep this if you want your data deleted later.")
-
 elif st.session_state.consent == "No, I do not consent":
     placeholder.empty()
     st.warning("⚠️ You did not consent. You can still use the app, but your logs will not be stored.")
 
+# -----------------------------
+# Privacy Policy Dropdown
+# -----------------------------
+with st.expander("Privacy Policy", expanded=False):
+    st.markdown("""
+    **Data Collection:** Only ChatGPT logs you upload will be analyzed.  
+    **PII Redaction:** All detected names, emails, and other personal info will be redacted.  
+    **Storage:** Data is stored anonymized in our research database.  
+    **Deletion:** You can request deletion using your anonymized ID.  
+    """)
+
 # -----------------------------
 # Parser Function
 # -----------------------------
@@ -82,39 +93,17 @@ if uploaded_file:
     data = json.load(uploaded_file)
     if isinstance(data, dict) and "conversations" in data:
         df = parse_chatgpt_export(data)
-        convo_titles = df["title"].unique()
     else:
         st.error("Unsupported JSON structure")
         st.stop()
 
     # -----------------------------
-    # Session State for Pagination
-    # -----------------------------
-    if "convo_index" not in st.session_state:
-        st.session_state.convo_index = 0
-
-    # -----------------------------
-    # Dropdown Selection
+    # Conversation Selector
     # -----------------------------
-    selected_title = st.selectbox("Select Conversation", options=convo_titles, index=st.session_state.convo_index)
-    st.session_state.convo_index = list(convo_titles).index(selected_title)
+    st.subheader("🗂 Select a Conversation")
+    convo_titles = df["title"].unique()
+    selected_title = st.selectbox("Choose conversation", convo_titles)
 
-    # -----------------------------
-    # Pagination Buttons
-    # -----------------------------
-    col1, col2, col3 = st.columns([1,2,1])
-    with col1:
-        if st.button("⬅ Previous") and st.session_state.convo_index > 0:
-            st.session_state.convo_index -= 1
-            selected_title = convo_titles[st.session_state.convo_index]
-    with col3:
-        if st.button("Next ➡") and st.session_state.convo_index < len(convo_titles)-1:
-            st.session_state.convo_index += 1
-            selected_title = convo_titles[st.session_state.convo_index]
-
-    # -----------------------------
-    # Filter Selected Conversation
-    # -----------------------------
     convo_df = df[df["title"] == selected_title].copy()
 
     # -----------------------------
@@ -127,9 +116,23 @@ if uploaded_file:
     convo_df["sentiment"] = convo_df["content"].apply(lambda x: analyzer.polarity_scores(str(x))["compound"])
 
     # -----------------------------
-    # Show Conversation
+    # Inline PII Editing
     # -----------------------------
-    st.subheader(f"💬 Conversation ({st.session_state.convo_index+1}/{len(convo_titles)}): {selected_title}")
+    st.subheader(f"💬 Conversation: {selected_title}")
+    editable_rows = []
+    for i, row in convo_df.iterrows():
+        st.markdown(f"**{row['role'].capitalize()} ({row['create_time']}):**")
+        edited_text = st.text_area(
+            f"Message {i}", 
+            value=row["redacted"], 
+            key=f"edit_{i}"
+        )
+        editable_rows.append({
+            **row,
+            "redacted": edited_text
+        })
+
+    convo_df = pd.DataFrame(editable_rows)
     st.dataframe(convo_df[["role", "redacted", "sentiment", "create_time"]])
 
     # -----------------------------
@@ -146,4 +149,4 @@ if uploaded_file:
                 "messages": convo_df.to_dict(orient="records")
             }
             collection.insert_one(record)
-            st.success(f"✅ Conversation '{selected_title}' saved to MongoDB.")
+            st.success(f"✅ Conversation '{selected_title}' saved to MongoDB.")