Update app.py

app.py

@@ -12,75 +12,53 @@ def get_phones(word):
     phones = phones_for_word[0].split()
     return phones
 
-def _get_vowel_index(phones):
-    for index, phone in enumerate(phones):
-        if phone[0] in 'AEIOU':
-            return index
-
-def _get_consonant_indices(phones):
-    indices = []
-    for index, phone in enumerate(phones):
-        if phone[0] in CONSONANTS and index < len(phones) - 1:
-            indices.append(index)
-    return indices
-
-def _get_rhyming_tail(phones):
-    index = _get_vowel_index(phones)
-    return phones[index:]
-
-def _slant_rhyme_consonants(phones):
-    consonant_indices = _get_consonant_indices(phones)
-    new_phones = copy.copy(phones)
-    for index in consonant_indices:
-        new_phones[index] = '.'
-    return new_phones
-
-def get_perfect_rhymes(phones):
-    rhyming_tail = " ".join(_get_rhyming_tail(phones))
-    return pronouncing.search(rhyming_tail + "$")
-
-def get_internal_rhymes(phones):
-    rhyming_tail = " ".join(_get_rhyming_tail(phones))
-    return pronouncing.search(rhyming_tail)
-
-def get_slant_rhymes(phones):
-    tail = _get_rhyming_tail(phones)
-    search = ' '.join(_slant_rhyme_consonants(tail))
-    return pronouncing.search(search)
-
-def find_rhymes(word):
-    phones = get_phones(word)
-    if phones is None:
-        return 'That word is not recognized. Please check the spelling and avoid proper nouns.'
+def _get_rhyming_tail(phones, syllable_count=2):
+    vowels = [phone for phone in phones if phone[0] in 'AEIOU']
+    if len(vowels) < syllable_count:
+        return None  # Not enough syllables for a multisyllabic rhyme
+    return phones[-syllable_count * 2:]
+
+def get_multisyllabic_rhymes(phones, syllable_count=2):
+    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 + "$")
+
+def find_rhymes_for_words(words, syllable_count=2):
+    # Split words by spaces
+    words = words.split()  
+    results = {}
     
-    perfect_rhymes = get_perfect_rhymes(phones)
-    internal_rhymes = get_internal_rhymes(phones)
-    slant_rhymes = get_slant_rhymes(phones)
-
-    perfects_without_word = set(perfect_rhymes) - {word}
-    internals_without_perfects = set(internal_rhymes) - set(perfect_rhymes) - {word}
-    imperfect_slants = set(slant_rhymes) - set(internal_rhymes) - {word}
-
-    result = ""
-    if perfects_without_word:
-        result += 'Perfect rhymes: ' + ", ".join(sorted(perfects_without_word)) + '\n\n'
-    else:
-        result += 'No perfect rhymes.\n\n'
-
-    if internals_without_perfects:
-        result += 'Internal rhymes: ' + ", ".join(sorted(internals_without_perfects)) + '\n\n'
-    else:
-        result += 'No internal rhymes.\n\n'
-
-    if imperfect_slants:
-        result += 'Slant rhymes: ' + ", ".join(sorted(imperfect_slants))
-    else:
-        result += 'No slant rhymes.'
+    for word in words:
+        phones = get_phones(word)
+        if phones is None:
+            results[word] = 'Not recognized'
+            continue
+        
+        multisyllabic_rhymes = get_multisyllabic_rhymes(phones, syllable_count)
+        results[word] = multisyllabic_rhymes if multisyllabic_rhymes else "No multisyllabic rhymes found"
+    
+    # Format results as a string
+    result_str = ""
+    for word, rhymes in results.items():
+        if isinstance(rhymes, list):
+            result_str += f"{word.capitalize()} ({syllable_count} syllables): " + ", ".join(rhymes) + "\n\n"
+        else:
+            result_str += f"{word.capitalize()}: {rhymes}\n\n"
 
-    return result
+    return result_str
 
 # Gradio interface
-iface = gr.Interface(fn=find_rhymes, inputs="text", outputs="text", description="Enter a word to find rhymes.")
+iface = gr.Interface(
+    fn=lambda words, syllables: find_rhymes_for_words(words, syllables), 
+    inputs=[
+        gr.inputs.Textbox(label="Enter words (space-separated)"), 
+        gr.inputs.Slider(1, 3, step=1, label="Number of Syllables")
+    ], 
+    outputs="text", 
+    description="Enter multiple words separated by spaces to find multisyllabic rhymes."
+)
 
 if __name__ == "__main__":
     iface.launch()