Update app.py

app.py

@@ -809,9 +809,75 @@ 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"):
+                                        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)} 🚫")
+
+
+
+                        
 
-                        elif selected_view == 'Clone Document':
+                        elif selected_view == 'Clone Old Document':
                             if st.button("📄 Clone Document"):
                                 clone_id = st.text_input("Enter Document ID to Clone")
                                 if clone_id: