Update app.py

app.py

@@ -1,107 +1,35 @@
 import streamlit as st
-from azure.cosmos import CosmosClient, PartitionKey
+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 create_stored_procedure(container):
-    stored_procedure_definition = {
-        'id': 'processQTPrompts',
-        'body': '''
-        function processQTPrompts(promptText) {
-            var context = getContext();
-            var container = context.getCollection();
-            var response = context.getResponse();
-
-            var prompts = promptText.split('\\n');
-            var results = [];
-
-            for (var i in prompts) {
-                var prompt = prompts[i].trim();
-                if (prompt.startsWith('QT ')) {
-                    var querySpec = {
-                        query: "SELECT * FROM c WHERE c.id = @id",
-                        parameters: [{ name: "@id", value: prompt }]
-                    };
-
-                    var isAccepted = container.queryDocuments(
-                        container.getSelfLink(),
-                        querySpec,
-                        function (err, items, responseOptions) {
-                            if (err) throw err;
-
-                            if (items.length > 0) {
-                                // Update existing record
-                                var item = items[0];
-                                item.occurrenceCount++;
-                                container.replaceDocument(
-                                    item._self,
-                                    item,
-                                    function (err, replacedItem) {
-                                        if (err) throw err;
-                                        results.push(replacedItem);
-                                    }
-                                );
-                            } else {
-                                // Create new record
-                                var newItem = {
-                                    id: prompt,
-                                    occurrenceCount: 1,
-                                    evaluation: ""
-                                };
-                                container.createDocument(
-                                    container.getSelfLink(),
-                                    newItem,
-                                    function (err, createdItem) {
-                                        if (err) throw err;
-                                        results.push(createdItem);
-                                    }
-                                );
-                            }
-                        }
-                    );
-
-                    if (!isAccepted) throw new Error("The query was not accepted by the server.");
-                }
-            }
-
-            response.setBody(results);
-        }
-        '''
-    }
-    container.scripts.create_stored_procedure(body=stored_procedure_definition)
-
-def ensure_stored_procedure_exists(container):
-    try:
-        container.scripts.get_stored_procedure('processQTPrompts')
-    except:
-        create_stored_procedure(container)
-
-def process_qt_prompts(container, prompt_text):
-    # Use the first prompt as the partition key
-    first_prompt = prompt_text.split('\n')[0].strip()
-    return container.scripts.execute_stored_procedure(
-        sproc='processQTPrompts',
-        params=[prompt_text],
-        partition_key=first_prompt
-    )
-
-def insert_record(container, record):
+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(container, long_text):
+    try:
+        result = container.scripts.execute_stored_procedure(
+            sproc='processQTPrompts',  # Replace with your actual stored procedure name
+            params=[long_text],
+            partition_key=None  # You may need to adjust this based on your container's partition key
+        )
+        return True, result
+    except Exception as e:
+        return False, str(e)
+
 # Streamlit app
-st.title("🌟 Cosmos DB Record Insertion and QT Prompt Processor")
+st.title("🌟 Cosmos DB Record Insertion and Stored Procedure")
 
 # Login section
 if 'logged_in' not in st.session_state:
@@ -124,48 +52,47 @@ else:
     database = client.get_database_client(DATABASE_NAME)
     container = database.get_container_client(CONTAINER_NAME)
 
-    # Ensure the stored procedure exists
-    ensure_stored_procedure_exists(container)
+    # Input fields for record insertion
+    st.subheader("📝 Enter Record Details")
+    id = st.text_input("ID")
+    name = st.text_input("Name")
+    age = st.number_input("Age", min_value=0, max_value=150)
+    city = st.text_input("City")
+
+    # Submit button for record insertion
+    if st.button("💾 Insert Record"):
+        record = {
+            "id": id,
+            "name": name,
+            "age": age,
+            "city": city
+        }
+        
+        success, response = insert_record(record)
+        if success:
+            st.success("✅ Record inserted successfully!")
+            st.json(response)
+        else:
+            st.error(f"❌ Failed to insert record: {response}")
 
-    # Tab for different operations
-    tab1, tab2 = st.tabs(["Insert Record", "Process QT Prompts"])
+    # Separator
+    st.markdown("---")
 
-    with tab1:
-        st.subheader("📝 Enter Record Details")
-        id = st.text_input("ID")
-        name = st.text_input("Name")
-        age = st.number_input("Age", min_value=0, max_value=150)
-        city = st.text_input("City")
+    # Input field for long text
+    st.subheader("📝 Enter Long Text for Stored Procedure")
+    long_text = st.text_area("Long Text", height=200)
 
-        if st.button("💾 Insert Record"):
-            record = {
-                "id": id,
-                "name": name,
-                "age": age,
-                "city": city
-            }
-            
-            success, response = insert_record(container, record)
+    # Button to call stored procedure
+    if st.button("🚀 Call Stored Procedure"):
+        if long_text:
+            success, result = call_stored_procedure(container, long_text)
             if success:
-                st.success("✅ Record inserted successfully!")
-                st.json(response)
+                st.success("✅ Stored procedure executed successfully!")
+                st.json(result)
             else:
-                st.error(f"❌ Failed to insert record: {response}")
-
-    with tab2:
-        st.subheader("📝 Enter QT Prompts")
-        default_text = "QT Crystal finders: Bioluminescent crystal caverns, quantum-powered explorers, prismatic hues, alien planet\nQT robot art: Cybernetic metropolis, sentient androids, rogue AI, neon-infused palette\nQT the Lava girl: Volcanic exoplanet, liquid metal rivers, heat-immune heroine, molten metallic palette"
-        qt_prompts = st.text_area("QT Prompts", value=default_text, height=300)
-
-        if st.button("🚀 Process QT Prompts"):
-            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)}")
+                st.error(f"❌ Failed to execute stored procedure: {result}")
+        else:
+            st.warning("Please enter some text before calling the stored procedure.")
 
     # Logout button
     if st.button("🚪 Logout"):