Update app.py

app.py

@@ -1,20 +1,43 @@
 import streamlit as st
 import itertools
-from nltk.corpus import wordnet
+from nltk.corpus import wordnet, words
 
-def get_synonyms(word):
+# Download the necessary resources
+nltk.download("wordnet")
+nltk.download("words")
+
+def get_related_words(word):
     synonyms = set()
+    antonyms = set()
+    hypernyms = set()
+    hyponyms = set()
+
     for syn in wordnet.synsets(word):
         for lemma in syn.lemmas():
             synonyms.add(lemma.name())
-    return list(synonyms)
+            if lemma.antonyms():
+                antonyms.add(lemma.antonyms()[0].name())
+
+        for hyper in syn.hypernyms():
+            hypernyms.update(lemma.name() for lemma in hyper.lemmas())
+
+        for hypo in syn.hyponyms():
+            hyponyms.update(lemma.name() for lemma in hypo.lemmas())
+
+    return {
+        "synonyms": list(synonyms),
+        "antonyms": list(antonyms),
+        "hypernyms": list(hypernyms),
+        "hyponyms": list(hyponyms),
+    }
 
 def generate_words(letters, length=None):
+    english_words = set(words.words())
     permutations = set()
     for i in range(1, len(letters) + 1):
         for p in itertools.permutations(letters, i):
             word = "".join(p)
-            if length is None or len(word) == length:
+            if (length is None or len(word) == length) and word in english_words:
                 permutations.add(word)
     return permutations
 
@@ -28,8 +51,15 @@ if letters:
     words = generate_words(letters, length=word_length if word_length > 0 else None)
     st.write(words)
 
-    st.header("Thesaurus Lookup")
-    selected_word = st.selectbox("Select a word to look up synonyms:", [""] + sorted(words))
+    st.header("Thesaurus and Related Words Lookup")
+    selected_word = st.selectbox("Select a word to look up related words:", [""] + sorted(words))
     if selected_word:
-        synonyms = get_synonyms(selected_word)
-        st.write(synonyms)
+        related_words = get_related_words(selected_word)
+        st.subheader("Synonyms")
+        st.write(related_words["synonyms"])
+        st.subheader("Antonyms")
+        st.write(related_words["antonyms"])
+        st.subheader("Hypernyms (more general terms)")
+        st.write(related_words["hypernyms"])
+        st.subheader("Hyponyms (more specific terms)")
+        st.write(related_words["hyponyms"])