Update app.py

app.py

@@ -1,6 +1,5 @@
 import streamlit as st
 from azure.cosmos import CosmosClient
-import os
 
 # Load and save query files
 def load_query(filename):
@@ -14,20 +13,21 @@ def save_query(filename, query):
 # Streamlit UI
 st.title("Azure Cosmos DB Explorer 👽")
 
-# Environment Variables for Cosmos DB Credentials
-ACCOUNT_URI = os.environ.get('ACCOUNT_URI', 'Your Cosmos DB URI here')
-ACCOUNT_KEY = os.environ.get('ACCOUNT_KEY', 'Your Cosmos DB Key here')
+# Default URI and KEY (You can set these values in code)
+default_account_uri = "Your Default Cosmos DB URI Here"
+default_account_key = "Your Default Cosmos DB Key Here"
 
 client = None
 
 # Connection Details Expander
 with st.expander("Connect 🌍"):
-    st.write("Environment Variables for Cosmos DB Credentials are pre-loaded.")
+    account_uri = st.text_input("Account URI:", default_account_uri)
+    account_key = st.text_input("Account Key:", default_account_key, type="password")
     database_name = st.text_input("Database Name:", "")
     container_name = st.text_input("Container Name:", "")
     if st.button("Connect"):
         try:
-            client = CosmosClient(ACCOUNT_URI, credential=ACCOUNT_KEY)
+            client = CosmosClient(account_uri, credential=account_key)
             database_client = client.get_database_client(database_name)
             container_client = database_client.get_container_client(container_name)
             st.success("Connected successfully! 🎉")
@@ -72,8 +72,4 @@ with st.expander("Query Editor 📝"):
 st.markdown("""
 ## Instructions to Run this App:
 
-1. **Environment Variables**: Set the `ACCOUNT_URI` and `ACCOUNT_KEY` environment variables with your Azure Cosmos DB URI and Key respectively.
-2. **Install Packages**: If you haven't, install the required Python packages:
-3. **Run App**: Save this code in a file, say `streamlit_cosmosdb_app.py`, and then run `streamlit run streamlit_cosmosdb_app.py`.
-4. **Execute**: Use the UI to connect and run SQL queries against your Cosmos DB.
-""")
+1. **Install Packages**: If you haven't, install the required Python packages: