Update app.py

app.py

@@ -6,16 +6,16 @@ from collections import defaultdict
 from datetime import datetime
 
 # AWS S3 setup
-os.environ['AWS_ACCESS_KEY']
-os.environ['AWS_SECRET_KEY']
-os.environ['BUCKET_NAME']
+aws_access_key_id=os.environ['AWS_ACCESS_KEY']
+aws_secret_access_key=os.environ['AWS_SECRET_KEY']
+aws_bucket=os.environ['BUCKET_NAME']
 
 # Initialize a session using DigitalOcean Spaces
 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_secret_access_key)
 
 def list_folders(bucket_name):
     try:
@@ -59,12 +59,12 @@ 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)
 selected_folder = st.selectbox('Select a folder', folders, disabled=True, index=folders.index('JUNIPER/') if 'JUNIPER/' in folders else 0)
 # Display file counts
 if selected_folder:
-    folder_counts = count_files_in_folder(BUCKET_NAME, selected_folder)
+    folder_counts = count_files_in_folder(aws_bucket, selected_folder)
     
     if folder_counts:
         # Collect all dates