Update app.py

app.py

@@ -39,9 +39,8 @@ def fetch_all_records():
     items = list(container.query_items(query=query, enable_cross_partition_query=True))
     return pd.DataFrame(items)
 
-def bulk_delete_records(ids):
+def delete_records(ids):
     try:
-        # Cosmos DB doesn't support bulk delete directly, so we'll delete items one by one
         for id in ids:
             container.delete_item(item=id, partition_key=id)
         return True, f"Successfully deleted {len(ids)} records"
@@ -51,14 +50,17 @@ def bulk_delete_records(ids):
 # Streamlit app
 st.title("🌟 Cosmos DB Record Management")
 
+# Initialize session state for selected IDs
+if 'selected_ids' not in st.session_state:
+    st.session_state.selected_ids = set()
+
 # Login section
 if 'logged_in' not in st.session_state:
     st.session_state.logged_in = False
 
 if not st.session_state.logged_in:
     st.subheader("🔐 Login")
-    #input_key = st.text_input("Enter your key", type="password")
-    input_key = Key
+    input_key = st.text_input("Enter your key", type="password")
     if st.button("🚀 Login"):
         if input_key:
             st.session_state.primary_key = input_key
@@ -77,20 +79,32 @@ else:
     df = fetch_all_records()
     
     # Add a checkbox column to the dataframe
-    df['select'] = False
+    df['select'] = df['id'].isin(st.session_state.selected_ids)
     
     # Use Streamlit's data editor
-    edited_df = st.data_editor(df, num_rows="dynamic", key="data_editor")
+    edited_df = st.data_editor(df, key="data_editor", disabled=["id", "name", "document", "evaluationText", "evaluationScore"])
+
+    # Update selected_ids based on the edited dataframe
+    selected_rows = edited_df[edited_df['select']]
+    st.session_state.selected_ids = set(selected_rows['id'])
+
+    # Display selected IDs with emoji checkboxes
+    if st.session_state.selected_ids:
+        st.markdown("### Selected Records:")
+        for id in st.session_state.selected_ids:
+            st.markdown(f"✅ {id}")
+    else:
+        st.markdown("### No Records Selected")
 
     # Add delete and download buttons
     col1, col2 = st.columns(2)
     with col1:
         if st.button("🗑️ Delete Selected"):
-            selected_ids = edited_df[edited_df['select']]['id'].tolist()
-            if selected_ids:
-                success, message = bulk_delete_records(selected_ids)
+            if st.session_state.selected_ids:
+                success, message = delete_records(list(st.session_state.selected_ids))
                 if success:
                     st.success(message)
+                    st.session_state.selected_ids.clear()  # Clear the selection after successful deletion
                 else:
                     st.error(message)
                 st.rerun()
@@ -98,7 +112,7 @@ else:
                 st.warning("No records selected for deletion.")
 
     with col2:
-        if st.download_button("📥 Download Data", edited_df.to_csv(index=False), "cosmos_db_data.csv", "text/csv"):
+        if st.download_button("📥 Download Data", df.to_csv(index=False), "cosmos_db_data.csv", "text/csv"):
             st.success("Data downloaded successfully!")
 
     # Input fields for new record
@@ -151,6 +165,7 @@ else:
     # Logout button
     if st.button("🚪 Logout"):
         st.session_state.logged_in = False
+        st.session_state.selected_ids.clear()  # Clear selected IDs on logout
         st.rerun()
 
     # Display connection info