Update app.py

app.py

@@ -798,165 +798,94 @@ def main():
                                                 display_glossary_entity(term)
                                                 search_glossary(term)
 
-                                    
-                        elif selected_view == 'Show as Run AI - Old':
-                            Label = '#### ✏️ Run AI with wisdom, save with precision'
-                            st.markdown(Label)                            
-                            num_cols = len(documents_to_display)
-                            cols = st.columns(num_cols)
-
-                            for idx, (col, doc) in enumerate(zip(cols, documents_to_display)):
-                                with col:
-                                    
-                                    # st.markdown(f"##### Document ID: {doc.get('id', '')}")
-                                    editable_id = st.text_input("ID", value=doc.get('id', ''), key=f'edit_id_{idx}')
-                                    editable_doc = doc.copy()
-                                    editable_doc.pop('id', None)
-
-                                    # st.markdown(f"##### Document Name: {doc.get('name', '')}")
-                                    editable_id = st.text_input("Name", value=doc.get('name', ''), key=f'edit_name_{idx}')
-                                    editable_doc = doc.copy()
-                                    editable_doc.pop('name', None)
-
-                                    doc_str = st.text_area("Document Content (in JSON format)", 
-                                                         value=json.dumps(editable_doc, indent=2), 
-                                                         height=300, 
-                                                         key=f'doc_str_{idx}')
-
-                                    # 💾🤖 Save and AI operations
-                                    col_ai, col_delete = st.columns(2)
-                                    
-                                    with col_ai:
-                                        if st.button("🤖 Run AI", key=f'run_with_ai_button_{idx}'):
-
-
-                                            
-
-                                            # Experiment to get content fields 🤖🤖🤖🤖🤖🤖🤖🤖🤖🤖🤖🤖🤖
-                                            total_docs = len(documents)
-                                            doc = documents[st.session_state.current_index]
-                                            values_with_space = []
-                                            def extract_values2(obj):
-                                                if isinstance(obj, dict):
-                                                    for k, v in obj.items():
-                                                        extract_values2(v)
-                                                elif isinstance(obj, list):
-                                                    for item in obj:
-                                                        extract_values2(item)
-                                                elif isinstance(obj, str):
-                                                    if ' ' in obj:
-                                                        values_with_space.append(obj)
-                                            extract_values2(doc)
-                                            #st.markdown("#### 🔗 Links for Extracted Texts")
-                                            for term in values_with_space:
-                                                display_glossary_entity(term)
-                                                search_glossary(term)
-                                                
-                                            #content = json.dumps(doc, indent=2)
-                                            #st.markdown(f"```json\n{content}\n```")
-                                            # Experiment to get content fields 🤖🤖🤖🤖🤖🤖🤖🤖🤖🤖🤖🤖🤖
-
-
-
-                                            
-                                            #search_glossary(json.dumps(editable_doc, indent=2))
-
-
-                    
-                    
+                        
                         elif selected_view == 'Clone Document':
-                            st.markdown("#### Clone a document:")
+                            st.markdown("#### 📄 Clone Document (Save As)")
                             
-                            for idx, doc in enumerate(documents_to_display):
-                                st.markdown(f"##### Original Document ID: {doc.get('id', '')}")
-                                
-                                if st.button("📄 Clone Document", key=f'clone_button_{idx}'):
-                                    # Generate unique filename using the provided function
-                                    unique_filename = gen_AI_IO_filename("Clone", doc.get('name', ''))
-                                    
-                                    # Create new document with unique ID and name based on timestamp
-                                    new_doc = {
-                                        'id': unique_filename,
-                                        'name': f"Clone_{unique_filename[:8]}",
-                                        **{k: v for k, v in doc.items() if k not in ['id', 'name', '_rid', '_self', '_etag', '_attachments', '_ts']}
-                                    }
-                                    
-                                    # Show editable preview
-                                    edited_doc = st.text_area(
-                                        "Edit cloned document:",
-                                        value=json.dumps(new_doc, indent=2),
-                                        height=300,
-                                        key=f'edit_clone_{idx}'
-                                    )
-                                    
-                                    col_save, col_update = st.columns(2)
-                                    
-                                    with col_save:
-                                        if st.button("💾 Save Clone", key=f'save_clone_{idx}'):
-                                            try:
-                                                final_doc = json.loads(edited_doc)
-                                                response = container.create_item(body=final_doc)
-                                                if response:
-                                                    st.success(f"New cloned document created with ID: {final_doc['id']}")
-                                                    st.rerun()
-                                                else:
-                                                    st.error("Failed to create new document")
-                                            except Exception as e:
-                                                st.error(f"Error creating document: {str(e)}")
-                                    
-                                    with col_update:
-                                        if st.button("🔄 Update Clone ID/Name", key=f'update_clone_{idx}'):
-                                            try:
-                                                # Generate new unique filename
-                                                new_unique_filename = gen_AI_IO_filename("Clone", doc.get('name', ''))
-                                                current_doc = json.loads(edited_doc)
-                                                current_doc['id'] = new_unique_filename
-                                                current_doc['name'] = f"Clone_{new_unique_filename[:8]}"
-                                                
-                                                # Update the text area with new values
-                                                st.session_state[f'edit_clone_{idx}'] = json.dumps(current_doc, indent=2)
-                                                st.rerun()
-                                            except Exception as e:
-                                                st.error(f"Error updating clone: {str(e)}")
-    
-
-                                            
-                                                
-                        elif selected_view == 'Clone Document - Old':
-                            st.markdown("#### Clone a document:")
+                            total_docs = len(documents)
+                            doc = documents[st.session_state.current_index]
                             
