Update app.py

app.py

@@ -263,6 +263,7 @@ def save_or_clone_to_cosmos_db(container, query=None, response=None, clone_id=No
 
             if clone_id:
                 try:
+                    
                     existing_doc = container.read_item(item=clone_id, partition_key=clone_id)
                     new_doc = {
                         'id': new_id,
@@ -312,55 +313,6 @@ def save_or_clone_to_cosmos_db(container, query=None, response=None, clone_id=No
     return None
 
     
-# 💾 Save or clone to Cosmos DB - Because every document deserves a twin
-def save_or_clone_to_cosmos_db2(container, query=None, response=None, clone_id=None):
-    try:
-        if not container:
-            st.error("Cosmos DB container is not initialized.")
-            return
-
-        # Generate a new unique ID
-        new_id = str(uuid.uuid4())
-
-        # Create a new document
-        if clone_id:
-            # If clone_id is provided, we're cloning an existing document
-            try:
-                existing_doc = container.read_item(item=clone_id, partition_key=clone_id)
-                new_doc = existing_doc.copy()
-                new_doc['id'] = new_id
-                new_doc['cloned_from'] = clone_id
-                new_doc['cloned_at'] = datetime.utcnow().isoformat()
-            except exceptions.CosmosResourceNotFoundError:
-                st.error(f"Document with ID {clone_id} not found for cloning.")
-                return
-        else:
-            # If no clone_id, we're creating a new document
-            new_doc = {
-                'id': new_id,
-                'query': query,
-                'response': response,
-                'created_at': datetime.utcnow().isoformat()
-            }
-
-        # Insert the new document
-        container.create_item(body=new_doc)
-
-        st.success(f"{'Cloned' if clone_id else 'New'} document saved successfully with ID: {new_id}")
-
-        # Refresh the documents in the session state
-        st.session_state.documents = list(container.query_items(
-            query="SELECT * FROM c ORDER BY c._ts DESC",
-            enable_cross_partition_query=True
-        ))
-
-        return new_id
-
-    except exceptions.CosmosHttpResponseError as e:
-        st.error(f"Error saving to Cosmos DB: {e}")
-    except Exception as e:
-        st.error(f"An unexpected error occurred: {str(e)}")
-
 # 📦 Archive current container - Packing up data like you're moving to a new digital house
 def archive_current_container(database_name, container_name, client):
     try:
@@ -500,6 +452,68 @@ def display_saved_files_in_sidebar():
                 st.rerun()
 
 
+def clone_record(container, clone_id):
+    try:
+        existing_doc = container.read_item(item=clone_id, partition_key=clone_id)
+        new_doc = existing_doc.copy()
+        new_doc['id'] = generate_unique_id()  # Generate new unique ID with timestamp
+        new_doc['createdAt'] = datetime.utcnow().isoformat()  # Update the creation time
+        new_doc['_rid'] = None  # Reset _rid or any system-managed fields
+        new_doc['_self'] = None
+        new_doc['_etag'] = None
+        new_doc['_attachments'] = None
+        new_doc['_ts'] = None  # Reset timestamp to be updated by Cosmos DB automatically
+
+        # Insert the cloned document
+        response = container.create_item(body=new_doc)
+        st.success(f"Cloned document saved successfully with ID: {new_doc['id']} 🎉")
+        
+        # Refresh the documents in session state
+        st.session_state.documents = list(container.query_items(
+            query="SELECT * FROM c ORDER BY c._ts DESC",
+            enable_cross_partition_query=True
+        ))
+
+    except exceptions.CosmosResourceNotFoundError:
+        st.error(f"Document with ID {clone_id} not found for cloning.")
+    except exceptions.CosmosHttpResponseError as e:
+        st.error(f"HTTP error occurred: {str(e)} 🚨")
+    except Exception as e:
+        st.error(f"An unexpected error occurred: {str(e)} 😱")
+
+
+
+
+def create_new_blank_record(container):
+    try:
+        # Get the structure of the latest document (to preserve schema)
+        latest_doc = container.query_items(query="SELECT * FROM c ORDER BY c._ts DESC", enable_cross_partition_query=True, max_item_count=1)
+        if latest_doc:
+            new_doc_structure = latest_doc[0].copy()
+        else:
+            new_doc_structure = {}
+
+        new_doc = {key: "" for key in new_doc_structure.keys()}  # Set all fields to blank
+        new_doc['id'] = generate_unique_id()  # Generate new unique ID
+        new_doc['createdAt'] = datetime.utcnow().isoformat()  # Set creation time
+
+        # Insert the new blank document
+        response = container.create_item(body=new_doc)
+        st.success(f"New blank document saved successfully with ID: {new_doc['id']} 🎉")
+        
+        # Refresh the documents in session state
+        st.session_state.documents = list(container.query_items(
+            query="SELECT * FROM c ORDER BY c._ts DESC",
+            enable_cross_partition_query=True
+        ))
+
+    except exceptions.CosmosHttpResponseError as e:
+        st.error(f"HTTP error occurred: {str(e)} 🚨")
+    except Exception as e:
+        st.error(f"An unexpected error occurred: {str(e)} 😱")
+
+
+
 
 
                 
@@ -782,6 +796,10 @@ def main():
                                         if st.button("🤖 Run AI", key=f'run_with_ai_button_{idx}'):
                                             search_glossary(json.dumps(editable_doc, indent=2))
                                             if st.button("📄 Clone Document", key=f'clone_button_{idx}'):
+
+
+
+                                                
                                                 with st.spinner("Cloning document..."):
                                                     cloned_id = save_or_clone_to_cosmos_db(container, 
                                                                                          clone_id=doc['id'])
@@ -794,6 +812,10 @@ def main():
                                                         
 
                         elif selected_view == 'Clone Document':
+                            if st.button("📄 Clone Document"):
+                                clone_id = st.text_input("Enter Document ID to Clone")
+                                if clone_id:
+                                    clone_record(container, clone_id)
                             Label = '# 🧬 Clone functionality - Copy wisely, paste precisely'
                             st.markdown(Label)
                             st.markdown("#### Clone a document:")
@@ -834,6 +856,11 @@ def main():
                         elif selected_view == 'New Record':
                             Label = '# 🆕 New Record view - Every new record is a new beginning'
                             st.markdown(Label)
+
+                            if st.button("➕ Create New Blank Document"):
+                                (container)
+
+        
                             st.markdown("#### Create a new document:")
                             #if st.button("🤖 Insert Auto-Generated Record"):
                             success, message = save_or_clone_to_cosmos_db(container,