Update src/streamlit_app.py

src/streamlit_app.py

@@ -15,13 +15,22 @@ st.markdown("This app uses **Machine Learning** (Naive Bayes + TF-IDF) to classi
 
 # ----------------- DATA LOADING -----------------
 @st.cache_data
-def load_data():
-    df = pd.read_csv("spam.csv", encoding='latin-1')[['v1', 'v2']]
+def load_data(file):
+    df = pd.read_csv(file, encoding='latin-1')[['v1', 'v2']]
     df.columns = ['label', 'message']
     df['label'] = df['label'].map({'ham': 0, 'spam': 1})
     return df
 
-df = load_data()
+# File uploader for user dataset
+st.subheader("📂 Upload Dataset")
+uploaded_file = st.file_uploader("Upload your spam dataset (CSV format)", type=["csv"])
+
+if uploaded_file is not None:
+    df = load_data(uploaded_file)
+    st.success("✅ Dataset loaded successfully from uploaded file.")
+else:
+    st.info("ℹ️ No file uploaded. Using default dataset (spam.csv).")
+    df = load_data("spam.csv")
 
 # ----------------- PREPROCESS FUNCTION -----------------
 def clean_text(text):