Spaces:
Sleeping
Sleeping
import streamlit as st | |
from azure.cosmos import CosmosClient | |
import os | |
# Cosmos DB configuration | |
ENDPOINT = "https://acae-afd.documents.azure.com:443/" | |
SUBSCRIPTION_ID = "003fba60-5b3f-48f4-ab36-3ed11bc40816" | |
# You'll need to set these environment variables or use Azure Key Vault | |
DATABASE_NAME = os.environ.get("COSMOS_DATABASE_NAME") | |
CONTAINER_NAME = os.environ.get("COSMOS_CONTAINER_NAME") | |
Key = os.environ.get("Key") | |
def insert_record(record): | |
try: | |
response = container.create_item(body=record) | |
return True, response | |
except Exception as e: | |
return False, str(e) | |
def call_stored_procedure(record): | |
try: | |
response = container.scripts.execute_stored_procedure( | |
sproc="processPrompt", | |
params=[record], | |
partition_key=record['id'] | |
) | |
return True, response | |
except Exception as e: | |
error_message = f"Error type: {type(e).__name__}\nError message: {str(e)}" | |
if hasattr(e, 'sub_status'): | |
error_message += f"\nSub-status: {e.sub_status}" | |
if hasattr(e, 'response'): | |
error_message += f"\nResponse: {e.response}" | |
return False, error_message | |
# Streamlit app | |
st.title("π Cosmos DB Record Insertion and Procedure Execution") | |
# Login section | |
if 'logged_in' not in st.session_state: | |
st.session_state.logged_in = False | |
if not st.session_state.logged_in: | |
st.subheader("π Login") | |
input_key = Key | |
if st.button("π Login"): | |
if input_key: | |
st.session_state.primary_key = input_key | |
st.session_state.logged_in = True | |
st.rerun() | |
else: | |
st.error("Please enter a valid key") | |
else: | |
# Initialize Cosmos DB client | |
client = CosmosClient(ENDPOINT, credential=st.session_state.primary_key) | |
database = client.get_database_client(DATABASE_NAME) | |
container = database.get_container_client(CONTAINER_NAME) | |
# Input fields | |
st.subheader("π Enter Record Details") | |
id = st.text_input("ID") | |
name = st.text_input("Name") | |
document = st.text_area("Document") | |
evaluation_text = st.text_area("Evaluation Text") | |
evaluation_score = st.number_input("Evaluation Score", min_value=0, max_value=100, step=1) | |
col1, col2 = st.columns(2) | |
# Insert Record button | |
with col1: | |
if st.button("πΎ Insert Record"): | |
record = { | |
"id": id, | |
"name": name, | |
"document": document, | |
"evaluationText": evaluation_text, | |
"evaluationScore": evaluation_score | |
} | |
success, response = insert_record(record) | |
if success: | |
st.success("β Record inserted successfully!") | |
st.json(response) | |
else: | |
st.error(f"β Failed to insert record: {response}") | |
# Call Procedure button | |
with col2: | |
if st.button("π§ Call Procedure"): | |
record = { | |
"id": id, | |
"name": name, | |
"document": document, | |
"evaluationText": evaluation_text, | |
"evaluationScore": evaluation_score | |
} | |
success, response = call_stored_procedure(record) | |
if success: | |
st.success("β Stored procedure executed successfully!") | |
st.json(response) | |
else: | |
st.error(f"β Failed to execute stored procedure: {response}") | |
# Logout button | |
if st.button("πͺ Logout"): | |
st.session_state.logged_in = False | |
st.rerun() | |
# Display connection info | |
st.sidebar.subheader("π Connection Information") | |
st.sidebar.text(f"Endpoint: {ENDPOINT}") | |
st.sidebar.text(f"Subscription ID: {SUBSCRIPTION_ID}") | |
st.sidebar.text(f"Database: {DATABASE_NAME}") | |
st.sidebar.text(f"Container: {CONTAINER_NAME}") |