Update app.py

app.py

@@ -3,6 +3,7 @@ import streamlit as st
 from azure.cosmos import CosmosClient, PartitionKey
 from azure.storage.blob import BlobServiceClient
 import requests
+import glob
 
 # Environment Variables
 COSMOS_CONNECTION_STRING = os.getenv('COSMOS_CONNECTION_STRING')
@@ -37,6 +38,42 @@ if st.button('Show Cosmos DB Structure'):
 
 
 
+# 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:
+        container.create_item(item_data)
+
+# Test Function to Insert PNG Images
+def test_insert_png_images(database_name, container_name):
+    png_files = glob.glob('*.png')
+    for file_name in png_files:
+        item_id = os.path.splitext(file_name)[0]  # Use file name without extension as ID
+        item_data = {"id": item_id, "file_name": file_name}
+        add_or_update_item(database_name, container_name, item_id, item_data)
+        st.write(f"Inserted: {file_name}")
+
+    # Displaying Images
+    st.subheader("Displaying Images in Container")
+    items = list(cosmos_client.get_database_client(database_name).get_container_client(container_name).read_all_items())
+    for item in items:
+        if 'file_name' in item and item['file_name'].endswith('.png'):
+            st.image(item['file_name'], caption=item['file_name'])
+
+# UI to Test Image Insertion Function
+if st.button('Test Insert PNG Images'):
+    test_database_name = st.text_input('Enter Test Database Name')
+    test_container_name = st.text_input('Enter Test Container Name')
+    if test_database_name and test_container_name:
+        test_insert_png_images(test_database_name, test_container_name)
+    else:
+        st.error('Please enter the test database and container names.')
+
+
 # Function to Add an Item
 def add_item_to_container(database_name, container_name, item):
     container = cosmos_client.get_database_client(database_name).get_container_client(container_name)