Update app.py

app.py

@@ -831,90 +831,90 @@ def main():
                                 # Button to initiate cloning of the document
                                 if st.button("📄 Clone Document", key=f'clone_button_{idx}'):
                                     with st.spinner("Cloning document..."):
+        
+                                        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['name'] = 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)} 😱")
+        
+                                        
+                                        #try:
+                                        # Copy the document and reset system-managed fields
+                                        cloned_doc = doc.copy()
+                                        cloned_doc['id'] = generate_unique_id()  # Generate new unique ID
+                                        cloned_doc['name'] = generate_unique_id()  # Generate new unique ID
+                                        cloned_doc['createdAt'] = datetime.utcnow().isoformat()  # Set new creation time
+                                        cloned_doc['_rid'] = None  # Clear system-managed fields
+                                        cloned_doc['_self'] = None
+                                        cloned_doc['_etag'] = None
+                                        cloned_doc['_attachments'] = None
+                                        cloned_doc['_ts'] = None
     
-                                    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['name'] = 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']} 🎉")
+                                        # Save the cloned document
                                         
-                                        # Refresh the documents in session state
+                                        response = container.create_item(body=cloned_doc)
+                                        st.success(f"Cloned document saved successfully with ID: {cloned_doc['id']} 🎉")
+    
+                                        # Refresh session state documents to show the new cloned document
                                         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)} 😱")
+                                            ))
     
+                                        #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)} 😱")
+    
+                                # If clone mode is activated, allow user to edit the cloned document
+                                if st.session_state.get('clone_mode', False):
+                                    st.markdown("#### Edit Cloned Document and Name Your Clone:")
+                                    cloned_doc_str = st.text_area(
+                                        "Cloned Document Content (in JSON format) - Update name before saving!", 
+                                        value=st.session_state.cloned_doc_str, 
+                                        height=300
+                                    )
                                     
-                                    #try:
-                                    # Copy the document and reset system-managed fields
-                                    cloned_doc = doc.copy()
-                                    cloned_doc['id'] = generate_unique_id()  # Generate new unique ID
-                                    cloned_doc['name'] = generate_unique_id()  # Generate new unique ID
-                                    cloned_doc['createdAt'] = datetime.utcnow().isoformat()  # Set new creation time
-                                    cloned_doc['_rid'] = None  # Clear system-managed fields
-                                    cloned_doc['_self'] = None
-                                    cloned_doc['_etag'] = None
-                                    cloned_doc['_attachments'] = None
-                                    cloned_doc['_ts'] = None
-
-                                    # Save the cloned document
-                                    
-                                    response = container.create_item(body=cloned_doc)
-                                    st.success(f"Cloned document saved successfully with ID: {cloned_doc['id']} 🎉")
-
-                                    # Refresh session state documents to show the new cloned document
-                                    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)} 😱")
-
-                            # If clone mode is activated, allow user to edit the cloned document
-                            if st.session_state.get('clone_mode', False):
-                                st.markdown("#### Edit Cloned Document and Name Your Clone:")
-                                cloned_doc_str = st.text_area(
-                                    "Cloned Document Content (in JSON format) - Update name before saving!", 
-                                    value=st.session_state.cloned_doc_str, 
-                                    height=300
-                                )
-                                
-                                # Save the edited cloned document
-                                if st.button("💾 Save Cloned Document"):
-                                    try:
-                                        new_doc = json.loads(cloned_doc_str)
-                                        success, message = insert_record(container, new_doc)
-                                        if success:
-                                            st.success(f"Cloned document saved with id: {new_doc['id']} 🎉")
-                                            st.session_state.selected_document_id = new_doc['id']
-                                            st.session_state.clone_mode = False
-                                            st.session_state.cloned_doc = None
-                                            st.session_state.cloned_doc_str = ''
-                                            st.rerun()
-                                        else:
-                                            st.error(message)
-                                    except json.JSONDecodeError as e:
-                                        st.error(f"Invalid JSON: {str(e)} 🚫")
+                                    # Save the edited cloned document
+                                    if st.button("💾 Save Cloned Document"):
+                                        try:
+                                            new_doc = json.loads(cloned_doc_str)
+                                            success, message = insert_record(container, new_doc)
+                                            if success:
+                                                st.success(f"Cloned document saved with id: {new_doc['id']} 🎉")
+                                                st.session_state.selected_document_id = new_doc['id']
+                                                st.session_state.clone_mode = False
+                                                st.session_state.cloned_doc = None
+                                                st.session_state.cloned_doc_str = ''
+                                                st.rerun()
+                                            else:
+                                                st.error(message)
+                                        except json.JSONDecodeError as e:
+                                            st.error(f"Invalid JSON: {str(e)} 🚫")