spam-filter/src/main.py
2020-03-06 20:37:03 +00:00

44 lines
1023 B
Python

#!/usr/bin/env python
import sys, json
sys.path.append('/home/spam-filter/')
from threading import Thread
from tweets.tweetFilter import tweetFilter
# from tweets.tweetFilter import tweetFilter
# from news.newsFilter import newsFilter
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')
result = filter.tweetFilterPredit(tweet)
return json.dumps({'result': result, 'tweet': 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__':
print("Console: ", "==== Spam Filter - Tweets & News ====")
sys.stdout.flush()
Thread(target=callProbes).start()
Thread(target=callTweetFilter).start()
# Thread(target=callNewsFilter).start()