Update app.py

app.py

@@ -1,13 +1,5 @@
 import streamlit as st
-
-# Display the logo with your updated link
-st.image(
-    "https://huggingface.co/spaces/taarhissian/SentinelCryptography/resolve/main/sentinelcryptography.png",
-    caption="SentinelCryptography Logo",
-    use_container_width=True
-)
-
-
+import os
 import random
 import base64
 from cryptography.hazmat.primitives import hashes
@@ -15,117 +7,123 @@ from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes
 from cryptography.hazmat.backends import default_backend
 from cryptography.hazmat.primitives.kdf.pbkdf2 import PBKDF2HMAC
 
+# --- Sentinel Engine Functions ---
 
-# Key generation function
-def generate_key(seed: str, dimensions: int):
+def generate_key(seed: str, dimensions: int = 5):
+    """Sentinel multi-dimensional key derivation"""
     kdf = PBKDF2HMAC(
-        algorithm=hashes.SHA256(), length=32, salt=b'salt', iterations=100000
+        algorithm=hashes.SHA256(),
+        length=32,
+        salt=b'sentinel_salt',  # change to a secure static salt or dynamic
+        iterations=200000,      # higher iterations for quantum resilience
+        backend=default_backend()
     )
     key = kdf.derive(seed.encode())
+    # Multi-dimensional expansion (adaptive randomness)
     key += bytes([random.randint(0, 255) for _ in range(dimensions * 16)])
     return key[:32]
 
-# Encryption function
-def encrypt_data(data, key):
-    iv = random.randbytes(16)
+def encrypt_data(data: bytes, key: bytes) -> bytes:
+    iv = os.urandom(16)  # cryptographically secure IV
     cipher = Cipher(algorithms.AES(key), modes.CBC(iv), backend=default_backend())
     encryptor = cipher.encryptor()
 
-    padding_length = 16 - len(data) % 16
-    data += bytes([padding_length] * padding_length)
+    padding_len = 16 - len(data) % 16
+    data += bytes([padding_len] * padding_len)
 
-    encrypted_data = encryptor.update(data) + encryptor.finalize()
-    return iv + encrypted_data
+    encrypted = encryptor.update(data) + encryptor.finalize()
+    return iv + encrypted
 
-# Decryption function
-def decrypt_data(encrypted_data, key):
+def decrypt_data(encrypted_data: bytes, key: bytes) -> bytes:
     iv = encrypted_data[:16]
     encrypted_data = encrypted_data[16:]
 
     cipher = Cipher(algorithms.AES(key), modes.CBC(iv), backend=default_backend())
     decryptor = cipher.decryptor()
-    decrypted_data = decryptor.update(encrypted_data) + decryptor.finalize()
+    decrypted = decryptor.update(encrypted_data) + decryptor.finalize()
+
+    padding_len = decrypted[-1]
+    return decrypted[:-padding_len]
 
-    padding_length = decrypted_data[-1]
-    return decrypted_data[:-padding_length]
+# --- Streamlit UI ---
+
+# Display Logo
+st.image(
+    "https://huggingface.co/spaces/taarhissian/SentinelCryptography/resolve/main/sentinelcryptography.png",
+    caption="SentinelCryptography Logo",
+    use_container_width=True
+)
 
-# Streamlit UI
-# --- Hero Section ---
 st.title("🛡 SentinelCryptography")
 st.subheader("A Quantum-Ready Encryption Guardian")
 st.write("""
-Not just another online AES tool. SentinelCryptography is designed as a living sentinel — 
-built for today’s digital security, aware of tomorrow’s quantum threats.
+SentinelCryptography is not just an AES tool — it is a living sentinel, designed to protect your messages 
+today and withstand the cryptographic challenges of tomorrow.
 """)
 
-# --- Encryption / Decryption Tool Placeholder ---
-st.markdown("---")
-st.header("🔑 Try it Out")
-st.write("Encrypt or decrypt your message with AES-256-CBC and our custom key derivation.")
-# (Insert your encrypt/decrypt input/output code here)
+# --- Session State for Seed ---
+if "seed" not in st.session_state:
+    st.session_state.seed = ""
 
