Update app.py

app.py

@@ -63,6 +63,20 @@ def get_base64_download_link(file_path, file_name):
     base64_encoded = base64.b64encode(contents).decode()
     return f'<a href="data:application/zip;base64,{base64_encoded}" download="{file_name}">Download {file_name}</a>'
 
+
+# New functions for dynamic sidebar
+def get_databases(client):
+    return [db['id'] for db in client.list_databases()]
+
+def get_containers(database):
+    return [container['id'] for container in database.list_containers()]
+
+def get_documents(container, limit=1000):
+    query = "SELECT * FROM c"
+    items = list(container.query_items(query=query, enable_cross_partition_query=True, max_item_count=limit))
+    return items
+
+
 # Cosmos DB functions
 def insert_record(record):
     try:
@@ -155,203 +169,101 @@ def archive_all_data(client):
 # Streamlit app
 st.title("🌟 Cosmos DB and GitHub Integration")
 
-# Initialize session state
-if 'logged_in' not in st.session_state:
-    st.session_state.logged_in = False
-if 'selected_records' not in st.session_state:
-    st.session_state.selected_records = []
-
-# Login section
-if not st.session_state.logged_in:
-    st.subheader("🔐 Login")
-    input_key = Key
-    if st.button("🚀 Login"):
-        if input_key:
-            st.session_state.primary_key = input_key
-            st.session_state.logged_in = True
-            st.rerun()
-        else:
-            st.error("Invalid key. Please check your environment variables.")
-else:
-    # Initialize Cosmos DB client
-    try:
-        client = CosmosClient(ENDPOINT, credential=st.session_state.primary_key)
-        database = client.get_database_client(DATABASE_NAME)
-        container = database.get_container_client(CONTAINER_NAME)
-    except exceptions.CosmosHttpResponseError as e:
-        st.error(f"Failed to connect to Cosmos DB. HTTP error: {str(e)}. Status code: {e.status_code}")
-        st.stop()
-    except Exception as e:
-        st.error(f"An unexpected error occurred while connecting to Cosmos DB: {str(e)}")
-        st.stop()
-
-    # 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.")
 
-    # Cosmos DB Operations
-    st.subheader("☁️ Cosmos DB Operations")
-    
-    # Archive all data
-    if st.button("📦 Archive All Cosmos DB Data"):
-        download_link = archive_all_data(client)
-        st.markdown(download_link, unsafe_allow_html=True)
 
-    # Fetch and display all records
-    st.subheader("📊 All Records")
-    df = fetch_all_records()
+# Modify the main app
+def main():
+    st.title("🌟 Cosmos DB and GitHub Integration")
 
-    if df.empty:
-        st.write("No records found in the database.")
+    # Initialize session state
+    if 'logged_in' not in st.session_state:
+        st.session_state.logged_in = False
+    if 'selected_records' not in st.session_state:
+        st.session_state.selected_records = []
+    if 'client' not in st.session_state:
+        st.session_state.client = None
+    if 'selected_database' not in st.session_state:
+        st.session_state.selected_database = None
+    if 'selected_container' not in st.session_state:
+        st.session_state.selected_container = None
+
+    # Login section
+    if not st.session_state.logged_in:
+        st.subheader("🔐 Login")
+        input_key = st.text_input("Enter your Cosmos DB key", type="password")
+        if st.button("🚀 Login"):
+            if input_key:
+                st.session_state.primary_key = input_key
+                st.session_state.logged_in = True
+                st.rerun()
+            else:
+                st.error("Invalid key. Please check your input.")
     else:
-        st.write("Records:")
-        for index, row in df.iterrows():
-            col1, col2, col3 = st.columns([5, 1, 1])
-
-            with col1:
-                st.write(f"ID: {row['id']}, Name: {row['name']}, Document: {row['document']}, "
-                         f"Evaluation Text: {row['evaluationText']}, Evaluation Score: {row['evaluationScore']}")
-
-            with col2:
-                key = f"select_{row['id']}"
-                if st.button(f"👉 Select", key=key):
-                    st.session_state.selected_record = row.to_dict()
-
-            with col3:
-                if st.button(f"🗑️ Delete", key=f"delete_{row['id']}"):
-                    success, message = delete_record(row['name'], row['id'])
-                    if success:
-                        st.success(message)
-                        st.rerun()
+        # Initialize Cosmos DB client
+        try:
+            if st.session_state.client is None:
+                st.session_state.client = CosmosClient(ENDPOINT, credential=st.session_state.primary_key)
+            
+            # Sidebar for database, container, and document selection
+            st.sidebar.title("🗄️ Cosmos DB Navigator")
+            
+            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.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.rerun()
+                
+                if st.session_state.selected_container:
+                    container = database.get_container_client(st.session_state.selected_container)
+                    
+                    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)
+                    
+                    if documents:
+                        document_ids = [doc.get('id', 'Unknown') for doc in documents]
+                        selected_document = st.sidebar.selectbox("📄 Select Document", document_ids)
+                        
+                        if selected_document:
+                            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)
                     else:
