Upload 2 files

app.py

@@ -0,0 +1,145 @@
+import streamlit as st
+import boto3
+import pandas as pd
+
+# Define AWS credentials and S3 resources
+ACCESS_KEY = "AKIAZPNYOGSLLAIPOMM2"
+SECRET_KEY = "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"
+REGION_NAME = "eu-west-2"
+BUCKET_NAME = "ogtl-analytics"
+
+# Define unique codes for each folder
+FOLDER_CODES = {
+    "flutterwave": "xwy3455",
+    "canvassing": "0thdkwn",
+    "test": "0thdkwn",
+    # Add more folders and codes here...
+}
+
+# Define subfolders for each folder
+SUBFOLDERS = {
+    "flutterwave": ["teammate-performance", "product"],
+    "canvassing": ["agent-performance-vicidial"],
+}
+
+# Connect to S3 client
+s3_client = boto3.client(
+    "s3",
+    aws_access_key_id=ACCESS_KEY,
+    aws_secret_access_key=SECRET_KEY,
+    region_name=REGION_NAME,
+)
+
+def preview_data(input_data, folder, subfolder):
+    if folder=='canvassing' and subfolder=='agent-performance-vicidial':
+        return pd.read_csv(input_data,header=3, nrows=10)
+
+    elif folder=='flutterwave' and subfolder=='teammate-performance':
+        return pd.read_csv(input_data,header=3, nrows=10)
+
+    elif folder=='flutterwave' and subfolder=='product':
+        return pd.read_csv(input_data,low_memory=False, nrows=10)
+
+
+# Get a list of folders in the bucket
+folders = [obj["Key"].split("/")[0] for obj in s3_client.list_objects_v2(Bucket=BUCKET_NAME)["Contents"]]
+
+# Streamlit app
+st.title("Upload CSV File to S3 Bucket")
+
+# Upload file
+uploaded_file = st.file_uploader("Choose a CSV file", type=".csv")
+
+# Check if file is None or not a CSV file
+if uploaded_file and uploaded_file.type != "text/csv":
+    st.error("Only CSV files are allowed!")
+    uploaded_file = None
+
+# Show the rest of the app only if a valid CSV file is uploaded
+if uploaded_file:
+
+    # Choose folder
+    selected_folder = st.selectbox("Select Campaign", set(folders))
+
+    # Check if folder has subfolders
+    if selected_folder in SUBFOLDERS:
+        # Show subfolder selection
+        selected_subfolder = st.selectbox("Select folder", SUBFOLDERS[selected_folder])
+    else:
+        selected_subfolder = None
+
+    # Preview data
+    # Read a few rows of data
+    data = preview_data(uploaded_file, selected_folder, selected_subfolder)
+
+    # Show data preview
+    st.subheader("Data preview: Please Confirm the Data Before Proceeding")
+    st.write(data)
+
+    # Input unique code
+    unique_code = st.text_input("Enter your unique code")
+
+    # Upload button
+    upload_button = st.button("Upload")
+
+    if upload_button:
+
+        # Validate unique code
+        if FOLDER_CODES.get(selected_folder) != unique_code:
+            st.error("Invalid unique code for selected folder")
+        else:
+            # Read entire uploaded file
+            data = pd.read_csv(uploaded_file)
+
+            # Create filename based on folder and subfolder
+            if selected_subfolder:
+                filename = f"{selected_folder}/{selected_subfolder}/{uploaded_file.name}"
+            else:
+                filename = f"{selected_folder}/{uploaded_file.name}"
+
+            try:
+                # Upload notification
+                st.info("Please wait, upload is in progress...")
+
+                # Upload file to S3
+                s3_client.put_object(Body=uploaded_file, Bucket=BUCKET_NAME, Key=filename)
+                st.empty()
+                st.success("File uploaded successfully!")
+
+            except:
+                st.write("An Error Occured")
+
+
+
+            # Clear file input and unique code
+            uploaded_file = None
+            unique_code = ""
+
+# # Choose folder
+# selected_folder = st.selectbox("Select folder", folders)
+#
+# # Input unique code
+# unique_code = st.text_input("Enter unique code")
+#
+# # Upload button
+# upload_button = st.button("Upload")
+#
+# if upload_button:
+#     # Validate unique code
+#     if FOLDER_CODES.get(selected_folder) != unique_code:
+#         st.error("Invalid unique code for selected folder")
+#     else:
+#         # Read uploaded file
+#         data = pd.read_csv(uploaded_file)
+#
+#         # Create filename
+#         filename = f"{selected_folder}/{uploaded_file.name}"
+#
+#         # Upload file to S3
+#         s3_client.put_object(Body=data.to_csv(), Bucket=BUCKET_NAME, Key=filename)
+#
+#         st.success("File uploaded successfully!")
+#
+#         # Clear file input and unique code
+#         uploaded_file = None
+#         unique_code = ""