-                            for idx, doc in enumerate(documents_to_display):
-                                st.markdown(f"##### Original Document ID: {doc.get('id', '')}")
-                                
-                                if st.button("📄 Clone Document", key=f'clone_button_{idx}'):
-                                    # Create new document with unique ID and name
-                                    new_doc = {
-                                        'id': str(uuid.uuid4()),
-                                        'name': f"Clone_{str(uuid.uuid4())[:8]}",
-                                        **{k: v for k, v in doc.items() if k not in ['id', 'name', '_rid', '_self', '_etag', '_attachments', '_ts']}
-                                    }
-                                    
-                                    # Show editable preview
-                                    edited_doc = st.text_area(
-                                        "Edit cloned document:",
-                                        value=json.dumps(new_doc, indent=2),
-                                        height=300,
-                                        key=f'edit_clone_{idx}'
-                                    )
-                                    
-                                    if st.button("💾 Save Clone", key=f'save_clone_{idx}'):
-                                        try:
-                                            final_doc = json.loads(edited_doc)
-                                            # Use container.create_item() instead of update_record()
-                                            response = container.create_item(body=final_doc)
-                                            if response:
-                                                st.success(f"New cloned document created with ID: {final_doc['id']}")
-                                                st.rerun()
-                                            else:
-                                                st.error("Failed to create new document")
-                                        except Exception as e:
-                                            st.error(f"Error creating document: {str(e)}")
-
+                            # Display current document info
+                            st.markdown(f"**Original Document ID:** {doc.get('id', '')}")
+                            st.markdown(f"**Original Document Name:** {doc.get('name', '')}")
+                            
+                            # Generate new unique ID and name
+                            unique_filename = gen_AI_IO_filename("Clone", doc.get('name', ''))
+                            new_id = st.text_input("New Document ID", value=unique_filename, key='new_clone_id')
+                            new_name = st.text_input("New Document Name", value=f"Clone_{unique_filename[:8]}", key='new_clone_name')
+                            
+                            # Create new document with all original content except system fields
+                            new_doc = {
+                                'id': new_id,
+                                'name': new_name,
+                                **{k: v for k, v in doc.items() if k not in ['id', 'name', '_rid', '_self', '_etag', '_attachments', '_ts']}
+                            }
+                            
+                            # Show editable preview of the new document
+                            doc_str = st.text_area(
+                                "Edit Document Content (in JSON format)",
+                                value=json.dumps(new_doc, indent=2),
+                                height=300,
+                                key='clone_preview'
+                            )
+                            
+                            col1, col2 = st.columns(2)
+                            
+                            with col1:
+                                if st.button("🔄 Generate New ID/Name", key='regenerate_id'):
+                                    # Generate new unique filename
+                                    new_unique_filename = gen_AI_IO_filename("Clone", doc.get('name', ''))
+                                    st.session_state.new_clone_id = new_unique_filename
+                                    st.session_state.new_clone_name = f"Clone_{new_unique_filename[:8]}"
+                                    st.rerun()
+                            
+                            with col2:
+                                if st.button("💾 Save As New Document", key='save_clone'):
+                                    try:
+                                        # Parse the edited document content
+                                        final_doc = json.loads(doc_str)
+                                        
+                                        # Ensure the new ID and name are used
+                                        final_doc['id'] = new_id
+                                        final_doc['name'] = new_name
+                                        
+                                        # Remove any system fields that might have been copied
+                                        system_fields = ['_rid', '_self', '_etag', '_attachments', '_ts']
+                                        for field in system_fields:
+                                            final_doc.pop(field, None)
+                                        
+                                        # Create the new document
+                                        response = container.create_item(body=final_doc)
+                                        
+                                        if response:
+                                            st.success(f"""
+                                            ✅ New document created successfully!
+                                            - ID: {final_doc['id']}
+                                            - Name: {final_doc['name']}
+                                            """)
+                                            # Update session state to show the new document
+                                            st.session_state.selected_document_id = final_doc['id']
+                                            st.rerun()
+                                        else:
+                                            st.error("Failed to create new document")
+                                    except json.JSONDecodeError as e:
+                                        st.error(f"Invalid JSON format: {str(e)}")
+                                    except Exception as e:
+                                        st.error(f"Error creating document: {str(e)}")
+                            
+                            # Navigation buttons for viewing other documents to clone
+                            col_prev, col_next = st.columns([1, 1])
+                            with col_prev:
+                                if st.button("⬅️ Previous", key='prev_clone'):
+                                    if st.session_state.current_index > 0:
+                                        st.session_state.current_index -= 1
+                                        st.rerun()
+                            with col_next:
+                                if st.button("➡️ Next", key='next_clone'):
+                                    if st.session_state.current_index < total_docs - 1:
+                                        st.session_state.current_index += 1
+                                        st.rerun()
+                                            
                                             
                         elif selected_view == 'New Record':
                             st.markdown("#### Create a new document:")