From 818024ffeca89cca9e17aa1013e648a85f7e07cd Mon Sep 17 00:00:00 2001
From: DeveloperSamir <61192453+DeveloperSamir@users.noreply.github.com>
Date: Sun, 1 Mar 2020 05:15:30 -0500
Subject: [PATCH] Update main.py

---
 main.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/main.py b/main.py
index 5dbd952..f9bc42d 100644
--- a/main.py
+++ b/main.py
@@ -1,6 +1,6 @@
 from flask import Flask, render_template, request, session, redirect
 from flask_sqlalchemy import SQLAlchemy
-from werkzeug import secure_filename
+from werkzeug.utils import secure_filename
 from flask_mail import Mail
 import json
 import os
@@ -149,7 +149,7 @@ def uploader():
 
 @app.route("/logout")
 def logout():
-    session.pop('user')
+    session.pop('user', None)
     return redirect('/dashboard')
 
 @app.route("/delete/<string:sno>", methods = ['GET', 'POST'])