Update app.py

app.py

@@ -3,12 +3,11 @@ 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 wordlist_generator import generate_wordlist  # Assuming your generate_wordlist function is imported here
 from dotenv import load_dotenv
 import os
 
-# Load environment variables from .env file
-load_dotenv()
+load_dotenv()  # Load environment variables from .env file
 access_token = os.getenv("HUGGINGFACE_ACCESS_TOKEN")
 
 # Page configuration
@@ -42,21 +41,18 @@ include_numbers = st.sidebar.checkbox("Include Numbers", value=True)
 # Display wordlist generation results
 st.header("Generated Wordlist Preview")
 
-# Call to a mock-up function for wordlist generation (you will replace this with your actual logic)
-try:
-    wordlist = generate_wordlist(
-        size=wordlist_size, 
-        min_length=min_length, 
-        max_length=max_length, 
-        special_chars=include_special_chars, 
-        numbers=include_numbers
-    )
-except Exception as e:
-    st.error(f"Error generating wordlist: {e}")
+# Call to the actual logic for wordlist generation
+wordlist = generate_wordlist(
+    size=wordlist_size, 
+    min_length=min_length, 
+    max_length=max_length, 
+    special_chars=include_special_chars, 
+    numbers=include_numbers
+)
 
 # Display the first 20 items in the wordlist
 st.write(f"Preview of {wordlist_size} words:")
-st.dataframe(pd.DataFrame(wordlist[:20], columns=["Generated Words"]))  # Display as a table for better interaction
+st.write(wordlist[:20])  # Show the first 20 words for brevity
 
 # Download link for the full wordlist
 st.markdown("### Download Full Wordlist")
@@ -103,6 +99,11 @@ if entropy < 50:
 else:
     st.success("Good entropy! This wordlist is secure against most brute-force attempts.")
 
+# Additional Analysis (Average Word Length)
+st.header("Additional Analysis: Average Word Length")
+avg_word_length = np.mean(word_lengths)
+st.write(f"Average Word Length: {avg_word_length:.2f} characters")
+
 # Footer
 st.markdown("---")
 st.markdown(