Update app.py

app.py

@@ -5,6 +5,7 @@ import pandas as pd
 import boto3
 import json
 import datetime
+import AWSHandler
 
 aws_access_key = os.getenv("aws_access_key")
 aws_secret_key = os.getenv("aws_secret_key")
@@ -32,10 +33,7 @@ for result in paginator.paginate(Bucket="gideon-corpus", Prefix='Cases/', Delimi
 subdirectories = list(subdirectories)
 subs = [s.split('/')[1] for s in subdirectories]
 
-case_docs = []
-for s in subs:
-    opinions_df, metadata = retrieve_casedocs(s)
-    case_docs.append((opinions_df, metadata))
+case_docs = AWSHandler.retrieve_all_casedocs(prefix='Cases/', aws_access_key_id=aws_access_key, aws_secret_access_key=aws_secret_key)
 
 for (opins_df, metadata) in case_docs:
     st.write(metadata)