-# --- Knowledge Section ---
+# --- Encryption Section ---
+st.markdown("---")
+st.header("🔒 Encrypt a Message")
+message = st.text_area("Enter Message to Encrypt:")
+st.session_state.seed = st.text_input("Enter Seed/Password for Key Generation:", st.session_state.seed)
+
+if st.button("Encrypt"):
+    if message and st.session_state.seed:
+        key = generate_key(st.session_state.seed, dimensions=5)
+        encrypted = encrypt_data(message.encode(), key)
+        st.success("✅ Encryption Complete!")
+        st.code(base64.b64encode(encrypted).decode(), language="text")
+    else:
+        st.error("Please enter both a message and a seed/password.")
+
+# --- Decryption Section ---
+st.markdown("---")
+st.header("🔓 Decrypt a Message")
+encrypted_input = st.text_area("Enter Encrypted Message (Base64):")
+
+if st.button("Decrypt"):
+    if encrypted_input and st.session_state.seed:
+        try:
+            encrypted_bytes = base64.b64decode(encrypted_input)
+            key = generate_key(st.session_state.seed, dimensions=5)
+            decrypted = decrypt_data(encrypted_bytes, key)
+            st.success("✅ Decryption Complete!")
+            st.code(decrypted.decode(), language="text")
+        except Exception as e:
+            st.error(f"Decryption failed: {e}")
+    else:
+        st.error("Please enter both an encrypted message and the seed/password.")
+
+# --- Educational & Roadmap Sections ---
 st.markdown("---")
 st.header("📜 The Legacy of Encryption")
-
-st.write("""
-Encryption has evolved across human history:
-- **Ancient Era** — Substitution ciphers like Caesar’s shift cipher.
-- **Mechanical Age** — Enigma machines and rotor-based systems in WWII.
-- **Modern Cryptography** — DES (broken by brute force), then AES chosen as the global standard.
-""")
-
 st.write("""
-But while AES became the backbone of secure communication, many **online tools** that claim to 
-offer AES encryption are built on outdated or weakened foundations.
+Encryption has evolved from ancient substitution ciphers to mechanical rotors in WWII, 
+and finally to AES in the modern era. SentinelCryptography pushes this evolution further, 
+preparing for the quantum era.
 """)
 
-# --- Weaknesses of Traditional Engines ---
 st.markdown("---")
-st.header("⚠️ Weaknesses of Traditional Online Engines")
-
-st.markdown("""
-- **AES-128 vs AES-256**  
-  AES-128 is mathematically strong today, but quantum algorithms (Grover’s) could cut its strength in half.  
-  AES-256 is far more resilient against this pressure.
-
-- **SHA-1 HMAC**  
-  Many tools use HMAC with SHA-1, which has been broken since 2017 (Google’s SHAttered attack).  
-  Even in HMAC, SHA-1 is considered unsafe for the future.
-
-- **CBC Mode Done Wrong**  
-  CBC requires a fresh random IV for each message. Many online engines ignore this, leaking patterns.
-
-- **No Key Derivation**  
-  User passwords are often treated directly as AES keys, making brute-force far easier.  
-  Sentinel uses a **custom Key Derivation Function (KDF)** for stronger defense.
-
-- **No Quantum Awareness**  
-  Traditional engines assume the world will remain pre-quantum. Sentinel does not.
-""")
-
-# --- Sentinel Difference ---
-st.markdown("---")
-st.header("🌐 The Sentinel Difference")
-
-st.markdown("""
-- **AES-256-CBC + Custom KDF** — stronger, adaptive protection.  
-- **Quantum-Resilient Design** — prepared for the next era of attacks.  
-- **Transparent & Open Source** — no black box, open to community audit.  
-- **Educational by Design** — teaching as much as it protects.  
-
-In short: where others offer a **lock**, Sentinel offers a **living shield**.
+st.header("🚀 Roadmap for 2025")
+st.write("""
+1. Q1: Expand post-quantum cryptographic support.  
+2. Q2: APIs for web & mobile integration.  
+3. Q3: Community challenges to improve security.  
+4. Q4: Launch full SentinelCryptography platform with analytics and advanced features.
 """)
+st.image("https://via.placeholder.com/800x400.png?text=SentinelCryptography+2025+Roadmap")
 
-# --- Future Roadmap ---
+# --- Donation Section ---
 st.markdown("---")
-st.header("🚀 Roadmap Ahead")
-
+st.header("💖 Support SentinelCryptography")
 st.write("""
-- Mobile and browser extension versions.  
-- File encryption and integration with decentralized storage.  
-- Visual encryption flow diagrams for education.  
-- Hackathons and community feature-building.  
+Your donations help us continue development and keep this project open source.
+**Donate via Tether (USDT-ERC20)**: `0xD5255523b1A6a144D9d034c70735b2dB729B49f0`
+Every contribution empowers global security for the quantum era.
 """)
 
-st.success("✨ SentinelCryptography — Built not just to encrypt, but to guard, teach, and endure.")
+st.info("SentinelCryptography is open source. Together, we secure the future.")