Wordblender.py

"""
Blends words to make Pokémon names
"""
import operator

import nltk.tokenize.sonority_sequencing as sequencing
import numpy as np
from ConceptNet import conceptnet_request
import string
import random


def select_answers(answers):
    """
    Selects the two random words from the answers and returns them.

    :param answers: list of given answers
    :return: two longest answers without the first two
    """
    word1 = random.choice(answers)
    answers.remove(word1)
    word2 = random.choice(answers)
    return word1, word2


def blend_a_word(answers):
    """
    Blends two words together in different variations.

    :param answers: list of given answers
    :return: list of possible blended words
    """
    word1, word2 = select_answers(answers)
    tok = sequencing.SyllableTokenizer()
    syl1 = tok.tokenize(word1)
    syl2 = tok.tokenize(word2)
    words = []
    if len(syl1) == len(syl2) == 1:
        endings = ['saur', 'bat', 'puff', 'don', 'gon', 'low', 'pede', 'no', 'ta']
        syl1 = syl1[0]
        syl2 = syl2[0]
        words.append(syl_to_vowel(syl2) + syl_from_vowel(syl1) + np.random.choice(endings))
        words.append(syl_to_vowel(syl1) + syl_from_vowel(syl2) + np.random.choice(endings))
    else:
        longer_word = syl1 if len(syl1) > len(syl2) else syl2
        shorter_word = syl2 if len(syl1) > len(syl2) else syl1
        for i in range(1, len(longer_word)):
            for j in range(1, len(shorter_word)+1):
                words.append(''.join(shorter_word[:j]) + ''.join(longer_word[i:]).lower())
        for i in range(0, len(shorter_word)):
            for j in range(1, len(longer_word)):
                words.append(''.join(longer_word[:j]) + ''.join(shorter_word[i:]).lower())
    return words, word1, word2


def syl_from_vowel(syl):
    """
    Finds the position of the first vowel for a syllable.
    
    :param syl: input syllable
    :return: syllable starting from the first vowel
    """
    vowels = ['a','e','i','u','o','y']
    i = 0
    while syl[i] not in vowels:
        if (len(syl)-1) <= i:
            if i == (len(syl)-1):
                return str(syl[i+1:] + np.random.choice(vowels))
            break
        i += 1
    return str(syl[i:])


def syl_to_vowel(syl):
    """
    Finds the position of the first vowel for a syllable.

    :param syl: input syllable
    :return: syllable ending at the first vowel
    """
    vowels = ['a', 'e', 'i', 'u', 'o', 'y']
    consonants = list(string.ascii_lowercase)
    [consonants.remove(v) for v in vowels]
    i = 0
    while syl[i] not in vowels:
        i += 1
        if (len(syl)) <= i:
            break
    if i == 0:
        return str(np.random.choice(consonants))
    return str(syl[:i])


def synonyms(word):
    """
    Returns a list of synonyms. NOT USED, might be interesting for future use.

    :param word: a word
    :return: list of synonyms
    """
    texts, synonyms = conceptnet_request(word, 'RelatedTo')
    return synonyms