Update app.py

app.py

@@ -17,10 +17,24 @@ if uploaded_file:
     df['YEAR'] = df['INVOICE_DATE'].dt.year
     df['MONTH'] = df['INVOICE_DATE'].dt.month
 
-    # User Inputs
+    # Clean whitespace in 'CUSTOMER_NAME' and 'DESCRIPTION'
+    df['CUSTOMER_NAME'] = df['CUSTOMER_NAME'].str.strip()
+    df['DESCRIPTION'] = df['DESCRIPTION'].str.strip()
+
+    # Sidebar Filters
     st.sidebar.header("Filter Options")
-    product_query = st.sidebar.text_input('Product Name:')
-    customer_query = st.sidebar.text_input('Customer Name:')
+
+    # Dropdown for Customer Selection
+    customer_list = df['CUSTOMER_NAME'].dropna().unique().tolist()
+    customer_list.sort()
+    customer_query = st.sidebar.selectbox('Select Customer:', options=["All"] + customer_list)
+
+    # Dropdown for Product Selection
+    product_list = df['DESCRIPTION'].dropna().unique().tolist()
+    product_list.sort()
+    product_query = st.sidebar.selectbox('Select Product:', options=["All"] + product_list)
+
+    # Date Range Selection
     start_date = st.sidebar.date_input('Start Date:', df['INVOICE_DATE'].min())
     end_date = st.sidebar.date_input('End Date:', df['INVOICE_DATE'].max())
 
@@ -29,11 +43,13 @@ if uploaded_file:
         (df['INVOICE_DATE'] >= pd.to_datetime(start_date)) &
         (df['INVOICE_DATE'] <= pd.to_datetime(end_date))
     ]
-    if product_query:
-        filtered_df = filtered_df[filtered_df['DESCRIPTION'].str.contains(product_query, case=False, na=False)]
-    if customer_query:
+
+    if customer_query != "All":
         filtered_df = filtered_df[filtered_df['CUSTOMER_NAME'].str.contains(customer_query, case=False, na=False)]
 
+    if product_query != "All":
+        filtered_df = filtered_df[filtered_df['DESCRIPTION'].str.contains(product_query, case=False, na=False)]
+
     # Display Filtered Data
     st.write(f"Filtered Data: {len(filtered_df)} records found.")
     st.dataframe(filtered_df)