Update app.py

app.py

@@ -1,22 +1,19 @@
-import streamlit as st
 import os
+import streamlit as st
 from azure.cosmos import CosmosClient, PartitionKey
 from azure.storage.blob import BlobServiceClient
 import requests
-from azure.identity import DefaultAzureCredential
 
 # Environment Variables
-COSMOS_ACCOUNT_URL = os.getenv('COSMOS_ACCOUNT_URL')
-COSMOS_ACCOUNT_KEY = os.getenv('COSMOS_ACCOUNT_KEY')
+COSMOS_CONNECTION_STRING = os.getenv('COSMOS_CONNECTION_STRING')
 BLOB_STORAGE_CONNECTION_STRING = os.getenv('BLOB_STORAGE_CONNECTION_STRING')
 
 # Initialize Azure Cosmos DB Client
-cosmos_client = CosmosClient(COSMOS_ACCOUNT_URL, credential=DefaultAzureCredential())
+cosmos_client = CosmosClient.from_connection_string(COSMOS_CONNECTION_STRING)
 
 # Initialize Azure Blob Storage Client
 blob_service = BlobServiceClient.from_connection_string(BLOB_STORAGE_CONNECTION_STRING)
 
-
 # Streamlit UI
 st.title('Azure Services Integration with Streamlit')
 
@@ -52,4 +49,3 @@ function_url = st.text_input('Function URL')
 if st.button('Call Azure Function'):
     response = requests.get(function_url)
     st.write('Function Response:', response.text)
-