Update app.py

app.py

@@ -40,36 +40,43 @@ def display_cosmos_db_structure():
                 else:
                     st.markdown(item_desc)
 
-# Function to Add or Update an Item
-def add_or_update_item(database_name, container_name, item_id, item_data):
+                    
+# Function to Add or Update PNG Images
+def add_or_update_png_images(database_name, container_name):
     container = cosmos_client.get_database_client(database_name).get_container_client(container_name)
+    existing_items = list(container.read_all_items())
+    existing_ids = {item['id'] for item in existing_items}
+
+    # Display and delete option for existing items
+    for item in existing_items:
+        if 'file_name' in item and item['file_name'].endswith('.png'):
+            st.markdown(f"Item: `{item['id']}`")
+            st.image(item['file_name'])
+            if st.button(f"🗑️ Delete {item['id']}", key=f"delete_{item['id']}"):
+                container.delete_item(item=item, partition_key=item['id'])
+
+    # Add or update PNG files from directory
+    png_files = glob.glob('*.png')
+    for file_name in png_files:
+        item_id = os.path.splitext(file_name)[0]
+        if item_id not in existing_ids:
+            item_data = {"id": item_id, "file_name": file_name}
+            container.create_item(body=item_data)
+            st.write(f"Added Item: {item_id}")
+        else:
+            st.write(f"Item already exists: {item_id}")
 
-    # Check if item exists
-    query = "SELECT * FROM c WHERE c.id = @id"
-    parameters = [{"name": "@id", "value": item_id}]
-    items = list(container.query_items(query=query, parameters=parameters, enable_cross_partition_query=True))
-
-    if items:
-        # Item exists, delete it
-        container.delete_item(item=items[0], partition_key=item_id)
-
-    # Add new item
-    container.create_item(body=item_data)
-    st.write(f"Added Item: {item_id}")
-
-    # Display item ID and image if available
-    display_item(database_name, container_name, item_id)
-
-
-
-# Function to Display an Item
-def display_item(database_name, container_name, item_id):
-    container = cosmos_client.get_database_client(database_name).get_container_client(container_name)
-    item = container.read_item(item_id, partition_key=PartitionKey(item_id))
-    st.markdown(f"- **Item ID**: {item_id}")
-    if 'file_name' in item and item['file_name'].endswith('.png'):
-        st.image(item['file_name'], caption=f"Image: {item['file_name']}")
+# UI to manage PNG images
+st.title('Manage PNG Images in Cosmos DB')
+selected_db_name = st.text_input('Enter Database Name')
+selected_container_name = st.text_input('Enter Container Name')
 
+if st.button('Manage Images'):
+    if selected_db_name and selected_container_name:
+        add_or_update_png_images(selected_db_name, selected_container_name)
+    else:
+        st.error('Please enter both database and container names.')
+        
                     
 # Button to Trigger Display of Cosmos DB Structure
 if st.button('Show Cosmos DB Structure'):