Update app.py

app.py

@@ -515,54 +515,7 @@ def main():
 
 
 
-
-
-                    # save_or_clone_to_cosmos_db special project
-                if st.session_state.selected_database:
-                    database = st.session_state.client.get_database_client(st.session_state.selected_database)
-                    containers = get_containers(database)
-                    selected_container = st.sidebar.selectbox("📁 Select Container", containers)
                     
-                    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:
-                        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)
-                        
-                        # Fetch documents
-                        st.session_state.documents = get_documents(container)
-                        total_docs = len(st.session_state.documents)
-            
-                        # Add buttons for saving new document and cloning
-                        if st.button("💾 Save New Document"):
-                            user_query = st.session_state.get('user_query', '')
-                            ai_response = st.session_state.get('ai_response', '')
-                            save_or_clone_to_cosmos_db(container, query=user_query, response=ai_response)
-            
-                        if st.button("🔄 Clone Document"):
-                            doc_to_clone = st.selectbox("Select document to clone", [doc['id'] for doc in st.session_state.documents])
-                            if doc_to_clone:
-                                save_or_clone_to_cosmos_db(container, clone_id=doc_to_clone)
-            
-                        # Display documents
-                        if st.session_state.documents:
-                            for doc in st.session_state.documents:
-                                st.write(f"ID: {doc['id']}")
-                                st.write(f"Query: {doc.get('query', 'N/A')}")
-                                st.write(f"Response: {doc.get('response', 'N/A')}")
-                                st.write("---")
-            
                                 
 
 
@@ -580,6 +533,28 @@ def main():
                     # Fetch documents
                     documents = get_documents(container)
                     total_docs = len(documents)
+        
+                    # Add buttons for saving new document and cloning
+                    if st.button("💾 Save New Document"):
+                        user_query = st.session_state.get('user_query', '')
+                        ai_response = st.session_state.get('ai_response', '')
+                        save_or_clone_to_cosmos_db(container, query=user_query, response=ai_response)
+        
+                    if st.button("🔄 Clone Document"):
+                        doc_to_clone = st.selectbox("Select document to clone", [doc['id'] for doc in st.session_state.documents])
+                        if doc_to_clone:
+                            save_or_clone_to_cosmos_db(container, clone_id=doc_to_clone)
+        
+                    # Display documents
+                    if st.session_state.documents:
+                        for doc in st.session_state.documents:
+                            st.write(f"ID: {doc['id']}")
+                            st.write(f"Query: {doc.get('query', 'N/A')}")
+                            st.write(f"Response: {doc.get('response', 'N/A')}")
+                            st.write("---")
+        
+
+
                     
                     if total_docs > 5:
                         documents_to_display = documents[:5]