Spam Shield is a machine learning-based phishing detection system designed to identify spam in SMS/mail and URLs. The system is built using HTML, CSS, and JavaScript for the frontend, with Python and Flask for backend integration. It also includes a browser extension to provide easy access to the main functionalities.
- SMS/Mail Spam Detection
- URL Spam Detection
- SMS/Mail Spam Detection: Collected from Kaggle, preprocessed, and trained using the Naive Bayes algorithm.
- URL Spam Detection: Collected from Kaggle, upgraded using the NLTK framework, and trained using Decision Tree, Random Forest, and Multilayer Perceptron algorithms.
- SMS/Mail Spam Detection:
- Algorithm: Multinomial Naive Bayes (best accuracy)
- Exported Models:
vectorizer.pkl,model.pkl
- URL Spam Detection:
- Algorithms: Decision Tree, Random Forest, Multilayer Perceptron
- Framework: TensorFlow
- Exported Model:
model2.pkl
- Main File:
main.py - Frontend Files:
- HTML:
html-spamshield.html - CSS:
css-spamshield.css
- HTML:
- Backend Files:
- API for URL Detection:
API.py - URL Feature Extraction:
URL_features.py,Feature_extract.py
- API for URL Detection:
- Frontend Files:
- HTML:
html-popup.html - CSS:
css-styles.css - JavaScript:
js-popup.js
- HTML:
- Manifest File:
manifest.json
-
Clone the Repository
git clone [https://github.com/your-repo/spam-shield.git](https://github.com/Janani-m17/SPAM-SHIELD.git) cd spam-shield -
Install Dependencies
pip install -r requirements.txt
-
Run the Flask Application
python main.py
-
Open the Browser Extension
- Load the extension in your browser via the developer mode.
- Select the folder containing
html-popup.html,css-styles.css,js-popup.js, andmanifest.json.
-
SMS/Mail Spam Detection:
- Navigate to the website.
- Input the SMS or mail content in the provided field.
- Click "Check" to see if the content is spam.
-
URL Spam Detection:
- Navigate to the website.
- Input the URL in the provided field.
- Click "Check" to see if the URL is spam.
-
Browser Extension:
- Open the extension.
- Use the same functionalities as the website for quick access.
https://github.com/RithikaSundaram
https://www.github.com/swetha5157
For any inquiries, please contact janani.mkgs@gmail.com.