Update app.py

app.py

@@ -5,17 +5,20 @@ import pandas as pd
 import boto3
 import AWSHandler
 import pinecone
+from sentence_transformers import SentenceTransformer
 
 aws_access_key = os.getenv("aws_access_key")
 aws_secret_key = os.getenv("aws_secret_key")
+pinecone_api_key = os.getenv("pinecone_api_key")
+pinecone_environment = os.getenv("pinecone_environment")
+
 s3 = boto3.client('s3', aws_access_key_id=aws_access_key, aws_secret_access_key=aws_secret_key)
+pinecone.init(api_key=pinecone_api_key, environment=pinecone_environment)
 
-st.set_page_config(layout="wide")
+model = SentenceTransformer('all-mpnet-base-v2')
 
-justices_response =  s3.get_object(Bucket='gideon-corpus', Key='Justices Table.csv')
-justices_df = pd.read_csv(justices_response['Body'])
+st.set_page_config(layout="wide")
 
-# Print the logs
 logs_response = s3.get_object(Bucket='gideon-corpus', Key='logs/logs.csv')
 logs_df = pd.read_csv(logs_response['Body'])
 logs_df['Last Upload'] = pd.to_datetime(logs_df['Last Upload'], unit='s')
@@ -24,6 +27,9 @@ processed  = len(logs_df) - len(logs_df[logs_df['Last Upload'].isna()])
 st.metric(label="Num Processed Cases", value=processed)
 st.divider()
 
+index = pinecone.Index("scotus")
+
+
 search = st.text_input('Search', placeholders='First Amendment cannot protect libel')