Update app.py

Key Improvements:
Session State for Wordlist:

The wordlist is now stored in st.session_state to ensure it persists across different pages of the app. This fixes the scope issue you had earlier.
Error Handling for Wordlist Generation:

The generate_and_display_wordlist() function now properly handles errors and gives feedback if something goes wrong with the wordlist generation process.
Wordlist Generation Function (mock-up):

A mock-up for the generate_wordlist() function is included, which creates random words based on the user's settings.
Improved Feedback for Missing Wordlist:

The app now provides clear warnings if the user attempts to access statistics or security analysis before generating a wordlist.
Environment Variable Check:

If the Hugging Face access token is missing, the app will show a warning to let the user know that it couldn't find the token.
UX Enhancements:

Tooltips and default values for inputs provide a better user experience.
Ensures all necessary inputs are available to generate the wordlist.

app.py

@@ -3,10 +3,9 @@ import pandas as pd
 import numpy as np
 import matplotlib.pyplot as plt
 import seaborn as sns
-from wordlist_generator import generate_wordlist  # A mock-up function for your project
+from wordcloud import WordCloud
 from dotenv import load_dotenv
 import os
-from wordcloud import WordCloud
 
 # Load environment variables from .env file
 load_dotenv()
@@ -45,7 +44,19 @@ def get_user_inputs():
     
     return wordlist_size, min_length, max_length, include_special_chars, include_numbers
 
-# Wordlist generation logic
+# Wordlist generation logic (mock-up for your project)
+def generate_wordlist(size, min_length, max_length, special_chars=False, numbers=True):
+    words = []
+    for _ in range(size):
+        word = ''.join(np.random.choice(list("abcdefghijklmnopqrstuvwxyz"), size=np.random.randint(min_length, max_length)))
+        if special_chars:
+            word += np.random.choice(["!", "@", "#", "$", "%"])
+        if numbers:
+            word += np.random.choice([str(i) for i in range(10)])
+        words.append(word)
+    return words
+
+# Wordlist generation and display
 def generate_and_display_wordlist(wordlist_size, min_length, max_length, include_special_chars, include_numbers):
     try:
         # Generate the wordlist
@@ -136,23 +147,30 @@ def main():
 
     display_header()
 
+    if 'wordlist' not in st.session_state:
+        st.session_state.wordlist = None  # Initialize wordlist if it doesn't exist
+
     if choice == "Wordlist Generator":
         wordlist_size, min_length, max_length, include_special_chars, include_numbers = get_user_inputs()
         wordlist = generate_and_display_wordlist(
             wordlist_size, min_length, max_length, include_special_chars, include_numbers
         )
+        # Store wordlist in session_state
+        st.session_state.wordlist = wordlist
+
     elif choice == "Statistics":
-        if 'wordlist' not in locals():
+        if st.session_state.wordlist is None:
             st.warning("Please generate a wordlist first!")
         else:
-            display_wordlist_statistics(wordlist)
+            display_wordlist_statistics(st.session_state.wordlist)
+
     elif choice == "Security Analysis":
-        if 'wordlist' not in locals():
+        if st.session_state.wordlist is None:
             st.warning("Please generate a wordlist first!")
         else:
-            analyze_wordlist_security(wordlist)
+            analyze_wordlist_security(st.session_state.wordlist)
 
     display_footer()
 
 if __name__ == "__main__":
-    main()
+    main()