#!/usr/bin/env python

import sys, json
sys.path.append('/home/spam-filter/')

from threading import Thread
from tweets.tweetFilter import tweetFilter

from utils.jsonLogger import setup_logging, log

from flask import Flask, request

from probes.probes import runFlaskProbes

app = Flask(__name__)

filter = tweetFilter()

@app.route('/predict', methods=['GET'])
def tweetPredict():
    tweet = request.args.get('tweet')

    log("Receiving Tweet to classify [{}]".format(tweet), 'INFO')

    result = filter.tweetFilterPredit(tweet)

    log("Returning classification result", 'INFO')
    return json.dumps({'result': result, 'tweet': tweet}), 200, {'ContentType':'application/json'}

@app.route('/predictProbeTest', methods=['GET'])
def predictProbeTest():
    return json.dumps({'result': 'spam', 'tweet': 'Fake Test Tweet'}), 200, {'ContentType':'application/json'}

def callTweetFilter():
    filter.tweetFilterTrain()

    app.run(port=9090, host="0.0.0.0")

# def callNewsFilter():
    # newsFilter()

def callProbes():
    runFlaskProbes()

if __name__ == '__main__':
    setup_logging()

    log("Starting Spam Filter...", 'INFO')
    sys.stdout.flush()

    Thread(target=callProbes).start()

    Thread(target=callTweetFilter).start()
    # Thread(target=callNewsFilter).start()