Update app.py

app.py

@@ -196,6 +196,52 @@ def archive_all_data(client):
     except Exception as e:
         return f"An error occurred while archiving data: {str(e)}"
 
+# Cosmos DB operations
+def create_item(client, db_name, container_name):
+    st.subheader("Create New Item")
+    item_id = st.text_input("Item ID")
+    item_data = st.text_area("Item Data (JSON format)")
+    
+    if st.button("💾 Save New Item"):
+        try:
+            container = client.get_database_client(db_name).get_container_client(container_name)
+            new_item = {"id": item_id, **json.loads(item_data)}
+            container.create_item(body=new_item)
+            st.success("New item created successfully!")
+            st.rerun()
+        except Exception as e:
+            st.error(f"Error creating item: {str(e)}")
+
+def edit_item(client, db_name, container_name, item_id):
+    container = client.get_database_client(db_name).get_container_client(container_name)
+    item = container.read_item(item=item_id, partition_key=item_id)
+    
+    st.subheader(f"Editing Item: {item_id}")
+    edited_item = {}
+    for key, value in item.items():
+        if key not in ['_rid', '_self', '_etag', '_attachments', '_ts']:
+            edited_item[key] = st.text_input(key, value)
+    
+    if st.button("💾 Save Changes"):
+        try:
+            container.upsert_item(body=edited_item)
+            st.success("Item updated successfully!")
+            st.rerun()
+        except Exception as e:
+            st.error(f"Error updating item: {str(e)}")
+
+def delete_item(client, db_name, container_name, item_id):
+    st.subheader(f"Delete Item: {item_id}")
+    st.write("Are you sure you want to delete this item?")
+    if st.button("🗑️ Confirm Delete"):
+        try:
+            container = client.get_database_client(db_name).get_container_client(container_name)
+            container.delete_item(item=item_id, partition_key=item_id)
+            st.success(f"Item {item_id} deleted successfully!")
+            st.rerun()
+        except Exception as e:
+            st.error(f"Error deleting item: {str(e)}")
+
 # Main Streamlit app
 def main():
     st.set_page_config(layout="wide")
@@ -212,14 +258,14 @@ def main():
     # Login section
     if not st.session_state.logged_in:
         st.subheader("🔐 Login")
-        input_key = st.text_input("Enter your Cosmos DB key", type="password")
+        input_key = Key  # Get the key from the environment variable
         if st.button("🚀 Login"):
             if input_key:
                 st.session_state.primary_key = input_key
                 st.session_state.logged_in = True
                 st.rerun()
             else:
-                st.error("Invalid key. Please enter a valid Cosmos DB key.")
+                st.error("Invalid key. Please check your environment variables.")
     else:
         # Initialize Cosmos DB client
         try: