Create app.py

app.py

@@ -0,0 +1,204 @@
+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.number_input("Evaluation Score", value=selected_record['evaluationScore'], min_value=0, max_value=100)
+
+        # 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
+            }
+
+            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.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": new_id,
+                "name": new_name,
+                "document": new_document,
+                "evaluationText": new_evaluation_text,
+                "evaluationScore": new_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}")
+            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}")