Update app.py

app.py

@@ -462,65 +462,8 @@ def main():
     if 'documents' not in st.session_state:
         st.session_state.documents = []
 
-    # Cosmos DB Connection (assuming this is done earlier in your code)
-    if st.session_state.client is None:
-        st.session_state.client = CosmosClient(ENDPOINT, credential=st.session_state.primary_key)
 
-    # Database and Container Selection
-    databases = get_databases(st.session_state.client)
-    selected_db = st.sidebar.selectbox("🗃️ Select Database", databases)
-    
-    if selected_db != st.session_state.selected_database:
-        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:
-        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("---")
+
 
     # Automatic Login
     if Key:
@@ -562,6 +505,69 @@ def main():
                 
                 if st.session_state.selected_container:
                     container = database.get_container_client(st.session_state.selected_container)
+
+                if selected_db != st.session_state.selected_database:
+                    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()
+
+
+
+
+
+                    # 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("---")
+            
+                                
+
+
+
+
                     
                     # Add Export button
                     if st.button("📦 Export Container Data"):