Update app.py

app.py

@@ -3,12 +3,10 @@ from azure.cosmos import CosmosClient, PartitionKey
 import os
 import pandas as pd
 
-# Streamlit page configuration for wide mode
-st.set_page_config(layout="wide")
-
 # Cosmos DB configuration
 ENDPOINT = "https://acae-afd.documents.azure.com:443/"
 SUBSCRIPTION_ID = "003fba60-5b3f-48f4-ab36-3ed11bc40816"
+# You'll need to set these environment variables or use Azure Key Vault
 DATABASE_NAME = os.environ.get("COSMOS_DATABASE_NAME")
 CONTAINER_NAME = os.environ.get("COSMOS_CONTAINER_NAME")
 Key = os.environ.get("Key")
@@ -49,26 +47,20 @@ 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")
 
+# 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 = Key
+    input_key = Key  # Use the predefined Key instead of asking for user input
     if st.button("🚀 Login"):
         if input_key:
             st.session_state.primary_key = input_key
@@ -82,135 +74,103 @@ else:
     database = client.get_database_client(DATABASE_NAME)
     container = database.get_container_client(CONTAINER_NAME)
 
-    # Fetch all records
-    records_df = fetch_all_records()
-
-    # Sidebar for data editing
-    st.sidebar.title("Edit Records")
-    edited_df = st.sidebar.data_editor(records_df, num_rows="dynamic", key="sidebar_editor")
-
-    # Main Content Area
-    col1, col2 = st.columns([2, 1])
-
-    with col1:
-        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)
-
-        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("🔍 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")
-
+    # Fetch and display all records
     st.subheader("📊 All Records")
-    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()
-
-    col6, col7 = st.columns(2)
-    with col6:
+    df = fetch_all_records()
+    
+    # Add a checkbox column to the dataframe
+    df['select'] = df['id'].isin(st.session_state.selected_ids)
+    
+    # Use Streamlit's 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'] == True]['id'].tolist()
-            if selected_ids:
-                success, message = 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()
             else:
                 st.warning("No records selected for deletion.")
 
-    with col7:
-        if st.download_button("📥 Download Data", edited_df.to_csv(index=False), "cosmos_db_data.csv", "text/csv"):
+    with col2:
+        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
+    st.subheader("📝 Enter New Record Details")
+    new_id = st.text_input("ID")
+    new_name = st.text_input("Name")
+    new_document = st.text_area("Document")
+    new_evaluation_text = st.text_area("Evaluation Text")
+    new_evaluation_score = st.number_input("Evaluation Score", min_value=0, max_value=100, step=1)
+
+    col1, col2 = st.columns(2)
+    
+    # Insert Record button
+    with col1:
+        if st.button("💾 Insert Record"):
+            record = {
+                "id": new_id,
+                "name": new_name,
+                "document": new_document,
+                "evaluationText": new_evaluation_text,
+                "evaluationScore": new_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}")
+            st.rerun()
+
+    # Call Procedure button
+    with col2:
+        if st.button("🔧 Call Procedure"):
+            record = {
+                "id": new_id,
+                "name": new_name,
+                "document": new_document,
+                "evaluationText": new_evaluation_text,
+                "evaluationScore": new_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}")
+
     # 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
-    st.sidebar.markdown("---")
     st.sidebar.subheader("🔗 Connection Information")
     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}")
-
-# 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']
+    st.sidebar.text(f"Container: {CONTAINER_NAME}")