Update app.py

app.py

@@ -1,14 +1,15 @@
 import streamlit as st
-from azure.cosmos import CosmosClient
+from azure.cosmos import CosmosClient, PartitionKey
 from azure.storage.blob import BlobServiceClient
 import requests
+from azure.identity import DefaultAzureCredential
 
-# Azure Services Initialization (Replace with actual credentials)
-cosmos_client = CosmosClient('<your_cosmos_account_url>', credential='<your_cosmos_key>')
+# Initialize Azure Cosmos DB Client
+cosmos_client = CosmosClient('<your_cosmos_account_url>', credential=DefaultAzureCredential())
 blob_service = BlobServiceClient.from_connection_string('<your_blob_storage_connection_string>')
 
 # Streamlit UI
-st.title('Azure Services - Streamlit Integration')
+st.title('Azure Services Integration with Streamlit')
 
 # Azure Cosmos DB - CRUD Operations
 st.subheader('Azure Cosmos DB - CRUD Operations')
@@ -18,12 +19,12 @@ item_id = st.text_input("Item ID (for Read, Update, Delete)")
 item_data = st.text_area("Item Data (JSON format, for Create and Update)")
 
 if st.button('Create Item in Cosmos DB'):
-    container = cosmos_client.get_container_client(cosmos_db, cosmos_container)
+    container = cosmos_client.get_database_client(cosmos_db).get_container_client(cosmos_container)
     container.create_item(item_data)
 
 if st.button('Read Item from Cosmos DB'):
-    container = cosmos_client.get_container_client(cosmos_db, cosmos_container)
-    item = container.read_item(item_id, partition_key=item_id)
+    container = cosmos_client.get_database_client(cosmos_db).get_container_client(cosmos_container)
+    item = container.read_item(item_id, partition_key=PartitionKey(item_id))
     st.json(item)
 
 # Azure Blob Storage - Upload/Download
@@ -38,6 +39,8 @@ if blob_file is not None and st.button('Upload to Blob'):
 # Azure Functions - Trigger
 st.subheader('Azure Functions - Trigger')
 function_url = st.text_input('Function URL')
+
 if st.button('Call Azure Function'):
     response = requests.get(function_url)
     st.write('Function Response:', response.text)
+