Update app.py

app.py

@@ -806,91 +806,67 @@ 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 Document':
-                            st.markdown("# 🧬 Clone functionality - Copy wisely, paste precisely")
+                            # 🧬 Clone Document per record
                             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', '')}")
-                                                                            
-                                clone_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..."):
-    
-                                        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
-                                        ))
-                                        
-                                        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
-                                            ))
-                                        
+                                    cloned_doc = doc.copy()
+                                    # Generate a unique ID
+                                    cloned_doc['id'] = generate_unique_id()
+                                    cloned_doc['name'] = generate_unique_id()
+                                    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)
+                                        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':
-                            Label = '# 🆕 New Record view - Every new record is a new beginning'
-                            st.markdown(Label)
-
-                            if st.button("➕ Create New Blank Document"):
-                                (container)
-
-        
+                            # 🆕 New Record
                             st.markdown("#### Create a new document:")
-                            #if st.button("🤖 Insert Auto-Generated Record"):
-                            success, message = save_or_clone_to_cosmos_db(container, 
-                                                                       query="Auto-generated", 
-                                                                       response="This is an auto-generated record.")
-                            if success:
-                                st.success(message)
-                                #st.rerun()
+                            if st.button("🤖 Insert Auto-Generated Record"):
+                                success, message = save_or_clone_to_cosmos_db(container, query="Auto-generated", response="This is an auto-generated record.")
+                                if success:
+                                    st.success(message)
+                                    st.rerun()
+                                else:
+                                    st.error(message)
                             else:
-                                st.error(message)
-                            #else:
-                                
-                            new_id = st.text_input("ID", value=generate_unique_id(), key='new_id')
-                            new_doc_str = st.text_area("Document Content (in JSON format)", 
-                                                     value='{}', height=300)
-                            if st.button("➕ Create New Document"):
-                                try:
-                                    new_doc = json.loads(new_doc_str)
-                                    new_doc['id'] = new_id
-                                    success, message = insert_record(container, new_doc)
-                                    if success:
-                                        st.success(f"New document created with id: {new_doc['id']} 🎉")
-                                        st.session_state.selected_document_id = new_doc['id']
-                                        #st.rerun()
-                                    else:
-                                        st.error(message)
-                                except json.JSONDecodeError as e:
-                                    st.error(f"Invalid JSON: {str(e)} 🚫")
-                                
-                    else:
-                        st.sidebar.info("No documents found in this container. 📭")
-
+                                new_id = st.text_input("ID", value=generate_unique_id(), key='new_id')
+                                new_doc_str = st.text_area("Document Content (in JSON format)", value='{}', height=300)
+                                if st.button("➕ Create New Document"):
+                                    try:
+                                        new_doc = json.loads(new_doc_str)
+                                        new_doc['id'] = new_id  # Use the provided ID
+                                        success, message = insert_record(container, new_doc)
+                                        if success:
+                                            st.success(f"New document created with id: {new_doc['id']} 🎉")
+                                            st.session_state.selected_document_id = new_doc['id']
+                                            # Switch to 'Show as Edit and Save' mode
+                                            st.rerun()
+                                        else:
+                                            st.error(message)
+                                    except json.JSONDecodeError as e:
+                                        st.error(f"Invalid JSON: {str(e)} 🚫")
 
             
             st.subheader(f"📊 Container: {st.session_state.selected_container}")