app.py

import streamlit as st
from streamlit_lottie import st_lottie
import pickle
import joblib
import numpy as np
import requests
from PIL import Image
from nltk.corpus import stopwords
from nltk.tokenize import word_tokenize
from nltk.stem import PorterStemmer,WordNetLemmatizer
import nltk
import string
nltk.download('stopwords')
nltk.download('punkt')
nltk.download('wordnet')


image = Image.open('./img/spam.png')
st.set_page_config(page_title='Spam Detection', page_icon=image)


def load_lottie(url):
    r = requests.get(url)
    if r.status_code != 200:
        return None
    return r.json()


# Initialize stemmer and lemmatizer
stemmer = PorterStemmer()
lemmatizer = WordNetLemmatizer()

# Define function to preprocess text
def preprocess_text(text):
    # Convert to lowercase
    text = text.lower()
    # Remove punctuation
    text = "".join([char for char in text if char not in string.punctuation])
    # Remove stopwords
    stop_words = set(stopwords.words('english'))
    text = " ".join([word for word in text.split() if word not in stop_words])
    # Tokenize
    tokens = nltk.word_tokenize(text)
    # Stem and lemmatize
    lemmatized_tokens = [lemmatizer.lemmatize(token) for token in tokens]
    stemmed_tokens = [stemmer.stem(token) for token in lemmatized_tokens]
    # Join tokens back into text string
    preprocessed_text = " ".join(stemmed_tokens)
    return preprocessed_text



# load the saved objects
cv = pickle.load(open('count_vectorizer.pkl', 'rb'))
model = joblib.load(open('model_NB','rb'))



st.title("Spam Detection System")
st.write("\t\t\tV 1.0")
st.write('---')

animation_spam = load_lottie('https://assets7.lottiefiles.com/private_files/lf30_prqvme9e.json')

with st.container():
    right_column, left_column = st.columns(2)
    with right_column:
        # Enter the message you want to classify
        new_message = st.text_area('Enter your message :', height=100)
        
        if st.button('Predict'):

            # Doind preprocessing on the text I get
            new_message = preprocess_text(new_message)

            # transform the new message using the loaded CountVectorizer object
            new_bow_features = cv.transform([new_message])

            # Predict
            result = model.predict(new_bow_features)

            # Display
            if result == 1:
                st.error("Spam")
            elif result == 0:
                st.success("Ham 'Not Spam'")

    with left_column:
        st_lottie(animation_spam, speed=0.5, height=250, key="secoend")



st.write('---')
st.write('')

animation_contact = load_lottie("https://assets5.lottiefiles.com/packages/lf20_ejryvjev.json")
with st.container():
    right_column, left_column = st.columns(2)
    with right_column:

        st.write('')
        st.write('')
        st.write('')
        st.write('_For any issue contact me via :_')

        '''

            [![Github](https://skillicons.dev/icons?i=github)](https://github.com/aliabdallah7)
              
            [![LinkedIn](https://skillicons.dev/icons?i=linkedin)](https://www.linkedin.com/in/ali-abdallah7/)
              
            [![Gmail](https://img.icons8.com/color/70/gmail-new.png)](mailto:ali.abdallah43792@gmail.com)

        '''

    with left_column:
        st_lottie(animation_contact, speed=1, height=200, key="third")


footer = """<style>
div.stButton > button:first-child {
  padding: 10px 20px;
  border: 2px solid #ff4d4d;
  background-color: #ffcdd2;
  color: #c62828;
  font-size: 16px;
  font-weight: bold;
  border-radius: 5px;
  box-shadow: 0px 2px 5px rgba(0, 0, 0, 0.5);
  transition: all 0.2s ease;
}
div.stButton > button:hover:first-child {
  background-color: #ff4d4d;
  color: #fff;
  box-shadow: 0px 2px 8px rgba(0, 0, 0, 0.8);
}
div.stButton > button:active:first-child {
  background-color: #c62828;
  color: #fff;
  box-shadow: inset 0px 2px 5px rgba(0, 0, 0, 0.5);
}

header {visibility: hidden;}

/* Light mode styles */
.my-paragraph {
  color: black;
}

/* Dark mode styles */
@media (prefers-color-scheme: dark) {
  .my-paragraph {
    color: white;
  }
}

a:link , a:visited{
color: #5C5CFF;
background-color: transparent;
text-decoration: none;
}

a:hover,  a:active {
color: red;
background-color: transparent;
text-decoration: underline;
}

:root {
  --footer-bg-color: #333;
}

@media (prefers-color-scheme: dark) {
  :root {
    --footer-bg-color: rgb(14, 17, 23);
  }
}

@media (prefers-color-scheme: light) {
  :root {
    --footer-bg-color: white;
  }
}

.footer {
position: fixed;
left: 0;
bottom: 0;
width: 100%;
background-color: var(--footer-bg-color);
color: black;
text-align: center;
}

</style>
<div class="footer">
<p class="my-paragraph">&copy; 2023 <a href="https://www.linkedin.com/in/ali-abdallah7/"> Ali Abdallah</a></p>
</div>
"""
st.markdown(footer,unsafe_allow_html=True)
st.markdown("<br>",unsafe_allow_html=True)