Update app.py

app.py

@@ -1,3 +1,4 @@
+
 import os
 import streamlit as st
 import boto3
@@ -14,8 +15,8 @@ aws_bucket=os.environ['BUCKET_NAME']
 session = boto3.session.Session()
 s3 = session.client('s3',
                     region_name='us-west-2',  # Update to the correct region
-                    aws_access_key_id=AWS_ACCESS_KEY,
-                    aws_secret_access_key=AWS_SECRET_KEY)
+                    aws_access_key_id=aws_access_key,
+                    aws_secret_access_key=aws_secret_key)
 
 def list_folders(bucket_name):
     try:
@@ -59,7 +60,7 @@ def count_files_in_folder(bucket_name, prefix):
 st.title('NeuroSinQ IAT Completions Dashboard')
 
 # Dropdown to select folder
-folders = list_folders(BUCKET_NAME)
+folders = list_folders(aws_bucket)
 selected_folder = st.selectbox('Select a folder', folders, index=folders.index('/') if '/' in folders else 0)
 
 # Display file counts