Update app.py

app.py

@@ -2,6 +2,8 @@ import os
 import streamlit as st
 from azure.cosmos import CosmosClient, PartitionKey
 from azure.storage.blob import BlobServiceClient
+from azure.cosmos.exceptions import CosmosResourceNotFoundError
+
 import requests
 import glob
 
@@ -39,15 +41,14 @@ def display_cosmos_db_structure():
                     st.markdown(item_desc)
 
                     
-
 # Function to Add or Update an Item
-def add_or_update_item(database_client, container_name, item_id, item_data):
-    container = database_client.get_container_client(container_name)
+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:
+    except CosmosResourceNotFoundError:
         container.create_item(item_data)
 
 # Function to Add or Update an Item