Update app.py

app.py

@@ -134,32 +134,27 @@ def delete_record(name, id):
     except Exception as e:
         return False, f"An unexpected error occurred: {traceback.format_exc()}"
 
-# New function to archive all databases and containers
-def archive_all_data(client):
+# New function to archive current container
+def archive_current_container(database_name, container_name, client):
     try:
-        base_dir = "./cosmos_archive"
+        base_dir = "./cosmos_archive_current_container"
         if os.path.exists(base_dir):
             shutil.rmtree(base_dir)
         os.makedirs(base_dir)
-
-        for database in client.list_databases():
-            db_name = database['id']
-            db_dir = os.path.join(base_dir, db_name)
-            os.makedirs(db_dir)
-
-            db_client = client.get_database_client(db_name)
-            for container in db_client.list_containers():
-                container_name = container['id']
-                container_dir = os.path.join(db_dir, container_name)
-                os.makedirs(container_dir)
-
-                container_client = db_client.get_container_client(container_name)
-                items = list(container_client.read_all_items())
-
-                with open(os.path.join(container_dir, f"{container_name}.json"), 'w') as f:
-                    json.dump(items, f, indent=2)
-
-        archive_name = f"cosmos_archive_{datetime.now().strftime('%Y%m%d_%H%M%S')}"
+        
+        db_client = client.get_database_client(database_name)
+        container_client = db_client.get_container_client(container_name)
+        items = list(container_client.read_all_items())
+        
+        container_dir = os.path.join(base_dir, container_name)
+        os.makedirs(container_dir)
+        
+        for item in items:
+            item_id = item.get('id', f"unknown_{datetime.now().strftime('%Y%m%d%H%M%S')}")
+            with open(os.path.join(container_dir, f"{item_id}.json"), 'w') as f:
+                json.dump(item, f, indent=2)
+        
+        archive_name = f"{container_name}_archive_{datetime.now().strftime('%Y%m%d%H%M%S')}"
         shutil.make_archive(archive_name, 'zip', base_dir)
         
         return get_base64_download_link(f"{archive_name}.zip", f"{archive_name}.zip")
@@ -226,6 +221,14 @@ def main():
                 if st.session_state.selected_container:
                     container = database.get_container_client(st.session_state.selected_container)
                     
+                    # Add Export button
+                    if st.button("📦 Export Container Data"):
+                        download_link = archive_current_container(st.session_state.selected_database, st.session_state.selected_container, st.session_state.client)
+                        if download_link.startswith('<a'):
+                            st.markdown(download_link, unsafe_allow_html=True)
+                        else:
+                            st.error(download_link)
+                    
                     limit_to_1000 = st.sidebar.checkbox("🔢 Limit to top 1000 documents", value=True)
                     documents = get_documents(container, limit=1000 if limit_to_1000 else None)
                     
@@ -237,7 +240,63 @@ def main():
                             st.subheader(f"📄 Document Details: {selected_document}")
                             selected_doc = next((doc for doc in documents if doc.get('id') == selected_document), None)
                             if selected_doc:
-                                st.json(selected_doc)
+                                # Add Viewer/Editor selection
+                                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)
+
+                                if selected_view == 'Show as Markdown':
+                                    # Show as Markdown
+                                    items = get_documents(container)
+                                    for item in items:
+                                        st.markdown(f"### ID: {item.get('id', 'Unknown')}")
+                                        content = item.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
+                                    items = get_documents(container)
+                                    for item in items:
+                                        st.code(json.dumps(item, indent=2), language='python')
+                                elif selected_view == 'Show as Edit and Save':
+                                    # Show as Edit and Save
+                                    doc_str = st.text_area("Edit Document", value=json.dumps(selected_doc, indent=2), height=300)
+                                    if st.button("💾 Save"):
+                                        try:
+                                            updated_doc = json.loads(doc_str)
+                                            success, message = update_record(updated_doc)
+                                            if success:
+                                                st.success(message)
+                                            else:
+                                                st.error(message)
+                                        except json.JSONDecodeError as e:
+                                            st.error(f"Invalid JSON: {str(e)}")
+                                elif selected_view == 'Clone Document':
+                                    # Clone Document
+                                    if st.button("📄 Clone Document"):
+                                        cloned_doc = selected_doc.copy()
+                                        timestamp = datetime.now().strftime('%Y%m%d%H%M%S')
+                                        cloned_doc['id'] = f"{cloned_doc['id']}_clone_{timestamp}"
+                                        success, message = insert_record(cloned_doc)
+                                        if success:
+                                            st.success(f"Document cloned with new id: {cloned_doc['id']}")
+                                        else:
+                                            st.error(message)
+                                elif selected_view == 'New Record':
+                                    # New Record
+                                    new_doc_str = st.text_area("New Document", value='{}', height=300)
+                                    if st.button("➕ Create New Document"):
+                                        try:
+                                            new_doc = json.loads(new_doc_str)
+                                            if 'id' not in new_doc:
+                                                new_doc['id'] = f"new_doc_{datetime.now().strftime('%Y%m%d%H%M%S')}"
+                                            success, message = insert_record(new_doc)
+                                            if success:
+                                                st.success(f"New document created with id: {new_doc['id']}")
+                                            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.")
             
@@ -252,7 +311,7 @@ def main():
             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"):
@@ -308,4 +367,3 @@ def main():
 
 if __name__ == "__main__":
     main()
-