import streamlit as st from azure.cosmos import CosmosClient, PartitionKey, exceptions import os import pandas as pd import traceback # For detailed error stacks st.set_page_config(layout="wide") # Cosmos DB configuration ENDPOINT = "https://acae-afd.documents.azure.com:443/" SUBSCRIPTION_ID = "003fba60-5b3f-48f4-ab36-3ed11bc40816" 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 exceptions.CosmosHttpResponseError as e: return False, f"HTTP error occurred: {str(e)}. Status code: {e.status_code}" except Exception as e: return False, f"An unexpected error occurred: {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 exceptions.CosmosHttpResponseError as e: error_message = f"HTTP error occurred: {str(e)}. Status code: {e.status_code}" return False, error_message except Exception as e: error_message = f"An unexpected error occurred: {str(e)}" return False, error_message def fetch_all_records(): try: query = "SELECT * FROM c" items = list(container.query_items(query=query, enable_cross_partition_query=True)) return pd.DataFrame(items) except exceptions.CosmosHttpResponseError as e: st.error(f"HTTP error occurred while fetching records: {str(e)}. Status code: {e.status_code}") return pd.DataFrame() except Exception as e: st.error(f"An unexpected error occurred while fetching records: {str(e)}") return pd.DataFrame() def update_record(updated_record): try: container.upsert_item(body=updated_record) # Upsert updates the item if it exists return True, f"Record with id {updated_record['id']} successfully updated." except exceptions.CosmosHttpResponseError as e: return False, f"HTTP error occurred: {str(e)}. Status code: {e.status_code}" except Exception as e: return False, f"An unexpected error occurred: {traceback.format_exc()}" # Streamlit app st.title("🌟 Cosmos DB Record Management") # Initialize session state for selected records if 'selected_records' not in st.session_state: st.session_state.selected_records = [] # 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 # Use the predefined Key instead of asking for user input 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("Invalid key. Please check your environment variables.") else: # Initialize Cosmos DB client try: client = CosmosClient(ENDPOINT, credential=st.session_state.primary_key) database = client.get_database_client(DATABASE_NAME) container = database.get_container_client(CONTAINER_NAME) except exceptions.CosmosHttpResponseError as e: st.error(f"Failed to connect to Cosmos DB. HTTP error: {str(e)}. Status code: {e.status_code}") st.stop() except Exception as e: st.error(f"An unexpected error occurred while connecting to Cosmos DB: {str(e)}") st.stop() # Fetch and display all records st.subheader("📊 All Records") df = fetch_all_records() if df.empty: st.write("No records found in the database.") else: st.write("Records:") for index, row in df.iterrows(): col1, col2 = st.columns([5, 1]) with col1: # Display all fields in the listing st.write(f"ID: {row['id']}, Name: {row['name']}, Document: {row['document']}, " f"Evaluation Text: {row['evaluationText']}, Evaluation Score: {row['evaluationScore']}") with col2: key = f"select_{row['id']}" if st.button("Select", key=key): st.session_state.selected_record = row.to_dict() # Display selected record for editing if 'selected_record' in st.session_state and st.session_state.selected_record: selected_record = st.session_state.selected_record st.subheader(f"Editing Record - ID: {selected_record['id']}") # Editable fields prefilled with the selected record updated_name = st.text_input("Name", value=selected_record['name']) updated_document = st.text_area("Document", value=selected_record['document']) updated_evaluation_text = st.text_area("Evaluation Text", value=selected_record['evaluationText']) updated_evaluation_score = st.text_input("Evaluation Score", value=str(selected_record['evaluationScore'])) # Update button if st.button("🔄 Update Record"): updated_record = { "id": selected_record['id'], "name": updated_name, "document": updated_document, "evaluationText": updated_evaluation_text, "evaluationScore": updated_evaluation_score # Now treated as a string } success, message = update_record(updated_record) if success: st.success(message) st.session_state.selected_record = updated_record # Update the session state with new values else: st.error(message) # Input fields for new record st.subheader("📝 Enter New Record Details") new_id = st.text_input("ID") new_name = st.text_input("Name") new_document = st.text_area("Document") new_evaluation_text = st.text_area("Evaluation Text") new_evaluation_score = st.text_input("Evaluation Score") # Now treated as a string col1, col2 = st.columns(2) # Insert Record button with col1: if st.button("💾 Insert Record"): record = { "id": new_id, "name": new_name, "document": new_document, "evaluationText": new_evaluation_text, "evaluationScore": new_evaluation_score # Insert as a string } success, response = insert_record(record) if success: st.success("✅ Record inserted successfully!") st.json(response) else: st.error(f"❌ Failed to insert record: {response}") st.rerun() # Call Procedure button with col2: if st.button("🔧 Call Procedure"): record = { "id": new_id, "name": new_name, "document": new_document, "evaluationText": new_evaluation_text, "evaluationScore": new_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.session_state.selected_records.clear() # Clear selected records on logout st.session_state.selected_record = None # Clear selected record 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}")