Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from azure.cosmos import CosmosClient, PartitionKey
+from azure.cosmos import CosmosClient, PartitionKey, exceptions
 import os
 import pandas as pd
 
@@ -17,8 +17,10 @@ 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, str(e)
+        return False, f"An unexpected error occurred: {str(e)}"
 
 def call_stored_procedure(record):
     try:
@@ -28,25 +30,35 @@ def call_stored_procedure(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"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}"
+        error_message = f"An unexpected error occurred: {str(e)}"
         return False, error_message
 
 def fetch_all_records():
-    query = "SELECT * FROM c"
-    items = list(container.query_items(query=query, enable_cross_partition_query=True))
-    return pd.DataFrame(items)
+    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 delete_record(name, id):
     try:
         container.delete_item(item=id, partition_key=id)
         return True, f"Successfully deleted record with name: {name} and id: {id}"
+    except exceptions.CosmosResourceNotFoundError:
+        return False, f"Record with id {id} not found. It may have been already deleted."
+    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"Error deleting record: {str(e)}"
+        return False, f"An unexpected error occurred: {str(e)}"
 
 # Streamlit app
 st.title("🌟 Cosmos DB Record Management")
@@ -71,31 +83,46 @@ if not st.session_state.logged_in:
             st.error("Invalid key. Please check your environment variables.")
 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)
+    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()
     
-    # Add a checkbox column to the dataframe
-    for index, row in df.iterrows():
-        key = f"select_{row['id']}"
-        if st.checkbox("Select", key=key):
-            if row.to_dict() not in st.session_state.selected_records:
-                st.session_state.selected_records.append(row.to_dict())
-        else:
-            st.session_state.selected_records = [r for r in st.session_state.selected_records if r['id'] != row['id']]
-        
-        st.write(f"ID: {row['id']}, Name: {row['name']}")
-        if st.button(f"Delete {row['name']}", key=f"delete_{row['id']}"):
-            success, message = delete_record(row['name'], row['id'])
-            if success:
-                st.success(message)
-            else:
-                st.error(message)
-            st.rerun()
+    if df.empty:
+        st.write("No records found in the database.")
+    else:
+        for index, row in df.iterrows():
+            col1, col2, col3 = st.columns([3, 1, 1])
+            
+            with col1:
+                st.write(f"ID: {row['id']}, Name: {row['name']}")
+            
+            with col2:
+                key = f"select_{row['id']}"
+                if st.checkbox("Select", key=key, value=row.to_dict() in st.session_state.selected_records):
+                    if row.to_dict() not in st.session_state.selected_records:
+                        st.session_state.selected_records.append(row.to_dict())
+                else:
+                    st.session_state.selected_records = [r for r in st.session_state.selected_records if r['id'] != row['id']]
+            
+            with col3:
+                if st.button(f"Delete", key=f"delete_{row['id']}"):
+                    success, message = delete_record(row['name'], row['id'])
+                    if success:
+                        st.success(message)
+                        st.rerun()
+                    else:
+                        st.error(message)
 
     # Display selected records
     st.subheader("Selected Records")