Update app.py

app.py

@@ -8,7 +8,7 @@ def categorize_letters():
     consonants = "".join([c for c in string.ascii_letters if c not in vowels])
     return vowels, consonants
 
-def generate_ciphertext(plaintext):
+def generate_cipher_map(plaintext):
     vowels, consonants = categorize_letters()
     symbol_pool = string.ascii_letters + string.digits + string.punctuation
     cipher_map = {}
@@ -22,46 +22,57 @@ def generate_ciphertext(plaintext):
         else:
             cipher_map[letter] = letter  # keep spaces and non-alphabetic characters as is
 
+    return cipher_map
+
+def generate_ciphertext(plaintext, cipher_map):
     # Create the encrypted text using the cipher map
-    ciphertext = "".join([cipher_map.get(char, char) for char in plaintext])
-    return ciphertext, cipher_map
+    return "".join([cipher_map.get(char, char) for char in plaintext])
 
-# Helper words list with repeated letters for each alphabet letter
+# Helper words list with common French words for each alphabet letter
 helper_words = {
-    'A': ["maman", "canal", "salade", "avalanche", "savane"],
-    'B': ["bobo", "babil", "babine", "bibine", "abbaye"],
-    'C': ["accroc", "accueil", "accuse", "cacao", "cocotte"],
-    'D': ["dada", "dedans", "dodo", "dédale", "doudou"],
-    'E': ["terre", "cerise", "serre", "essence", "berce"],
-    'F': ["fifre", "fouffe", "souffre", "coiffe", "affreux"],
-    'G': ["gaga", "gigogne", "gogo", "grignote", "goguette"],
-    'H': ["hahaha", "hache", "haha", "hérissé", "hocher"],
-    'I': ["ironie", "criique", "illisible", "risible", "immobile"],
-    'J': ["joujou", "jardinier", "jonjonaire", "jujube", "juxtapose"],
-    'K': ["kakapo", "kaki", "kéké", "kitoko", "kiwi"],
-    'L': ["loulou", "bouillon", "lila", "ballon", "molle"],
-    'M': ["maman", "mimique", "mamie", "moment", "murmure"],
-    'N': ["nanan", "manana", "ronron", "néné", "nananère"],
-    'O': ["oligo", "mollo", "bourgeois", "bouton", "tonton"],
-    'P': ["papa", "popote", "poupon", "pipeau", "appétit"],
-    'Q': ["croque", "quadruple", "aquatique", "qualité", "quinte"],
-    'R': ["rare", "terre", "arbre", "brouillard", "rire"],
-    'S': ["sissi", "assise", "saison", "ressasser", "cuisse"],
-    'T': ["tata", "totem", "tatouage", "tarte", "montant"],
-    'U': ["tutu", "futul", "perdu", "moule", "lutulage"],
-    'V': ["vavavoom", "vivant", "sauvage", "veau", "velours"],
-    'W': ["woow", "watt", "wowow", "wargame", "wawa"],
-    'X': ["axel", "axiome", "relaxe", "excitation", "sexe"],
-    'Y': ["yaya", "yayou", "yonis", "yeux", "joyeux"],
-    'Z': ["zigzag", "zizi", "zazou", "blizzard", "zaza"]
+    'A': ["salade", "banane", "flamme", "chaise", "carafe"],
+    'B': ["bison", "cabine", "blouse", "abri", "bonbon"],
+    'C': ["cerise", "cactus", "canard", "chaud", "cirque"],
+    'D': ["dinde", "décor", "donner", "douleur", "début"],
+    'E': ["pêche", "mercredi", "effet", "plage", "épice"],
+    'F': ["fête", "fleur", "framboise", "frère", "façade"],
+    'G': ["groupe", "gagne", "galerie", "grille", "guerre"],
+    'H': ["histoire", "hasard", "hiver", "homme", "horizon"],
+    'I': ["étudier", "milieu", "jardin", "écrire", "grill"],
+    'J': ["jaune", "jupe", "joli", "jambe", "jouet"],
+    'K': ["kilo", "kiwi", "kangourou", "kermesse", "kayak"],
+    'L': ["lumière", "lien", "lire", "loup", "lait"],
+    'M': ["mère", "mignon", "monde", "musique", "moulin"],
+    'N': ["nuage", "nuit", "nommer", "neige", "nourrir"],
+    'O': ["orange", "origine", "opéra", "objet", "ombrelle"],
+    'P': ["poire", "piano", "pluie", "parc", "porte"],
+    'Q': ["quatre", "qualité", "quitte", "queue", "quart"],
+    'R': ["rose", "rire", "réveil", "route", "réponse"],
+    'S': ["soleil", "salle", "sourire", "sombre", "salut"],
+    'T': ["table", "téléphone", "tigre", "tarte", "triste"],
+    'U': ["univers", "unité", "unique", "urgent", "utile"],
+    'V': ["vache", "valise", "visage", "vitrine", "vendre"],
+    'W': ["wagon", "weekend", "whisky", "web", "watt"],
+    'X': ["xénon", "exemple", "toxique", "fox", "lux"],
+    'Y': ["yeux", "yo-yo", "yacht", "yeti", "yogourt"],
+    'Z': ["zèbre", "zone", "zinc", "zodiac", "zigzag"]
 }
 
-def generate_helper_phrase(ciphertext):
+def generate_helper_phrase(ciphertext, cipher_map):
     # Generate a helper phrase using the helper words list
     helper_phrase = []
-    for letter in set(ciphertext):
-        if letter.isalpha() and letter.upper() in helper_words:
-            helper_phrase.append(random.choice(helper_words[letter.upper()]))
+    desired_length = 3 * len(ciphertext)  # Desired length for the helper phrase
+
+    while len(" ".join(helper_phrase)) < desired_length:
+        for letter in set(ciphertext):
+            if letter.isalpha() and letter.upper() in helper_words:
+                word = random.choice(helper_words[letter.upper()])
+                # Encrypt the helper word using the same cipher map
+                encrypted_word = generate_ciphertext(word, cipher_map)
+                helper_phrase.append(encrypted_word)
+                if len(" ".join(helper_phrase)) >= desired_length:
+                    break
+
     return " ".join(helper_phrase)
 
 # Streamlit App
@@ -72,8 +83,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"):
-    ciphertext, cipher_map = generate_ciphertext(plaintext)
-    helper_phrase = generate_helper_phrase(ciphertext)
+    cipher_map = generate_cipher_map(plaintext)
+    ciphertext = generate_ciphertext(plaintext, cipher_map)
+    helper_phrase = generate_helper_phrase(ciphertext, cipher_map)
 
     # Display the encrypted text and helper phrase
     st.write("Texte chiffré:")