Update app.py

app.py

@@ -32,8 +32,12 @@ if uploaded_file is not None:
 
     df = data.copy()
 
-    # Rename columns
-    df = df.rename(columns={'Order Date': 'order_date', 'Category': 'category', 'Sub-Category': 'sub_category', 'Sales': 'sales'})
+    # Rename columns (Check if 'Sales' exists)
+    if 'Sales' in df.columns:
+        df = df.rename(columns={'Order Date': 'order_date', 'Category': 'category', 'Sub-Category': 'sub_category', 'Sales': 'sales'})
+    else:
+        st.error("🚨 'Sales' column not found in the uploaded file. Please ensure the correct dataset is uploaded.")
+        st.stop()
 
     st.subheader("📊 Data Summary")
     st.write(df.describe())