Update app.py

app.py

@@ -1,51 +1,25 @@
 import streamlit as st
 from azure.cosmos import CosmosClient
 import os
-import traceback
-import json
 
 # 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 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, partition_key_type):
-    try:
-        if partition_key_type == "first_word":
-            partition_key = long_text.split()[0]
-        elif partition_key_type == "first_two_words":
-            partition_key = " ".join(long_text.split()[:2])
-        elif partition_key_type == "first_line":
-            partition_key = long_text.split('\n')[0].strip()
-        elif partition_key_type == "json":
-            # Assume the text is a JSON string and extract a specific field
-            json_data = json.loads(long_text)
-            partition_key = json_data.get("partitionKey", "default")
-        else:
-            partition_key = "default"
-
-        result = container.scripts.execute_stored_procedure(
-            sproc='processQTPrompts',  # Make sure this matches your stored procedure name
-            params=[long_text],
-            partition_key=partition_key
-        )
-        return True, result
-    except Exception as e:
-        error_msg = f"Error: {str(e)}\n\nTraceback:\n{traceback.format_exc()}"
-        return False, error_msg
-
 # Streamlit app
-st.title("🌟 Cosmos DB Record Insertion and Stored Procedure Caller")
+st.title("🌟 Cosmos DB Record Insertion")
 
 # Login section
 if 'logged_in' not in st.session_state:
@@ -53,7 +27,8 @@ 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", value=Key)
+    #input_key = st.text_input("Enter your Cosmos DB Primary Key", type="password")
+    input_key = Key
     if st.button("🚀 Login"):
         if input_key:
             st.session_state.primary_key = input_key
@@ -67,52 +42,28 @@ else:
     database = client.get_database_client(DATABASE_NAME)
     container = database.get_container_client(CONTAINER_NAME)
 
-    # Create tabs for different operations
-    tab1, tab2 = st.tabs(["Insert Record", "Call Stored Procedure"])
+    # Input fields
+    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")
 
-    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")
-
-        if st.button("💾 Insert Record"):
-            record = {
-                "id": id,
-                "name": name,
-                "age": age,
-                "city": city
-            }
-            
-            success, response = insert_record(container, record)
-            if success:
-                st.success("✅ Record inserted successfully!")
-                st.json(response)
-            else:
-                st.error(f"❌ Failed to insert record: {response}")
-
-    with tab2:
-        st.subheader("📝 Enter Long Text for Stored Procedure")
-        long_text = st.text_area("Long Text", height=200)
-
-        partition_key_type = st.selectbox(
-            "Select Partition Key Type",
-            ["first_word", "first_two_words", "first_line", "json", "default"]
-        )
-
-        if st.button("🚀 Call Stored Procedure"):
-            if long_text:
-                with st.spinner('Calling stored procedure...'):
-                    success, result = call_stored_procedure(container, long_text, partition_key_type)
-                    if success:
-                        st.success("✅ Stored procedure executed successfully!")
-                        st.json(result)
-                    else:
-                        st.error("❌ Failed to execute stored procedure")
-                        st.text(result)  # This will show the detailed error message
-            else:
-                st.warning("Please enter some text before calling the stored procedure.")
+    # Submit button
+    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}")
 
     # Logout button
     if st.button("🚪 Logout"):