Update app.py

app.py

@@ -806,78 +806,60 @@ def main():
                                         if st.button("🤖 Run AI", key=f'run_with_ai_button_{idx}'):
                                             search_glossary(json.dumps(editable_doc, indent=2))
 
-                                                       
-                        elif selected_view == 'Clone Document2':
-                            # 🧬 Clone Document per record
-                            st.markdown("#### Clone a document:")
-                            for idx, doc in enumerate(documents_to_display):
-                                st.markdown(f"##### Document ID: {doc.get('id', '')}")
-                                if st.button("📄 Clone Document", key=f'clone_button_{idx}'):
-                                    cloned_doc = doc.copy()
-                                    
-                                    # Generate a unique ID
-                                    st.code(cloned_doc, )
-                                    cloned_doc['id'] = generate_unique_id()
-                                    cloned_doc['name'] = generate_unique_id()
-                                    cloned_doc_str = st.text_area("Cloned Document Content (in JSON format) - Update name please to make it unique before saving!", value=cloned_doc, height=300)
-
-                                    st.markdown("#### Edit Cloned Document and Name Your Clone:")
-                                    if st.button("💾 Save Cloned Document"):
-                                        success, message = insert_record(container, cloned_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)
                                             
                         elif selected_view == 'Clone Document':
-                            # 🧬 Clone Document per record
                             st.markdown("#### Clone a document:")
                             
                             for idx, doc in enumerate(documents_to_display):
                                 st.markdown(f"##### Document ID: {doc.get('id', '')}")
                                 
                                 if st.button("📄 Clone Document", key=f'clone_button_{idx}'):
+                                    # Create copy of original document
                                     cloned_doc = doc.copy()
                                     
-                                    # Generate new unique IDs for both 'id' and 'name'
-                                    cloned_doc['id'] = generate_unique_id()
-                                    cloned_doc['name'] = generate_unique_id()
+                                    # Generate new unique IDs
+                                    new_id = generate_unique_id()
+                                    new_name = generate_unique_id()
                                     
-                                    # Display the cloned document in JSON format with a prompt to edit
-                                    cloned_doc_str = st.text_area("Cloned Document Content (in JSON format) - You can edit before saving!", 
-                                                                  value=str(cloned_doc), height=300)
-                                    cloned_doc_str = cloned_doc_str.replace("'id': None,", "'id': 'test123',")
-                                    cloned_doc_str = cloned_doc_str.replace("'name': None,", "'name': 'test123',")
-                                    st.write(cloned_doc_str)
-
+                                    # Update ID and name fields at root level
+                                    cloned_doc['id'] = new_id
+                                    if 'name' in cloned_doc:
+                                        cloned_doc['name'] = new_name
+                                    
+                                    # Update nested ID and name fields if they exist
+                                    def update_nested_ids(obj):
+                                        if isinstance(obj, dict):
+                                            if 'id' in obj:
+                                                obj['id'] = new_id
+                                            if 'name' in obj:
+                                                obj['name'] = new_name
+                                            for value in obj.values():
+                                                update_nested_ids(value)
+                                        elif isinstance(obj, list):
+                                            for item in obj:
+                                                update_nested_ids(item)
                                     
-                                    st.markdown("#### Edit Cloned Document and Name Your Clone:")
+                                    update_nested_ids(cloned_doc)
+                                    
+                                    # Display editable JSON
+                                    edited_doc_str = st.text_area(
+                                        "Edit cloned document before saving:",
+                                        value=json.dumps(cloned_doc, indent=2),
+                                        height=300
+                                    )
                                     
-                                    # Allow user to save the cloned document
                                     if st.button("💾 Save Cloned Document"):
-                                        # Convert the string back to dictionary (if edited)
                                         try:
-                                            # Assuming that cloned_doc_str will be valid JSON
-                                            cloned_doc = json.loads(cloned_doc_str)
-                                            success, message = insert_record(container, cloned_doc)
+                                            final_doc = json.loads(edited_doc_str)
+                                            success, message = insert_record(container, final_doc)
                                             
                                             if success:
-                                                st.success(f"Cloned document saved with id: {cloned_doc['id']} 🎉")
-                                                st.session_state.selected_document_id = cloned_doc['id']
-                                                st.session_state.clone_mode = False
-                                                st.session_state.cloned_doc = None
-                                                st.session_state.cloned_doc_str = ''
+                                                st.success(f"Cloned document saved successfully with ID: {final_doc['id']}")
                                                 st.rerun()
                                             else:
-                                                st.error(message)
+                                                st.error(f"Failed to save clone: {message}")
                                         except json.JSONDecodeError as e:
-                                            st.error(f"Error decoding the document. Please ensure it's valid JSON: {e}")
-
+                                            st.error(f"Invalid JSON format: {str(e)}")
 
                     
                         elif selected_view == 'New Record':