-                        st.error(message)
-
-    # Display selected record for editing
-    if 'selected_record' in st.session_state and st.session_state.selected_record:
-        selected_record = st.session_state.selected_record
-
-        st.subheader(f"Editing Record - ID: {selected_record['id']}")
-
-        updated_name = st.text_input("Name", value=selected_record['name'])
-        updated_document = st.text_area("Document", value=selected_record['document'])
-        updated_evaluation_text = st.text_area("Evaluation Text", value=selected_record['evaluationText'])
-        updated_evaluation_score = st.text_input("Evaluation Score", value=str(selected_record['evaluationScore']))
-
-        if st.button("💾 Save Changes"):
-            updated_record = {
-                "id": selected_record['id'],
-                "name": updated_name,
-                "document": updated_document,
-                "evaluationText": updated_evaluation_text,
-                "evaluationScore": updated_evaluation_score
-            }
-
-            success, message = update_record(updated_record)
-            if success:
-                st.success(message)
-                st.session_state.selected_record = updated_record
-            else:
-                st.error(message)
-
-    # Input fields for new record
-    st.subheader("📝 Enter New Record Details")
-    new_id = st.text_input("ID")
-    new_name = st.text_input("Name")
-    new_document = st.text_area("Document")
-    new_evaluation_text = st.text_area("Evaluation Text")
-    new_evaluation_score = st.text_input("Evaluation Score")
-
-    col1, col2 = st.columns(2)
-
-    with col1:
-        if st.button("💾 Insert Record"):
-            record = {
-                "id": new_id,
-                "name": new_name,
-                "document": new_document,
-                "evaluationText": new_evaluation_text,
-                "evaluationScore": new_evaluation_score
-            }
-
-            success, response = insert_record(record)
-            if success:
-                st.success("✅ Record inserted successfully!")
-                st.json(response)
-            else:
-                st.error(f"❌ Failed to insert record: {response}")
-            st.rerun()
-
-    with col2:
-        if st.button("🔧 Call Procedure"):
-            record = {
-                "id": new_id,
-                "name": new_name,
-                "document": new_document,
-                "evaluationText": new_evaluation_text,
-                "evaluationScore": new_evaluation_score
-            }
-
-            success, response = call_stored_procedure(record)
-            if success:
-                st.success("✅ Stored procedure executed successfully!")
-                st.json(response)
-            else:
-                st.error(f"❌ Failed to execute stored procedure: {response}")
+                        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:
+                df = pd.DataFrame(documents)
+                st.dataframe(df)
+            
+            # ... (keep the rest of your existing code for GitHub operations, etc.)
+
+        except exceptions.CosmosHttpResponseError as e:
+            st.error(f"Failed to connect to Cosmos DB. HTTP error: {str(e)}. Status code: {e.status_code}")
+        except Exception as e:
+            st.error(f"An unexpected error occurred: {str(e)}")
 
     # Logout button
-    if st.button("🚪 Logout"):
+    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.selected_record = None
+        st.session_state.client = None
+        st.session_state.selected_database = None
+        st.session_state.selected_container = None
         st.rerun()
 
-    # Display connection info
-    st.sidebar.subheader("🔗 Connection Information")
-    st.sidebar.text(f"Endpoint: {ENDPOINT}")
-    st.sidebar.text(f"Subscription ID: {SUBSCRIPTION_ID}")
-    st.sidebar.text(f"Database: {DATABASE_NAME}")
-    st.sidebar.text(f"Container: {CONTAINER_NAME}")
-
-#if __name__ == "__main__":
-#    st.write("Application is running")
+if __name__ == "__main__":
+    main()
+