In [3]:

    

import json
import urllib.request
import random
import PIL
import word2vec
import sys
import memegenerator
import scipy.ndimage
import pickle


    

In [2]:

    

model = word2vec.load('text8.bin')


    

In [16]:

    

public_tweets = pickle.load(open('tweets.pkl', 'rb'))


    

In [17]:

    

valid_tweets = []
best_tweets = []
for t in public_tweets:
    words = t.text.split()
    words = [w for w in words if w in model.vocab]
    words2 = [w for w in words if len(w) > 3]
    if len(words) < 3:
        continue
        
    if len(words2) >= 3:
        best_tweets.append(t)
        
    valid_tweets.append(t)

tweet = random.choice(public_tweets)
if best_tweets:
    tweet = random.choice(best_tweets)


    

In [18]:

    

words = tweet.text.split()
words2 = [w for w in words if len(w) > 3]
words2 = [w for w in words2 if w in model.vocab]

ws = []
while(len(ws) <= 2):
    
    using = words
    if words2:
        using = words2
    
    w = random.choice(using)
    using.pop(using.index(w))
    if w in model.vocab:
        ws.append(w)


    

In [19]:

    

indexes, metrics = model.analogy(pos=[ws[0], ws[1]], neg=[ws[2]], n=10)


    

In [20]:

    

ws.append(model.vocab[indexes][0])


    

In [21]:

    

if using:
    ws.append(using[0])
else:
    ws.append(ws[0])


    

In [22]:

    

print(tweet.text)
print(ws)


    

    




"If you think women in tech is just a pipeline problem, you haven’t been paying attention" https://t.co/VACr7JdaAb by brilliant @math_rachel
['pipeline', 'just', 'think', 'bus', 'women']

In [23]:

    

url_template = 'https://ajax.googleapis.com/ajax/services/search/images?v=1.0&q=%s&imgsz=xlarge&imgtype=photo&as_sitesearch=flickr.com'
url = url_template % ws[4]
result = urllib.request.urlopen(url)
js = json.loads(result.read().decode())
image_url = js['responseData']['results'][0]['unescapedUrl']

filename = urllib.request.urlretrieve(image_url)[0]
topString = '%s is to %s' % (ws[0], ws[1])
bottomString = "as %s is to %s" % (ws[2], ws[3])
memegenerator.make_meme(topString, bottomString, filename)


    

In [24]:

    

imagename = 'temp.png'
%matplotlib inline


    

In [25]:

    

from PIL import Image
im = Image.open(imagename)
im.show()


    

In [ ]:

    




    

In [ ]: