Update app.py

app.py

@@ -2,17 +2,39 @@ import streamlit as st
 from random import choice, sample
 import pronouncing
 import emoji
+import nltk
+from itertools import cycle
+from nltk.corpus import wordnet
+
+# Ensure NLTK resources are downloaded
+nltk.download('wordnet')
+nltk.download('averaged_perceptron_tagger')
 
 # Function to add emojis to a line
 def add_emojis(line, emojis):
     return ' '.join([emoji.emojize(f":{e}:") + word for word, e in zip(line.split(), cycle(emojis))])
 
+# Function to ensure the rhyming words fit grammatically
+def filter_rhymes(rhymes, pos_tag):
+    filtered_rhymes = []
+    for word in rhymes:
+        word_pos = nltk.pos_tag([word])[0][1]
+        if word_pos in pos_tag:
+            filtered_rhymes.append(word)
+    return filtered_rhymes
+
 # Function to generate a line of the tongue twister
 def generate_line(keyword, humor_words, complexity, emojis):
     rhymes = pronouncing.rhymes(keyword)
     if not rhymes:
         return "No rhymes found."
-    rhyme_words = sample(rhymes, min(len(rhymes), complexity))
+    
+    # Identify part-of-speech for keyword
+    keyword_pos = nltk.pos_tag([keyword])[0][1]
+    pos_tags = [wordnet.synsets(word)[0].pos() for word in rhymes if wordnet.synsets(word)]
+    filtered_rhymes = filter_rhymes(rhymes, pos_tags)
+    
+    rhyme_words = sample(filtered_rhymes, min(len(filtered_rhymes), complexity))
     line = f"{keyword.capitalize()} {choice(humor_words)} {' '.join(rhyme_words)},"
     return add_emojis(line, emojis)