Update app.py

app.py

@@ -18,50 +18,55 @@ def _get_rhyming_tail(phones, syllable_count=1):
         return None  # Not enough syllables for the rhyme
     return phones[-syllable_count * 2:]
 
-def get_multisyllabic_rhymes(phones, syllable_count=1):
+def get_exact_rhymes(phones, syllable_count=1):
     rhyming_tail = _get_rhyming_tail(phones, syllable_count)
     if not rhyming_tail:
         return []
     
     rhyming_tail_str = " ".join(rhyming_tail)
-    return pronouncing.search(rhyming_tail_str + "$")
+    matches = pronouncing.search(rhyming_tail_str + "$")
+    
+    exact_rhymes = [match for match in matches if rhyming_tail == _get_rhyming_tail(get_phones(match), syllable_count)]
+    return exact_rhymes
 
-def get_soft_rhymes(phones, syllable_count=1):
+def get_loose_rhymes(phones, syllable_count=1):
+    """
+    Fallback function to find words with a similar ending sound by allowing slight variations.
+    """
     rhyming_tail = _get_rhyming_tail(phones, syllable_count)
     if not rhyming_tail:
         return []
     
-    # Generate a "soft rhyme" by allowing slight variations in the ending sounds
-    search_pattern = " ".join(phone[:-1] + "." for phone in rhyming_tail)  # Replace last digit in each phoneme with "."
-    return pronouncing.search(search_pattern)
+    # Allow for near matches by replacing the last character in each phoneme with a wildcard
+    search_pattern = " ".join(phone[:-1] + "." for phone in rhyming_tail)
+    matches = pronouncing.search(search_pattern)
+    return matches
 
 def find_rhymes_for_phrase(phrase, syllable_count=1):
-    # Split phrase into individual words
     words = phrase.split()
     rhyming_options = []
 
-    # Get rhymes for each word in the phrase
     for word in words:
         phones = get_phones(word)
         if phones is None:
             rhyming_options.append([f"{word} (Not recognized)"])
             continue
         
-        # Get perfect and soft rhymes
-        perfect_rhymes = get_multisyllabic_rhymes(phones, syllable_count)
-        soft_rhymes = get_soft_rhymes(phones, syllable_count)
+        # Try to find exact rhymes first
+        exact_rhymes = get_exact_rhymes(phones, syllable_count)
         
-        # Combine both types of rhymes and add to options
-        all_rhymes = set(perfect_rhymes + soft_rhymes)
-        if not all_rhymes:
-            rhyming_options.append([f"{word} (No rhymes found)"])
+        # If no exact rhymes are found, fallback to looser rhymes
+        if exact_rhymes:
+            rhyming_options.append(exact_rhymes)
         else:
-            rhyming_options.append(list(all_rhymes))
+            loose_rhymes = get_loose_rhymes(phones, syllable_count)
+            if loose_rhymes:
+                rhyming_options.append(loose_rhymes)
+            else:
+                rhyming_options.append([f"{word} (No rhymes found)"])
     
-    # Generate all possible combinations of rhyming words
     combined_results = list(itertools.product(*rhyming_options))
     
-    # Format combined results as rhyming lines
     if combined_results:
         result_str = f"Rhyming combinations for '{phrase}':\n"
         result_str += "\n".join(" ".join(combination) for combination in combined_results)