From 2d407a53f0f3579a58bb7262d74fd5a536a0a311 Mon Sep 17 00:00:00 2001
From: happyZYM <zhuangyumin@sjtu.edu.cn>
Date: Sat, 22 Jun 2024 04:49:45 +0000
Subject: [PATCH] add login failure warning

---
 UI/main.py              |  2 +-
 UI/templates/login.html | 15 +++++++++++++++
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/UI/main.py b/UI/main.py
index 04a3e95..2afeaee 100644
--- a/UI/main.py
+++ b/UI/main.py
@@ -83,7 +83,7 @@ def login():
         else:
             error = 'An unknown error occurred.'
         
-        flash(error, 'error')
+        flash(error, 'danger')
         return redirect(url_for('login'))
     
     return render_template('login.html', form=form)
diff --git a/UI/templates/login.html b/UI/templates/login.html
index 7690815..a5551d9 100644
--- a/UI/templates/login.html
+++ b/UI/templates/login.html
@@ -6,6 +6,21 @@
 <div class="row justify-content-center">
     <div class="col-md-6">
         <h2 class="text-center">Login</h2>
+        
+        <!-- Flash messages -->
+        {% with messages = get_flashed_messages(with_categories=true) %}
+            {% if messages %}
+                {% for category, message in messages %}
+                    <div class="alert alert-{{ category }} alert-dismissible fade show" role="alert">
+                        {{ message }}
+                        <button type="button" class="close" data-dismiss="alert" aria-label="Close">
+                            <span aria-hidden="true">&times;</span>
+                        </button>
+                    </div>
+                {% endfor %}
+            {% endif %}
+        {% endwith %}
+        
         <form id="loginForm" class="needs-validation" novalidate method="POST" action="/login">
             {{ form.hidden_tag() }}
             <div class="form-group">