app.py

from flask import Flask, render_template,request,redirect,url_for,session,flash
import pickle
import string
import nltk
from nltk.stem import PorterStemmer
import mysql.connector 

app = Flask(__name__)
app.secret_key = '1c8073775dbc85a92ce20ebd44fd6a4fd832078f59ef16ec'
ps = PorterStemmer()
with open('vectorizer.pkl', 'rb') as file:
    tfidf = pickle.load(file)
with open('model.pkl', 'rb') as file:
    model = pickle.load(file)

nltk.download('punkt')

def transform_text(text):
    text = text.lower()
    text = nltk.word_tokenize(text)
    
    y = []
    for i in text:
        # Check if the word is alphanumeric
        if i.isalnum():
            y.append(i)
    
    text = y[:]
    y.clear()  #The list y is cleared to free up memory
    
    for i in text:
        y.append(ps.stem(i))  #Stemming Each Word:(արմատը)
        
    return " ".join(y)   #join the processed words back into a single string separated by spaces

#Defining database connction
db = mysql.connector.connect(
    host = "localhost",
    user = "root",
    password = "ասդֆ",
    database = "smc"
)

@app.route('/')
def home():
    return render_template('home.html')

@app.route('/about')
def about():
    return render_template('about.html')

@app.route('/index')
def index():
    if 'user' in session:    # Check if the 'user' session variable exists (i.e., the user is logged in)
        return render_template('index.html')  
    else:
        return redirect(url_for('signin'))   # Redirect to the sign-in page if the user is not logged in
    
@app.route('/predict', methods=['POST'])
def predict():
    input_email = request.form.get('message')
    transformed_email =transform_text(input_email)
    vector_input = tfidf.transform([transformed_email])
    result = model.predict(vector_input)[0]
    if result == 0:
        prediction = 'Spam'
    else:
        prediction = 'Ham/not spam'
    return render_template('result.html',prediction=prediction)

@app.route('/signin')
def signin():
    if 'user' in session:
        return redirect(url_for('index'))
    return render_template('signin.html')

@app.route('/signup', methods=['GET'])
def register():
    if request.method == 'POST':
        full_name = request.form['full_name']   
        username = request.form['username']
        email = request.form['email']
        phone = request.form['phone']
        password = request.form['password']
        
        # Ensure the password and confirm_password match
        confirm_password = request.form['confirm_password']
        if password != confirm_password:
            return "Password and confirm_password don't match."
        
        # Insert data into MySQL
        cur = db.cursor()
        cur.execute("INSERT INTO users  (full_name, username, emsil, phone, password) VALUES (%s, %s, %s, %s, %s)",
                    (full_name, username, email, phone, password))   
        db.commit()
        cur.close()
        
        flash('Registration successful','success')
        return redirect('/signin')
    
    return "Invalid request method"

@app.route('/login', methods=['POST'])
def login():
    if request.method == 'POST':
        email = request.form['email']
        password = request.form['password']
        remember_me = request.form.get('remember_me')
        
        # Query the database to check if the email and password match
        cur = db.cursor()
        cur.execute("SELECT * FROM users WHERE email = %s AND password = %s", (email,password))
        user = cur.fetchone()  #If a user with the given email and password is found,  user will contain that user's data.
        cur.close()
        
        if user:
            session['user'] = user
            
            if remember_me:
                session.permanent = True
            return redirect(url_for('index'))
        else:
            return "Login failed.Check your email and password."
    return "Invalid request method"

@app.route('/logout')
def logout():
    session.pop('user',None)
    return redirect(url_for('home'))  # Redirect to the sign-in page after logging out

if __name__ == '__main__':
    app.run(debug=True)