Update app.py

app.py

@@ -809,69 +809,69 @@ def main():
                                                     else:
                                                         st.error("Failed to clone document. Please try again.")
 
-                                                                                
-                            elif selected_view == 'Clone Document':
-                                st.markdown("# 🧬 Clone functionality - Copy wisely, paste precisely")
-                                st.markdown("#### Clone a document:")
-    
-                                # Displaying document IDs and clone buttons
-                                for idx, doc in enumerate(documents_to_display):
-                                    st.markdown(f"##### Document ID: {doc.get('id', '')}")
-                                    
-                                    # Button to initiate cloning of the document
-                                    if st.button("📄 Clone Document", key=f'clone_button_{idx}'):
-                                        with st.spinner("Cloning document..."):
-                                            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['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"):
+                                                                            
+                        elif selected_view == 'Clone Document':
+                            st.markdown("# 🧬 Clone functionality - Copy wisely, paste precisely")
+                            st.markdown("#### Clone a document:")
+
+                            # Displaying document IDs and clone buttons
+                            for idx, doc in enumerate(documents_to_display):
+                                st.markdown(f"##### Document ID: {doc.get('id', '')}")
+                                
+                                # Button to initiate cloning of the document
+                                if st.button("📄 Clone Document", key=f'clone_button_{idx}'):
+                                    with st.spinner("Cloning 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)} 🚫")
+                                            # Copy the document and reset system-managed fields
+                                            cloned_doc = doc.copy()
+                                            cloned_doc['id'] = 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)} 🚫")