Update app.py

app.py

@@ -48,15 +48,16 @@ helper_words = {
     'Z': ["zèbre", "zone", "zinc", "zodiac", "zigzag"]
 }
 
-def generate_helper_phrase(ciphertext, mapping):
+def generate_helper_phrase(plaintext, mapping):
+    unique_letters = set(char for char in plaintext if char.isalpha())  # Get unique letters from plaintext
     helper_phrase = []
 
-    # Generate three words for each letter in the ciphertext
-    for letter in set(ciphertext):
-        if letter.isalpha() and letter.upper() in helper_words:
-            for _ in range(3):  # Add three words for each letter
-                word = random.choice(helper_words[letter.upper()])
-                helper_phrase.append(word)
+    # Generate five words for each unique letter in the plaintext
+    for letter in unique_letters:
+        letter_upper = letter.upper()
+        if letter_upper in helper_words:
+            selected_words = random.sample(helper_words[letter_upper], 5)  # Select 5 random words
+            helper_phrase.extend(selected_words)
 
     # Join the helper phrase into a single string
     helper_phrase_str = " ".join(helper_phrase)
@@ -76,7 +77,7 @@ plaintext = st.text_input("Entrez le texte à chiffrer:", "Exemple de texte")
 if st.button("Chiffrer"):
     mapping = generate_variable_mapping()
     ciphertext = encrypt_text(plaintext, mapping)
-    helper_phrase = generate_helper_phrase(ciphertext, mapping)
+    helper_phrase = generate_helper_phrase(plaintext, mapping)
 
     # Display the encrypted text and helper phrase
     st.write("Texte chiffré:")