Update app.py

app.py

@@ -165,6 +165,10 @@ def main():
         st.session_state.selected_container = None
     if 'selected_document_id' not in st.session_state:
         st.session_state.selected_document_id = None
+    if 'current_index' not in st.session_state:
+        st.session_state.current_index = 0
+    if 'cloned_doc' not in st.session_state:
+        st.session_state.cloned_doc = None
 
     # 🔐 Automatic Login
     if Key:
@@ -190,6 +194,7 @@ def main():
                 st.session_state.selected_database = selected_db
                 st.session_state.selected_container = None
                 st.session_state.selected_document_id = None
+                st.session_state.current_index = 0
                 st.rerun()
             
             if st.session_state.selected_database:
@@ -200,6 +205,7 @@ def main():
                 if selected_container != st.session_state.selected_container:
                     st.session_state.selected_container = selected_container
                     st.session_state.selected_document_id = None
+                    st.session_state.current_index = 0
                     st.rerun()
                 
                 if st.session_state.selected_container:
@@ -213,10 +219,11 @@ def main():
                         else:
                             st.error(download_link)
                     
-                    # Fetch up to 6 documents to check if more than 5 exist
-                    documents = get_documents(container, limit=6)
+                    # Fetch documents
+                    documents = get_documents(container)
+                    total_docs = len(documents)
                     
-                    if len(documents) > 5:
+                    if total_docs > 5:
                         documents_to_display = documents[:5]
                         st.info("Showing top 5 most recent documents.")
                     else:
@@ -228,8 +235,45 @@ def main():
                         view_options = ['Show as Markdown', 'Show as Code Editor', 'Show as Edit and Save', 'Clone Document', 'New Record']
                         selected_view = st.selectbox("Select Viewer/Editor", view_options, index=2)
                         
-                        if selected_view == 'Show as Edit and Save':
-                            # ✏️ Show as Edit and Save
+                        if selected_view == 'Show as Markdown':
+                            # 🖌️ Show each record as Markdown
+                            for idx, doc in enumerate(documents_to_display):
+                                st.markdown(f"### **🆔 ID: {doc.get('id', 'Unknown')}**")
+                                content = doc.get('content', '')
+                                if isinstance(content, dict) or isinstance(content, list):
+                                    content = json.dumps(content, indent=2)
+                                st.markdown(content)
+                        elif selected_view == 'Show as Code Editor':
+                            # 💻 Show each record in a code editor with navigation
+                            total_docs = len(documents)
+                            doc = documents[st.session_state.current_index]
+                            st.markdown(f"#### Document ID: {doc.get('id', '')}")
+                            doc_str = st.text_area("Edit Document", value=json.dumps(doc, indent=2), height=300, key=f'code_editor_{st.session_state.current_index}')
+                            col_prev, col_next = st.columns([1, 1])
+                            with col_prev:
+                                if st.button("⬅️ Previous"):
+                                    if st.session_state.current_index > 0:
+                                        st.session_state.current_index -= 1
+                                        st.rerun()
+                            with col_next:
+                                if st.button("➡️ Next"):
+                                    if st.session_state.current_index < total_docs - 1:
+                                        st.session_state.current_index += 1
+                                        st.rerun()
+                            if st.button("💾 Save Changes", key=f'save_button_{st.session_state.current_index}'):
+                                try:
+                                    updated_doc = json.loads(doc_str)
+                                    success, message = update_record(container, updated_doc)
+                                    if success:
+                                        st.success(f"Document {updated_doc['id']} saved successfully.")
+                                        st.session_state.selected_document_id = updated_doc['id']
+                                        st.rerun()
+                                    else:
+                                        st.error(message)
+                                except json.JSONDecodeError as e:
+                                    st.error(f"Invalid JSON: {str(e)} 🚫")
+                        elif selected_view == 'Show as Edit and Save':
+                            # ✏️ Show as Edit and Save in columns
                             st.markdown("#### Edit the document fields below:")
                             
                             # Create columns for each document
@@ -257,61 +301,56 @@ def main():
                                                 st.error(message)
                                         except json.JSONDecodeError as e:
                                             st.error(f"Invalid JSON: {str(e)} 🚫")
-                        else:
-                            # Allow user to select a document from the sidebar
-                            document_ids = [doc.get('id', 'Unknown') for doc in documents_to_display]
-                            selected_document = st.sidebar.selectbox("📄 Select Document", document_ids, index=document_ids.index(st.session_state.selected_document_id) if st.session_state.selected_document_id in document_ids else 0)
-                            
-                            if selected_document != st.session_state.selected_document_id:
-                                st.session_state.selected_document_id = selected_document
-                                st.rerun()
-                            
-                            if st.session_state.selected_document_id:
-                                st.subheader(f"📄 Document Details: {st.session_state.selected_document_id}")
-                                selected_doc = next((doc for doc in documents_to_display if doc.get('id') == st.session_state.selected_document_id), None)
-                                if selected_doc:
-                                    if selected_view == 'Show as Markdown':
-                                        # 🖌️ Show as Markdown
-                                        content = selected_doc.get('content', '')
-                                        if isinstance(content, dict) or isinstance(content, list):
-                                            content = json.dumps(content, indent=2)
-                                        st.markdown(content)
-                                    elif selected_view == 'Show as Code Editor':
-                                        # 💻 Show as Code Editor
-                                        st.code(json.dumps(selected_doc, indent=2), language='python')
-                                    elif selected_view == 'Clone Document':
-                                        # 🧬 Clone Document
-                                        if st.button("📄 Clone Document"):
-                                            cloned_doc = selected_doc.copy()
-                                            # Generate a unique ID
-                                            cloned_doc['id'] = generate_unique_id()
-                                            success, message = insert_record(container, cloned_doc)
-                                            if success:
-                                                st.success(f"Document cloned with new id: {cloned_doc['id']} 🎉")
-                                                st.session_state.selected_document_id = cloned_doc['id']
-                                                # Switch to 'Show as Edit and Save' mode
-                                                st.rerun()
-                                            else:
-                                                st.error(message)
-                                    elif selected_view == 'New Record':
-                                        # 🆕 New Record
-                                        st.markdown("#### Create a new document:")
-                                        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)} 🚫")
+                        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}'):
+                                    cloned_doc = doc.copy()
+                                    # Generate a unique ID
+                                    cloned_doc['id'] = 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:")
+                                cloned_doc_str = st.text_area("Cloned Document Content (in JSON format)", 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':
+                            # 🆕 New Record
+                            st.markdown("#### Create a new document:")
+                            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)} 🚫")
                     else:
                         st.sidebar.info("No documents found in this container. 📭")
             
@@ -382,6 +421,7 @@ def main():
         st.session_state.selected_database = None
         st.session_state.selected_container = None
         st.session_state.selected_document_id = None
+        st.session_state.current_index = 0
         st.rerun()
 
 if __name__ == "__main__":