Update app.py

app.py

@@ -7,7 +7,8 @@ from wordlist_generator import generate_wordlist  # A mock-up function for your
 from dotenv import load_dotenv
 import os
 
-load_dotenv()  # Load environment variables from .env file
+# Load environment variables from .env file
+load_dotenv()
 access_token = os.getenv("HUGGINGFACE_ACCESS_TOKEN")
 
 # Page configuration
@@ -42,17 +43,20 @@ include_numbers = st.sidebar.checkbox("Include Numbers", value=True)
 st.header("Generated Wordlist Preview")
 
 # Call to a mock-up function for wordlist generation (you will replace this with your actual logic)
-wordlist = generate_wordlist(
-    size=wordlist_size, 
-    min_length=min_length, 
-    max_length=max_length, 
-    special_chars=include_special_chars, 
-    numbers=include_numbers
-)
+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}")
 
 # Display the first 20 items in the wordlist
 st.write(f"Preview of {wordlist_size} words:")
-st.write(wordlist[:20])  # Show the first 20 words for brevity
+st.dataframe(pd.DataFrame(wordlist[:20], columns=["Generated Words"]))  # Display as a table for better interaction
 
 # Download link for the full wordlist
 st.markdown("### Download Full Wordlist")
@@ -104,6 +108,3 @@ st.markdown("---")
 st.markdown(
     "Made with ❤️ by Canstralian. For more information on ReconNinja, visit our [GitHub](https://github.com/Canstralian)."
 )
-
-
-