Update app.py

app.py

@@ -306,4 +306,176 @@ def main():
                             with col_prev:
                                 if st.button("⬅️ Previous", key='prev_code'):
                                     if st.session_state.current_index > 0:
-  
+                                        st.session_state.current_index -= 1
+                                        st.rerun()
+                            with col_next:
+                                if st.button("➡️ Next", key='next_code'):
+                                    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
+                            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}')
+                                    # Remove 'id' from the document for editing other fields
+                                    editable_doc = doc.copy()
+                                    editable_doc.pop('id', 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}')
+                                    if st.button("💾 Save Changes", key=f'save_button_{idx}'):
+                                        try:
+                                            updated_doc = json.loads(doc_str)
+                                            updated_doc['id'] = editable_id  # Include the possibly edited ID
+                                            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 == '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. 📭")
+            
+            # 🎉 Main content area
+            st.subheader(f"📊 Container: {st.session_state.selected_container}")
+            if st.session_state.selected_container:
+                if documents_to_display:
+                    df = pd.DataFrame(documents_to_display)
+                    st.dataframe(df)
+                else:
+                    st.info("No documents to display. 🧐")
+                    
+            # 🐙 GitHub section
+            st.subheader("🐙 GitHub Operations")
+            github_token = os.environ.get("GITHUB")  # Read GitHub token from environment variable
+            source_repo = st.text_input("Source GitHub Repository URL", value="https://github.com/AaronCWacker/AIExamples-8-24-Streamlit")
+            new_repo_name = st.text_input("New Repository Name (for cloning)", value=f"AIExample-Clone-{datetime.now().strftime('%Y%m%d_%H%M%S')}")
+
+            col1, col2 = st.columns(2)
+            with col1:
+                if st.button("📥 Clone Repository"):
+                    if github_token and source_repo:
+                        try:
+                            local_path = f"./temp_repo_{datetime.now().strftime('%Y%m%d%H%M%S')}"
+                            download_github_repo(source_repo, local_path)
+                            zip_filename = f"{new_repo_name}.zip"
+                            create_zip_file(local_path, zip_filename[:-4])
+                            st.markdown(get_base64_download_link(zip_filename, zip_filename), unsafe_allow_html=True)
+                            st.success("Repository cloned successfully! 🎉")
+                        except Exception as e:
+                            st.error(f"An error occurred: {str(e)} 😢")
+                        finally:
+                            if os.path.exists(local_path):
+                                shutil.rmtree(local_path)
+                            if os.path.exists(zip_filename):
+                                os.remove(zip_filename)
+                    else:
+                        st.error("Please ensure GitHub token is set in environment variables and source repository URL is provided. 🔑❓")
+        
+            with col2:
+                if st.button("📤 Push to New Repository"):
+                    if github_token and source_repo:
+                        try:
+                            g = Github(github_token)
+                            new_repo = create_repo(g, new_repo_name)
+                            local_path = f"./temp_repo_{datetime.now().strftime('%Y%m%d%H%M%S')}"
+                            download_github_repo(source_repo, local_path)
+                            push_to_github(local_path, new_repo, github_token)
+                            st.success(f"Repository pushed successfully to {new_repo.html_url} 🚀")
+                        except Exception as e:
+                            st.error(f"An error occurred: {str(e)} 😢")
+                        finally:
+                            if os.path.exists(local_path):
+                                shutil.rmtree(local_path)
+                    else:
+                        st.error("Please ensure GitHub token is set in environment variables and source repository URL is provided. 🔑❓")
+    
+        except exceptions.CosmosHttpResponseError as e:
+            st.error(f"Failed to connect to Cosmos DB. HTTP error: {str(e)} 🚨")
+        except Exception as e:
+            st.error(f"An unexpected error occurred: {str(e)} 😱")
+
+    # 🚪 Logout button
+    if st.session_state.logged_in and st.sidebar.button("🚪 Logout"):
+        st.session_state.logged_in = False
+        st.session_state.selected_records.clear()
+        st.session_state.client = None
+        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__":
+    main()