omroy07 · vakrahul · Aug 15, 2025
diff --git a/Labour_Alerts/templates/forgot_password.html b/Labour_Alerts/templates/forgot_password.html
@@ -0,0 +1,24 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <title>Forgot Password</title>
+</head>
+<body>
+    <h2>Forgot Password</h2>
+    {% with messages = get_flashed_messages(with_categories=true) %}
+      {% if messages %}
+        <ul class="flashes">
+        {% for category, message in messages %}
+          <li class="{{ category }}">{{ message }}</li>
+        {% endfor %}
+        </ul>
+      {% endif %}
+    {% endwith %}
+    <form action="{{ url_for('forgot_password') }}" method="post">
+        <label for="email">Enter your email:</label>
+        <input type="email" id="email" name="email" required>
+        <button type="submit">Reset Password</button>
+    </form>
+</body>
+</html>
diff --git a/Labour_Alerts/templates/reset_password.html b/Labour_Alerts/templates/reset_password.html
@@ -0,0 +1,28 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <title>Reset Password</title>
+</head>
+<body>
+    <h2>Reset Your Password</h2>
+    {% with messages = get_flashed_messages(with_categories=true) %}
+      {% if messages %}
+        <ul class="flashes">
+        {% for category, message in messages %}
+          <li class="{{ category }}">{{ message }}</li>
+        {% endfor %}
+        </ul>
+      {% endif %}
+    {% endwith %}
+    <form action="{{ url_for('reset_password', token=token) }}" method="post">
+        <label for="new_password">New Password:</label>
+        <input type="password" id="new_password" name="new_password" required>
+        <br>
+        <label for="confirm_password">Confirm Password:</label>
+        <input type="password" id="confirm_password" name="confirm_password" required>
+        <br>
+        <button type="submit">Submit</button>
+    </form>
+</body>
+</html>
diff --git a/app.py b/app.py
@@ -1,40 +1,131 @@
-from flask import Flask, render_template, request
-from utils import load_keras_model, predict_image_keras
+# --- All necessary imports for the Flask app and password reset ---
 import os
+import secrets
+import smtplib
+from email.mime.text import MIMEText
+from datetime import datetime, timedelta
+
+from flask import Flask, render_template, request, redirect, url_for, flash
 
 app = Flask(__name__)
+# IMPORTANT: Replace this with a strong, random key in a production environment
+app.config['SECRET_KEY'] = 'your_very_secret_key_here'
+
+# --- In-memory user data (simulating a database) ---
+# This data is not persistent. In a real application, you would use a database.
+# The 'password' value should be a hashed password in a real application.
+users = {
+    'testuser@example.com': {
+        'password': 'hashed_password_placeholder',
+        'reset_token': None,
+        'reset_token_expiration': None
+    }
+}
+# --- End of in-memory data ---
+
+# --- Email configuration ---
+# You MUST replace these with your actual email service details
+# Example for Gmail: EMAIL_HOST = 'smtp.gmail.com', EMAIL_PORT = 587
+EMAIL_HOST = 'smtp.example.com'
+EMAIL_PORT = 587
+EMAIL_USERNAME = 'your_email@example.com'
+EMAIL_PASSWORD = 'your_email_password'
+SENDER_EMAIL = 'your_email@example.com'
+
+# --- Email Sending Function ---
+def send_reset_email(user_email, reset_link):
+    """Sends a password reset email to the user."""
+    try:
+        msg = MIMEText(f"Click the link to reset your password: {reset_link}")
+        msg['Subject'] = "Password Reset Request"
+        msg['From'] = SENDER_EMAIL
+        msg['To'] = user_email
+
+        with smtplib.SMTP(EMAIL_HOST, EMAIL_PORT) as server:
+            server.starttls()
+            server.login(EMAIL_USERNAME, EMAIL_PASSWORD)
+            server.sendmail(SENDER_EMAIL, [user_email], msg.as_string())
+        return True
+    except Exception as e:
+        print(f"Error sending email: {e}")
+        return False
+
+# ------------------------------------------------------------------
+# --- Your Original Application Routes Go Here ---
+# You can copy your existing routes (e.g., '/', '/main', '/disease') here.
+# For example:
+# @app.route('/')
+# def index():
+#     return render_template('index.html')
+#
+# @app.route('/login')
+# def login():
+#     return render_template('login.html')
+# ------------------------------------------------------------------
+
+# --- New Password Reset Routes ---
+@app.route('/forgot_password', methods=['GET', 'POST'])
+def forgot_password():
+    """Handles the request for a password reset email."""
+    if request.method == 'POST':
+        email = request.form.get('email')
+        user = users.get(email)
+
+        if user:
+            # Generate a secure token with a one-hour expiration time
+            token = secrets.token_urlsafe(32)
+            expiration_time = datetime.now() + timedelta(hours=1)
+
+            user['reset_token'] = token
+            user['reset_token_expiration'] = expiration_time
 
-UPLOAD_FOLDER = r'disease/static/uploads'
-app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
-os.makedirs(UPLOAD_FOLDER, exist_ok=True)
+            reset_link = url_for('reset_password', token=token, _external=True)
+
+            if send_reset_email(email, reset_link):
+                flash("A password reset link has been sent to your email.", "success")
+            else:
+                flash("Failed to send email. Please try again.", "danger")
+        else:
+            flash("If the email is valid, a password reset link will be sent.", "info")
 
-# Load the Keras model
-model = load_keras_model(r'disease/model.h5')
+        return redirect(url_for('forgot_password'))
 
-# Route for homepage
-@app.route('/')
-def index():
-    return render_template('index.html')
+    return render_template('forgot_password.html')
 
-# Route for prediction
-@app.route('/predict', methods=['POST'])
-def predict():
-    if 'file' not in request.files:
-        return "No file uploaded."
-    file = request.files['file']
-    if file.filename == '':
-        return "No selected file."
+@app.route('/reset_password/<token>', methods=['GET', 'POST'])
+def reset_password(token):
+    """Handles the password reset form and logic."""
+    user_to_reset = None
+    for email, user in users.items():
+        # Find the user with the valid, non-expired token
+        if user['reset_token'] == token and user['reset_token_expiration'] > datetime.now():
+            user_to_reset = user
+            break
+
+    if not user_to_reset:
+        flash("Invalid or expired token.", "danger")
+        return redirect(url_for('forgot_password'))
 
-    filepath = os.path.join(app.config['UPLOAD_FOLDER'], file.filename)
-    file.save(filepath)
+    if request.method == 'POST':
+        new_password = request.form.get('new_password')
+        confirm_password = request.form.get('confirm_password')
 
-    predicted_class, description = predict_image_keras(model, filepath)
+        if new_password != confirm_password:
+            flash("Passwords do not match.", "danger")
+            return render_template('reset_password.html', token=token)
+
+        # In a real app, you would hash the password before storing it
+        user_to_reset['password'] = new_password
+
+        # Invalidate the token after use
+        user_to_reset['reset_token'] = None
+        user_to_reset['reset_token_expiration'] = None
 
-    return render_template('result.html',
-                           prediction=predicted_class,
-                           description=description,
-                           image_path=filepath)
+        flash("Your password has been reset successfully.", "success")
+        return redirect(url_for('forgot_password'))
 
+    return render_template('reset_password.html', token=token)
 
+# --- App entry point ---
 if __name__ == '__main__':
-    app.run(debug=True)
+    app.run(debug=True)