Haiku2.py

# Haiku bot - Chris Pocock
# Reads some file of random text and detects sentences with 5 and 7 syllables
# and turns them into Haiku poems

import sys
import re
import textstat
from textstat.textstat import textstat

import random
from twython import Twython
from twython import TwythonStreamer
import json

#Function to count syllables
def CountSyllables(word, isName=True):
    vowels = "aeiouy"
    #single syllables in words like bread and lead, but split in names like Breanne and Adreann
    specials = ["ia","ea"] if isName else ["ia"]
    specials_except_end = ["ie","ya","es","ed"]  #seperate syllables unless ending the word
    currentWord = word.lower()
    numVowels = 0
    lastWasVowel = False
    last_letter = ""

    for letter in currentWord:
        if letter in vowels:
            #don't count diphthongs unless special cases
            combo = last_letter+letter
            if lastWasVowel and combo not in specials and combo not in specials_except_end:
                lastWasVowel = True
            else:
                numVowels += 1
                lastWasVowel = True
        else:
            lastWasVowel = False

        last_letter = letter

    #remove es & ed which are usually silent
    if len(currentWord) > 2 and currentWord[-2:] in specials_except_end:
        numVowels -= 1

    #remove silent single e, but not ee since it counted it before and we should be correct
    elif len(currentWord) > 2 and currentWord[-1:] == "e" and currentWord[-2:] != "ee":
        numVowels -= 1

    return numVowels
#End function    
    
#success = False
#count = 0

#Define our constant variables, this is all the data we wrote down in the first part of the tutorial.
CONSUMER_KEY = 	
CONSUMER_SECRET = 
ACCESS_KEY = 
ACCESS_SECRET = 

#Create a copy of the Twython object with all our keys and secrets to allow easy commands.
api = Twython(CONSUMER_KEY,CONSUMER_SECRET,ACCESS_KEY,ACCESS_SECRET) 


twitter=Twython(
    CONSUMER_KEY,
    CONSUMER_SECRET,
    ACCESS_KEY,
    ACCESS_SECRET
    )


success = False

while success == False:

	#optionally generate random word
	word = random.choice(open('20k.txt').readlines())[:-1]
	#word = 'the'

	print 'Attempting to construct Haiku using: ', word, '\n'
	
	
	# Find first line - 5 syllables
	
	results = twitter.search(q = word, count = 100)


	tweets = results['statuses']
	
	count = 0
	
	for tweet in tweets:
		tw = tweet['text']
		tw_clean = re.sub(r"http\S+", "", tw)
		if CountSyllables(tw_clean) == 5:
			print(tw_clean)
			success = True
		#break
		
		else:
			count += 1
			if count > 100:
				break
		
	# Find second line - 7 syllables
	
	results = twitter.search(q = word, count = 100)
	
	count = 0
	
	for tweet in tweets:
		tw = tweet['text']
		tw_clean = re.sub(r"http\S+", "", tw)
		if CountSyllables(tw_clean) == 7:
			print(tw_clean)
			success = True
		else:
			count += 1
			if count > 100:
				break

	# Find third line - 5 syllables
	
	results = twitter.search(q = word, count = 100)
	
	count = 0

	for tweet in tweets:
		tw = tweet['text']
		tw_clean = re.sub(r"http\S+", "", tw)
		if CountSyllables(tw_clean) == 5:
			print(tw_clean)
			success = True
			
		else:
			count += 1
			if count > 100:
				break

#Using our newly created object, utilize the update_status to send in the text passed in through CMD
#api.update_status(status=s1)