Update app.py

app.py

@@ -1,7 +1,6 @@
 import streamlit as st
 from azure.cosmos import CosmosClient, PartitionKey
 import os
-import re
 
 # Cosmos DB configuration
 ENDPOINT = "https://acae-afd.documents.azure.com:443/"
@@ -10,7 +9,6 @@ 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 create_stored_procedure(container):
     stored_procedure_definition = {
@@ -28,8 +26,8 @@ def create_stored_procedure(container):
                 var prompt = prompts[i].trim();
                 if (prompt.startsWith('QT ')) {
                     var querySpec = {
-                        query: "SELECT * FROM c WHERE c.prompt = @prompt",
-                        parameters: [{ name: "@prompt", value: prompt }]
+                        query: "SELECT * FROM c WHERE c.id = @id",
+                        parameters: [{ name: "@id", value: prompt }]
                     };
 
                     var isAccepted = container.queryDocuments(
@@ -53,7 +51,7 @@ def create_stored_procedure(container):
                             } else {
                                 // Create new record
                                 var newItem = {
-                                    prompt: prompt,
+                                    id: prompt,
                                     occurrenceCount: 1,
                                     evaluation: ""
                                 };
@@ -86,10 +84,12 @@ def ensure_stored_procedure_exists(container):
         create_stored_procedure(container)
 
 def process_qt_prompts(container, prompt_text):
+    # Use a dummy partition key value
+    partition_key = "dummy_partition_key"
     return container.scripts.execute_stored_procedure(
         sproc='processQTPrompts',
         params=[prompt_text],
-        partition_key=None
+        partition_key=partition_key
     )
 
 # Streamlit app
@@ -101,13 +101,12 @@ if 'logged_in' not in st.session_state:
 
 if not st.session_state.logged_in:
     st.subheader("🔐 Login")
-    #input_key = st.text_input("Enter your Cosmos DB Primary Key", type="password")
-    input_key = Key
+    input_key = st.text_input("Enter your Cosmos DB Primary Key", type="password")
     if st.button("🚀 Login"):
         if input_key:
             st.session_state.primary_key = input_key
             st.session_state.logged_in = True
-            st.rerun()
+            st.experimental_rerun()
         else:
             st.error("Please enter a valid key")
 else:
@@ -126,11 +125,14 @@ else:
 
     # Submit button
     if st.button("🚀 Process QT Prompts"):
-        results = process_qt_prompts(container, qt_prompts)
-        
-        # Display results in a dataframe
-        df_data = [{"Prompt": item['prompt'], "Occurrence Count": item['occurrenceCount']} for item in results]
-        st.dataframe(df_data)
+        try:
+            results = process_qt_prompts(container, qt_prompts)
+            
+            # Display results in a dataframe
+            df_data = [{"Prompt": item['id'], "Occurrence Count": item['occurrenceCount']} for item in results]
+            st.dataframe(df_data)
+        except Exception as e:
+            st.error(f"An error occurred: {str(e)}")
 
     # Logout button
     if st.button("🚪 Logout"):