Update app.py

app.py

@@ -1,11 +1,6 @@
 import streamlit as st
-import json
 from azure.cosmos import CosmosClient
-
-# Initialize variables
-connection_str = ""
-database_name = ""
-container_name = ""
+import os
 
 # Load and save query files
 def load_query(filename):
@@ -16,31 +11,32 @@ def save_query(filename, query):
     with open(filename, 'w') as file:
         file.write(query)
 
-# Connect to Azure Cosmos DB
-def connect_to_cosmosdb(url, key, database, container):
-    client = CosmosClient(url, credential=key)
-    db = client.get_database_client(database)
-    container = db.get_container_client(container)
-    return container
-
 # 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')
+
+client = None
+
 # Connection Details Expander
 with st.expander("Connect 🌍"):
-    connection_str = st.text_input("Connection String (URL):", connection_str)
-    database_name = st.text_input("Database Name:", database_name)
-    container_name = st.text_input("Container Name:", container_name)
+    st.write("Environment Variables for Cosmos DB Credentials are pre-loaded.")
+    database_name = st.text_input("Database Name:", "")
+    container_name = st.text_input("Container Name:", "")
     if st.button("Connect"):
         try:
-            container = connect_to_cosmosdb(connection_str, database_name, container_name)
+            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! 🎉")
         except Exception as e:
             st.error(f"Failed to connect: {e}")
 
 # Query Editor Expander
 with st.expander("Query Editor 📝"):
-    query = st.text_area("Enter your SQL query here:")
+    query = st.text_area("Enter your SQL query here:", "")
     file_option = st.selectbox("File Options", ["New", "Open", "Save", "Save As"])
     
     if file_option == "New":
@@ -59,9 +55,22 @@ with st.expander("Query Editor 📝"):
             save_query(saveas_filename, query)
     
     if st.button("Execute Query 🚀"):
-        try:
-            results = list(container.query_items(query=query, enable_cross_partition_query=True))
-            st.write("Results 📋:")
-            st.json(results)
-        except Exception as e:
-            st.error(f"Query failed: {e}")
+        if client:
+            try:
+                items = list(container_client.query_items(
+                    query=query,
+                    enable_cross_partition_query=True
+                ))
+                st.write("Results 📋:")
+                st.json(items)
+            except Exception as e:
+                st.error(f"Query failed: {e}")
+        else:
+            st.warning("Not connected to any Cosmos DB. Please connect first.")
+
+# Instructions
+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: