Update app.py

app.py

@@ -43,13 +43,13 @@ import ezdxf
 from flask import Flask, render_template, session, redirect, url_for
 from flask_session import Session  # Capital "S"
 app = Flask(__name__)
-  
+
 
 # Configure Flask-Session for server-side session storage
-app.config["SESSION_TYPE"] = "filesystem"  # Using filesystem for session storage (change as needed)
-app.config["SESSION_PERMANENT"] = False  # Sessions are not permanent
-app.config["SESSION_FILE_DIR"] = "/tmp"  # Temporary storage directory (adjust for Hugging Face)
-app.secret_key = "your_secret_key"  # Replace with a secure key for session signing
+app.config["SESSION_TYPE"] = "filesystem"  
+app.config["SESSION_PERMANENT"] = False
+app.config["SESSION_FILE_DIR"] = "./flask_session_files"  # Local directory for session files (use a persistent directory)
+app.secret_key = "your_secret_key"  # Replace with a secure key
 
 # Initialize Flask-Session
 sess = Session()
@@ -65,24 +65,25 @@ global pdflink
 # For 2.7
 global hatched_areas2_7
 
+# Home Route (Displays password page initially)
 @app.route("/", methods=["GET", "POST"])
 def getInfotoMeasure():
     return render_template("gui2.html")
 
-# Password Page Route
+# Password Page Route (For entering the password)
 @app.route("/password", methods=["GET", "POST"])
 def password_page():
     return render_template("gui2.html")
 
-# Check Password Route
+# Check Password Route (Validates the password)
 @app.route("/check_password", methods=["POST"])
 def check_password():
     password = request.form.get("password")
-    correct_password = "c900"  # Correct password
+    correct_password = "c900"
 
-    # Check if the entered password matches the correct password
     if password == correct_password:
         session["authenticated"] = True  # Store authentication in session
+        session.modified = True  # Mark session as modified to ensure it is saved
         print("Password correct. Session data set:", session)  # Debugging: Print session after setting it
         return jsonify({"authenticated": True}), 200
     else:
@@ -93,13 +94,13 @@ def check_password():
 def main_gui():
     print("Session data at mainGUI:", session)  # Debugging: Check session data
 
-    # Debugging session contents before redirection
-    if "authenticated" not in session:
-        print("Session does not contain 'authenticated'")  # Debugging
+    # Check if the session contains 'authenticated' and its value
+    if "authenticated" in session:
+        print(f"Session contains 'authenticated': {session['authenticated']}")  # Debugging
     else:
-        print(f"Session['authenticated'] = {session['authenticated']}")  # Debugging
+        print("Session does not contain 'authenticated'")  # Debugging
 
-    # If the session does not have "authenticated" or it's False, redirect to the password page
+    # If session does not have "authenticated" or it's False, redirect to password page
     if "authenticated" not in session or not session["authenticated"]:
         print("User not authenticated, redirecting to password page.")  # Debugging
         return redirect(url_for("password_page"))