Update app.py

app.py

@@ -7,7 +7,6 @@ import os
 
 app = Flask(__name__, static_folder='static')
 app.secret_key = b'_5#y2L"F4Q8z\n\xec]/' # Change this key for secure deployment
-csrf = CSRFProtect(app)
 
 SESSION_DICT = {}
 
@@ -55,13 +54,13 @@ def id_type():
 @app.route("/id_type/", methods=['POST'])
 def id_type_done():
     add_to_session('id_type', request.values['ocr_type'])
-    return redirect('/id_file/')
+    return redirect(url_for('id_file'))
 
 @app.route("/id_file/", methods=['GET'])
 def id_file():
     id_type = get_session_data('id_type')
     if id_type is None:
-        return redirect('/id_type')
+        return redirect(url_for('id_type'))
     
     id_file = get_session_data('id_file')
     if id_type == 'ic':
@@ -82,19 +81,19 @@ def id_file_done():
         if id_file is not None and id_file['image2'] == "":
             id_file['image2'] = id_base64
             add_to_session('id_file', id_file)
-            return redirect('/id_detail/')
+            return redirect(url_for('id_detail'))
         else:
             add_to_session('id_file', {
                 "image": id_base64,
                 "image2": ""
             })
-            return redirect('/id_file/')
+            return redirect(url_for('id_file'))
     else:
         add_to_session('id_file', {
             "image": id_base64,
             "image2": ""
         })
-        return redirect('/id_detail/')
+        return redirect(url_for('id_detail'))
 
 @app.route("/id_detail/", methods=['GET'])
 def id_detail():
@@ -115,11 +114,11 @@ def id_detail():
         
         return render_template('id_detail.html', ocr_result = json.dumps(res))
     else:
-        return redirect('/id_type/')
+        return redirect(url_for('id_type'))
 
 @app.route("/id_detail/", methods=['POST'])
 def id_detail_done():
-    return redirect('/fr_file/')
+    return redirect(url_for('fr_file'))
 
 @app.route("/fr_file/", methods=['GET'])
 def fr_file():
@@ -129,13 +128,13 @@ def fr_file():
 def fr_file_done():
     fr_base64 = request.values['fr_file']
     add_to_session('fr_selfie', fr_base64)
-    return redirect('/fr_detail')
+    return redirect(url_for('fr_detail'))
 
 @app.route("/fr_detail/", methods=['GET'])
 def fr_detail():
     sess_id_front = get_session_data('id_front')
     if sess_id_front is None:
-        return redirect('/id_type/')
+        return redirect(url_for('id_type'))
     
     sess_fr_selfie = get_session_data('fr_selfie')
     if sess_fr_selfie is not None:
@@ -171,7 +170,7 @@ def fr_detail():
                                recog_res = recog_res.upper(), 
                                recog_similarity = recog_similarity * 100)
     else:
-        return redirect('/fr_file/')
+        return redirect(url_for('fr_file'))
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)