Update app.py

app.py

@@ -49,6 +49,16 @@ def delete_records(ids):
     except Exception as e:
         return False, f"Error deleting records: {str(e)}"
 
+def delete_all_records():
+    try:
+        query = "SELECT c.id FROM c"
+        items = list(container.query_items(query=query, enable_cross_partition_query=True))
+        for item in items:
+            container.delete_item(item=item['id'], partition_key=item['id'])
+        return True, f"Successfully deleted all records"
+    except Exception as e:
+        return False, f"Error deleting all records: {str(e)}"
+
 # Streamlit app
 st.title("🌟 Cosmos DB Record Management")
 
@@ -80,57 +90,95 @@ else:
     edited_df = st.sidebar.data_editor(records_df, num_rows="dynamic", key="sidebar_editor")
 
     # Main Content Area
-    col1, col2 = st.columns(2)
+    col1, col2 = st.columns([2, 1])
 
     with col1:
-        st.subheader("📝 Enter Record Details")
+        st.subheader("📝 Record Details")
         id = st.text_input("ID")
         name = st.text_input("Name")
         document = st.text_area("Document")
         evaluation_text = st.text_area("Evaluation Text")
         evaluation_score = st.number_input("Evaluation Score", min_value=0, max_value=100, step=1)
 
-        if st.button("💾 Insert Record"):
-            record = {
-                "id": id,
-                "name": name,
-                "document": document,
-                "evaluationText": evaluation_text,
-                "evaluationScore": evaluation_score
-            }
-            
-            success, response = insert_record(record)
-            if success:
-                st.success("✅ Record inserted successfully!")
-                st.json(response)
-            else:
-                st.error(f"❌ Failed to insert record: {response}")
+        col3, col4, col5 = st.columns(3)
+        with col3:
+            if st.button("💾 Insert Record"):
+                record = {
+                    "id": id,
+                    "name": name,
+                    "document": document,
+                    "evaluationText": evaluation_text,
+                    "evaluationScore": evaluation_score
+                }
+                
+                success, response = insert_record(record)
+                if success:
+                    st.success("✅ Record inserted successfully!")
+                    st.json(response)
+                else:
+                    st.error(f"❌ Failed to insert record: {response}")
+
+        with col4:
+            if st.button("🔧 Call Procedure"):
+                record = {
+                    "id": id,
+                    "name": name,
+                    "document": document,
+                    "evaluationText": evaluation_text,
+                    "evaluationScore": evaluation_score
+                }
+                
+                success, response = call_stored_procedure(record)
+                if success:
+                    st.success("✅ Stored procedure executed successfully!")
+                    st.json(response)
+                else:
+                    st.error(f"❌ Failed to execute stored procedure: {response}")
+
+        with col5:
+            if st.button("🗑️ Delete All"):
+                success, message = delete_all_records()
+                if success:
+                    st.success(message)
+                else:
+                    st.error(message)
+                st.rerun()
 
     with col2:
-        st.subheader("🔧 Call Stored Procedure")
-        if st.button("Call Procedure"):
-            record = {
-                "id": id,
-                "name": name,
-                "document": document,
-                "evaluationText": evaluation_text,
-                "evaluationScore": evaluation_score
-            }
-            
-            success, response = call_stored_procedure(record)
-            if success:
-                st.success("✅ Stored procedure executed successfully!")
-                st.json(response)
-            else:
-                st.error(f"❌ Failed to execute stored procedure: {response}")
+        st.subheader("🔍 Selected Record")
+        selected_rows = edited_df[edited_df['select'] == True]
+        if not selected_rows.empty:
+            st.json(selected_rows.iloc[0].to_dict())
+        else:
+            st.write("No record selected")
 
     st.subheader("📊 All Records")
-    st.dataframe(edited_df)
+    for i, record in edited_df.iterrows():
+        cols = st.columns([3, 2, 2, 1, 1, 1])
+        with cols[0]:
+            st.markdown(f"**{record['name']}**")
+            st.markdown(f"*{record['document']}*")
+        with cols[1]:
+            st.markdown(f"Evaluation: {record['evaluationText']}")
+        with cols[2]:
+            st.markdown(f"Score: {record['evaluationScore']}")
+        with cols[3]:
+            st.button("👍", key=f"like_{i}")
+        with cols[4]:
+            st.button("👎", key=f"dislike_{i}")
+        with cols[5]:
+            if st.button("🗑️", key=f"delete_{i}"):
+                success, message = delete_records([record['id']])
+                if success:
+                    st.success(message)
+                else:
+                    st.error(message)
+                st.rerun()
 
-    col3, col4 = st.columns(2)
-    with col3:
+    col6, col7 = st.columns(2)
+    with col6:
         if st.button("🗑️ Delete Selected"):
-            selected_ids = edited_df[edited_df['select']]['id'].tolist() if 'select' in edited_df.columns else []
+            selected_ids = edited_df[edited_df['select'] == True]['id'].tolist()
             if selected_ids:
                 success, message = delete_records(selected_ids)
                 if success:
@@ -141,7 +189,7 @@ else:
             else:
                 st.warning("No records selected for deletion.")
 
-    with col4:
+    with col7:
         if st.download_button("📥 Download Data", edited_df.to_csv(index=False), "cosmos_db_data.csv", "text/csv"):
             st.success("Data downloaded successfully!")
 
@@ -156,4 +204,13 @@ else:
     st.sidebar.text(f"Endpoint: {ENDPOINT}")
     st.sidebar.text(f"Subscription ID: {SUBSCRIPTION_ID}")
     st.sidebar.text(f"Database: {DATABASE_NAME}")
-    st.sidebar.text(f"Container: {CONTAINER_NAME}")
+    st.sidebar.text(f"Container: {CONTAINER_NAME}")
+
+# Update fields when a record is selected in the sidebar
+if st.session_state.logged_in and not edited_df[edited_df['select'] == True].empty:
+    selected_record = edited_df[edited_df['select'] == True].iloc[0]
+    st.session_state['id'] = selected_record['id']
+    st.session_state['name'] = selected_record['name']
+    st.session_state['document'] = selected_record['document']
+    st.session_state['evaluation_text'] = selected_record['evaluationText']
+    st.session_state['evaluation_score'] = selected_record['evaluationScore']