Update app.py

app.py

@@ -820,30 +820,31 @@ def main():
                                     st.session_state.cloned_doc = cloned_doc
                                     st.session_state.cloned_doc_str = json.dumps(cloned_doc, indent=2)
                                     st.session_state.clone_mode = True
-                                    st.rerun()
-                            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 please to make it unique before saving!", value=st.session_state.cloned_doc_str, height=300)
-                                if st.button("💾 Save Cloned Document"):
-                                    try:
-                                        new_doc = json.loads(cloned_doc_str)
-
-                                        # Generate new name and id
-                                        new_doc['id'] = generate_unique_id()
-                                        new_doc['name'] = generate_unique_id()
-                                        
-                                        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)} 🚫")
+                                                                            
+                                    new_doc = json.loads(cloned_doc_str)
+                                    cloned_doc_str = st.text_area("Cloned Document Content (in JSON format) - Update name please to make it unique before saving!", value=new_doc, height=300)
+
+                                    st.markdown("#### Edit Cloned Document and Name Your Clone:")
+                                    if st.button("💾 Save Cloned Document"):
+                                        try:
+                                            new_doc = json.loads(cloned_doc_str)
+    
+                                            # Generate new name and id
+                                            new_doc['id'] = generate_unique_id()
+                                            new_doc['name'] = generate_unique_id()
+                                            
+                                            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 == 'New Record':
                             # 🆕 New Record