graffittiBot

#!/usr/bin/env python2
# -*- coding: utf-8 -*- #

from sys import argv
from PIL import Image
import cv2
import StringIO

from newsGraffiti_1 import comic_element
from twitterbot import TwitterBot
from picbot import PicBot
import keys



class graffitiBot2(PicBot, TwitterBot):
    """
    Whenever you tweet a photo at this bot, it will tweet back at you with an image
    with the faces flipped
    """
    def bot_init(self):
        """ Initialize and configure the bot """

        ############################
        # SETUP  FACE DETECTION    #
        ############################
        classifier = "haarcascade_frontalface_default.xml"
        self.faceCascade = cv2.CascadeClassifier(classifier)

        ############################
        # REQUIRED: LOGIN DETAILS! #
        ############################
        self.config['api_key'] = keys.consumer_key
        self.config['api_secret'] = keys.consumer_secret
        self.config['access_key'] = keys.access_token
        self.config['access_secret'] = keys.access_token_secret


        ######################################
        # SEMI-OPTIONAL: OTHER CONFIG STUFF! #
        ######################################

        # how often to tweet, in seconds
        self.config['tweet_interval'] = 1 * 60     # default: 1 minutes

        # use this to define a (min, max) random range of how often to tweet
        # e.g., self.config['tweet_interval_range'] = (5*60, 10*60) # tweets every 5-10 minutes
        self.config['tweet_interval_range'] = None

        # only reply to tweets that specifically mention the bot
        self.config['reply_direct_mention_only'] = True

        # the bot will listen to everything on the timeline
        self.config['ignore_timeline_mentions'] = False

        # only include bot followers (and original tweeter) in @-replies
        self.config['reply_followers_only'] = False

        # fav any tweets that mention this bot?
        self.config['autofav_mentions'] = False

        # fav any tweets containing these keywords?
        self.config['autofav_keywords'] = []

        # follow back all followers?
        self.config['autofollow'] = False



    def on_scheduled_tweet(self):
        """ Make a public tweet to the bot's own timeline. """
        # We might take senteces from somewhere and tweet them on a regular basis ...
        pass # don't do anything here ...


    def on_timeline(self, tweet, prefix):
        """ Actions to take on a timeline tweet. """

        #pass # Don't do anything here ...
        print "I am watching the timeline"

        #watch timeline and grab images form martins bot
        image = self.get_image(tweet)

        if image is not None:
            print "1"

            text = "{} This is this what you really think ?.".format(prefix)

            image, rectangles = comic_element(image)

            num = len(rectangles)
            if num == 1:
                comment = "Now I know why did clicked this picture".format(prefix)
            elif num > 1:
                comment = "{} You look like a character from a comic strip  ".format(prefix)
            else:
                print "2"
                return
            print comment

            # filename and format for uplaoding
            filename, format = "result.jpg", "JPEG"

            image.show()

            # write image to a StringIO file as JPG
            file = StringIO.StringIO()
            image.save(file, format= format)

            self.post_tweet(text[:140], reply_to=tweet, media=image, file=file)


    def on_mention_without_image(self, tweet, prefix):
        """
        React to text-only mentions
        """
        text = "{} Come on, give me a picture and I'll tell you what I think about it".format(prefix)
        try:
            self.post_tweet(text[:140], reply_to=tweet)
        except Exception as e:
            print(e)



    def process_image(self, image, prefix=None, fn=comic_element):
        """
        Process the image, place the speech bubble
        """

        image, rectangles = fn(image)

        # return new images with comment
        num = len(rectangles)
        if num == 1:
            comment = "{} Cool pic :) Do you have one with more faces in it?".format(prefix)
        elif num > 1:
            comment = "{} your photo says something more! ".format(prefix)
        else:
            comment = "Nice pic. But I couldn't spot any faces :("
        return image, comment


if __name__ == '__main__':

    bot = graffitiBot2()

    if len(argv) == 1:
        # just run the bot
        print("Running the bot.")
        bot.run()
        print bot.state['recent_timeline']

    elif argv[1] == "test":
        # various tests
        if(len(argv) == 2):
            print "USAGE: graffiti Bot.py test [image|mention]"
        else:
            test = argv[2]
            if test == "image":
                print("Testing image ")
                img = Image.open("test2.jpg")
                img, comment = bot.process_image(img, "@test", fn=comic_element )
                img.show()
                print("COMMENT: {}".format(comment))
            elif test == "mention":
                print("Testing on_mention hook of the bot")
                api = bot.api
                tweet = api.get_status("655719153102401537")
                prefix = bot.get_mention_prefix(tweet)
                bot.on_mention(tweet, prefix)
            else:
                print("Unknown test: {}".format(test))