Update app.py

app.py

@@ -1,24 +1,23 @@
 
-# Function to hash passkey
 def hash_passkey(passkey):
     return hashlib.sha256(passkey.encode()).hexdigest()
 
-# Function to generate a key from passkey (for encryption)
 def generate_key_from_passkey(passkey):
     hashed = hashlib.sha256(passkey.encode()).digest()
     return base64.urlsafe_b64encode(hashed[:32])
 
-# Function to encrypt data
 def encrypt_data(text, passkey):
     key = generate_key_from_passkey(passkey)
     cipher = Fernet(key)
     return cipher.encrypt(text.encode()).decode()
 
-# Function to decrypt data
 def decrypt_data(encrypted_text, passkey, data_id):
     try:
         hashed_passkey = hash_passkey(passkey)
-        if data_id in streamlit.session_state.stored_data and streamlit.session_state.stored_data[data_id]["passkey"] == hashed_passkey:
+        if (
+            data_id in streamlit.session_state.stored_data and
+            streamlit.session_state.stored_data[data_id]["passkey"] == hashed_passkey
+        ):
             key = generate_key_from_passkey(passkey)
             cipher = Fernet(key)
             decrypted = cipher.decrypt(encrypted_text.encode()).decode()
@@ -33,23 +32,23 @@ def decrypt_data(encrypted_text, passkey, data_id):
         streamlit.session_state.last_attempt_time = time.time()
         return None
 
-# Function to reset failed attempts
 def reset_failed_attempts():
     streamlit.session_state.failed_attempts = 0
 
-# Function to change page
 def change_page(page):
     streamlit.session_state.current_page = page
 
+# ---------------------- Main App ----------------------
+
 def main():
-    # Initialize session state variables
-    if 'failed_attempts' not in streamlit.session_state:
+    # Initialize session state
+    if "failed_attempts" not in streamlit.session_state:
         streamlit.session_state.failed_attempts = 0
-    if 'stored_data' not in streamlit.session_state:
+    if "stored_data" not in streamlit.session_state:
         streamlit.session_state.stored_data = {}
-    if 'current_page' not in streamlit.session_state:
+    if "current_page" not in streamlit.session_state:
         streamlit.session_state.current_page = "Home"
-    if 'last_attempt_time' not in streamlit.session_state:
+    if "last_attempt_time" not in streamlit.session_state:
         streamlit.session_state.last_attempt_time = 0
 
     streamlit.title("🔒 Secure Data Encryption System")
@@ -59,12 +58,14 @@ def main():
     choice = streamlit.sidebar.selectbox("Navigation", menu, index=menu.index(streamlit.session_state.current_page))
     streamlit.session_state.current_page = choice
 
-    # Block if too many failed attempts
+    # Handle lockout
     if streamlit.session_state.failed_attempts >= 3:
         streamlit.session_state.current_page = "Login"
         streamlit.warning("🔒 Too many failed attempts! Reauthorization required.")
 
-    # HOME PAGE
+    # ---------------------- Pages ----------------------
+
+    # Home Page
     if streamlit.session_state.current_page == "Home":
         streamlit.subheader("🏠 Welcome to the Secure Data System")
         streamlit.write("Use this app to **securely store and retrieve data** using unique passkeys.")
@@ -79,7 +80,7 @@ def main():
 
         streamlit.info(f"Currently storing {len(streamlit.session_state.stored_data)} encrypted data entries.")
 
-    # STORE DATA PAGE
+    # Store Data Page
     elif streamlit.session_state.current_page == "Store Data":
         streamlit.subheader("📂 Store Data Securely")
         user_data = streamlit.text_area("Enter Data:")
@@ -106,7 +107,7 @@ def main():
             else:
                 streamlit.error("⚠️ All fields are required!")
 
-    # RETRIEVE DATA PAGE
+    # Retrieve Data Page
     elif streamlit.session_state.current_page == "Retrieve Data":
         streamlit.subheader("🔍 Retrieve Your Data")
         attempts_remaining = 3 - streamlit.session_state.failed_attempts
@@ -137,7 +138,7 @@ def main():
             else:
                 streamlit.error("⚠️ Both fields are required!")
 
-    # LOGIN PAGE
+    # Login Page
     elif streamlit.session_state.current_page == "Login":
         streamlit.subheader("🔑 Reauthorization Required")
 
@@ -160,6 +161,7 @@ def main():
     streamlit.markdown("---")
     streamlit.markdown("🔐 Secure Data Encryption System | Educational Project")
 
-# Run app
+# ---------------------- Entry Point ----------------------
+
 if __name__ == "__main__":
     main()