Update app.py

app.py

@@ -9,7 +9,6 @@ import streamlit as st
 import boto3
 from botocore.exceptions import ClientError, NoCredentialsError
 
-read_file_path = "160/task_for_you.pdf"
 def check_pdf(read_file_path):
     try:
         parser = LlamaParse(result_type="markdown", api_key=os.environ['LLAMA_CLOUD_API_KEY'], ignore_errors=False)
@@ -83,12 +82,11 @@ def download_files_from_s3(bucket_name, local_folder, file_path_list):
         st.write(f"An error occurred: {e}")
 
 bucket_name = st.text_input("Enter bucket name", key="bucket_name")
-user_name = st.text_input("Enter user name", key="user_name")
-file_name = st.text_input("Enter file name", key="file_name")
+key = st.text_input("Enter key", key="key")
 
 if st.button("Submit"):
     with tempfile.TemporaryDirectory() as temp_dir:
-        download_files_from_s3(bucket_name, temp_dir, [os.path.join(user_name, file_name)])
+        download_files_from_s3(bucket_name, temp_dir, [key])
         if check_pdf(os.path.join(temp_dir, file_name)):
             st.success("File downloaded successfully")
         else: