Update app.py

app.py

@@ -15,6 +15,7 @@ cosmos_client = CosmosClient.from_connection_string(COSMOS_CONNECTION_STRING)
 # Initialize Azure Blob Storage Client
 blob_service = BlobServiceClient.from_connection_string(BLOB_STORAGE_CONNECTION_STRING)
 
+
 # Function to Retrieve and Display Cosmos DB Structure
 def display_cosmos_db_structure():
     st.subheader('Azure Cosmos DB Structure')
@@ -30,11 +31,14 @@ def display_cosmos_db_structure():
             
             items = list(container_client.read_all_items())
             for item in items:
-                st.markdown(f"  - Item: `{item['id']}`")  # Replace 'id' with the appropriate key if different
+                item_desc = f"  - Item: `{item['id']}`"
+                if 'file_name' in item and item['file_name'].endswith('.png'):
+                    st.markdown(item_desc)
+                    st.image(item['file_name'])
+                else:
+                    st.markdown(item_desc)
 
-# Button to Trigger Display of Cosmos DB Structure
-if st.button('Show Cosmos DB Structure'):
-    display_cosmos_db_structure()
+                    
 
 # Function to Add or Update an Item
 def add_or_update_item(database_client, container_name, item_id, item_data):
@@ -46,6 +50,20 @@ def add_or_update_item(database_client, container_name, item_id, item_data):
     except:
         container.create_item(item_data)
 
+# Function to Add or Update an Item
+def add_or_update_item(database_name, container_name, item_id, item_data):
+    container = cosmos_client.get_database_client(database_name).get_container_client(container_name)
+    try:
+        existing_item = container.read_item(item_id, partition_key=PartitionKey(item_id))
+        existing_item.update(item_data)
+        container.replace_item(item_id, existing_item)
+    except exceptions.CosmosResourceNotFoundError:
+        container.create_item(item_data)
+        
+# Button to Trigger Display of Cosmos DB Structure
+if st.button('Show Cosmos DB Structure'):
+    display_cosmos_db_structure()
+        
 # Test Function to Insert PNG Images
 def test_insert_png_images():
     # Get the first database and container