Spaces:
Sleeping
Sleeping
File size: 3,046 Bytes
c55395e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 |
import streamlit as st
from azure.cosmos import CosmosClient
import os
# Load and save query files
def load_query(filename):
with open(filename, 'r') as file:
return file.read()
def save_query(filename, query):
with open(filename, 'w') as file:
file.write(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')
client = None
# Connection Details Expander
with st.expander("Connect 🌍"):
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:
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:", "")
file_option = st.selectbox("File Options", ["New", "Open", "Save", "Save As"])
if file_option == "New":
query = ""
elif file_option == "Open":
open_file = st.file_uploader("Choose a file:", type=["txt"])
if open_file is not None:
query = load_query(open_file)
elif file_option == "Save":
save_filename = st.text_input("Enter filename to save:", "my_query.txt")
if st.button("Save Query"):
save_query(save_filename, query)
elif file_option == "Save As":
saveas_filename = st.text_input("Enter new filename:", "my_new_query.txt")
if st.button("Save As"):
save_query(saveas_filename, query)
if st.button("Execute Query 🚀"):
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:
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.
""")
|