Update app.py

app.py

@@ -2,31 +2,21 @@ import streamlit as st
 import random
 import string
 
-# Helper functions for substitution cipher
-def categorize_letters():
-    vowels = "AEIOUYaeiouy"
-    consonants = "".join([c for c in string.ascii_letters if c not in vowels])
-    return vowels, consonants
+# Generate a unique random character for each letter of the alphabet
+def generate_variable_mapping():
+    letters = string.ascii_letters  # Includes both lowercase and uppercase letters
+    shuffled_characters = list(set(string.ascii_letters + string.digits + string.punctuation))
+    random.shuffle(shuffled_characters)
 
-def generate_cipher_map(plaintext):
-    vowels, consonants = categorize_letters()
-    symbol_pool = string.ascii_letters + string.digits + string.punctuation
-    cipher_map = {}
+    variable_mapping = {}
+    for letter in letters:
+        variable_mapping[letter] = shuffled_characters.pop()
 
-    # Create substitution map for each letter in the plaintext
-    for letter in set(plaintext):
-        if letter in vowels:
-            cipher_map[letter] = random.choice([c for c in symbol_pool if c in vowels])
-        elif letter in consonants:
-            cipher_map[letter] = random.choice([c for c in symbol_pool if c in consonants])
-        else:
-            cipher_map[letter] = letter  # keep spaces and non-alphabetic characters as is
+    return variable_mapping
 
-    return cipher_map
-
-def generate_ciphertext(plaintext, cipher_map):
-    # Create the encrypted text using the cipher map
-    return "".join([cipher_map.get(char, char) for char in plaintext])
+def encrypt_text(plaintext, mapping):
+    # Encrypt the plaintext using the variable mapping
+    return "".join(mapping.get(char, char) for char in plaintext)
 
 # Helper words list with common French words for each alphabet letter
 helper_words = {
@@ -58,7 +48,7 @@ helper_words = {
     'Z': ["zèbre", "zone", "zinc", "zodiac", "zigzag"]
 }
 
-def generate_helper_phrase(ciphertext, cipher_map):
+def generate_helper_phrase(ciphertext, mapping):
     helper_phrase = []
 
     # Generate three words for each letter in the ciphertext
@@ -71,8 +61,8 @@ def generate_helper_phrase(ciphertext, cipher_map):
     # Join the helper phrase into a single string
     helper_phrase_str = " ".join(helper_phrase)
     
-    # Encrypt the entire helper phrase using the same cipher map
-    encrypted_helper_phrase = generate_ciphertext(helper_phrase_str, cipher_map)
+    # Encrypt the entire helper phrase using the same variable mapping
+    encrypted_helper_phrase = encrypt_text(helper_phrase_str, mapping)
 
     return encrypted_helper_phrase
 
@@ -84,9 +74,9 @@ st.write("Ce système remplace chaque lettre d'un texte par un caractère aléat
 plaintext = st.text_input("Entrez le texte à chiffrer:", "Exemple de texte")
 
 if st.button("Chiffrer"):
-    cipher_map = generate_cipher_map(plaintext)
-    ciphertext = generate_ciphertext(plaintext, cipher_map)
-    helper_phrase = generate_helper_phrase(ciphertext, cipher_map)
+    mapping = generate_variable_mapping()
+    ciphertext = encrypt_text(plaintext, mapping)
+    helper_phrase = generate_helper_phrase(ciphertext, mapping)
 
     # Display the encrypted text and helper phrase
     st.write("Texte chiffré:")
@@ -95,4 +85,4 @@ if st.button("Chiffrer"):
     st.write("Phrase d'aide:")
     st.code(helper_phrase, language="")
 
-    st.write("Note: Utilisez la phrase d'aide pour déduire les correspondances des lettres.")
+    st.write("Note: Utilisez la phrase d'aide pour déduire les correspondances des lettres.")