Update app.py

app.py

@@ -40,27 +40,36 @@ 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):
     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))
+    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, update it
+        existing_item = items[0]
         existing_item.update(item_data)
-        container.replace_item(item_id, existing_item)
-    except CosmosResourceNotFoundError:
-        container.create_item(item_data)
+        container.replace_item(item=existing_item, body=existing_item)
+        st.write(f"Updated Item: {item_id}")
+    else:
+        # Item does not exist, create new
+        container.create_item(body=item_data)
+        st.write(f"Created New Item: {item_id}")
 
-# Function to Add or Update an Item
-def add_or_update_item(database_name, container_name, item_id, item_data):
+    # 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)
-    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)
-        
+    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']}")
+
+                    
 # Button to Trigger Display of Cosmos DB Structure
 if st.button('Show Cosmos DB Structure'):
     display_cosmos_db_structure()