views.py

import gc
import os.path

from flask import Flask, render_template, redirect, url_for, request, session, jsonify

from helpers.decorators import login_required
from helpers.funcs import *
from models import user
from models.search import Search
from models.similarity import most_similar, similar_by_word, doesnt_match


app = Flask(__name__)
app.config['UPLOAD_FOLDER'] = config.UPLOAD_FOLDER
app.secret_key = config.SECRET_KEY


# INDEX PAGE
@app.route('/')
def index():
    return render_template('index.html')


# UPLOAD DECORATOR
@app.route('/upload/', methods=['POST'])
def upload():
    try:
        image = None
        if request.files['img-file'].filename:
            image = request.files['img-file']
        elif request.files['img-camera'].filename:
            image = request.files['img-camera']
        elif request.form['img-url']:
            image = request.form['img-url']
        if image:
            if upload_file(image, image.filename, app.config['UPLOAD_FOLDER']):
                flash('Upload success')
                return redirect(url_for('search', img=image.filename))
            else:
                flash('Could not upload {}'.format(image.filename))
        else:
            flash("Give us an image to work with.")
    except Exception as e:
        flash('ERROR: {}'.format(e))
    return redirect(url_for('index'))


# SEARCH PAGE
@app.route('/search/')
@app.route('/search/<img>')
@app.route('/search/<img>/<int:page>/')
def search(img=None, page=1):
    results = None
    if not img:
        return render_template('search.html', img=img, results=results)
    try:
        # image_filename = os.path.join(config.UPLOAD_FOLDER, img)
        # r = Recognizer()
        # image_label = r.recognize(image_filename)     # Returns a string of image label
        image_label = 'technology'
        _search = Search()
        results = _search.search(image_label, page)
    except Exception as e:
        flash('ERROR: {}'.format(e))
    return render_template('search.html', img=img, page=page, results=results)


# REGISTER FUNCTION
@app.route('/register/', methods=['POST'])
def register():
    status = user.register(request.form['username'], request.form['password'],
                           request.form['confirm-password'])
    if status == user.SUCCESS:
        flash('Registration Successful!')
        return redirect(url_for('login'))
    return render_template('login.html', register_error=status)


# LOG IN PAGE
@app.route('/login/', methods=['GET', 'POST'])
def login():
    if request.method == 'POST':
        status = user.login(request.form['username'], request.form['password'])
        if status == user.SUCCESS:
            flash('Login Successful!')
            return redirect(url_for('index'))
        return render_template('login.html', login_error=status)
    return render_template('login.html')


# LOGOUT PAGE
@app.route('/logout/')
@login_required
def logout():
    session.clear()
    gc.collect()
    return redirect(url_for('login'))


# SIMILARITY PAGE
@app.route('/similarity/', methods=['GET', 'POST'])
@app.route('/similarity', methods=['GET', 'POST'])
def similarity():
    if request.method == 'POST':
        if request.args.get('action') == 'this-to-that':
            word1 = request.form['word-1']
            word2 = request.form['word-2']
            word3 = request.form['word-3']
            answer = most_similar(word1, word2, word3)
            flash(answer)
        elif request.args.get('action') == 'most-similar':
            word = request.form['word']
            answer = similar_by_word(word)
            flash(answer)
        elif request.args.get('action') == 'phrase-fitness':
            phrase = request.form['phrase']
            answer = doesnt_match(phrase)
            flash(answer)
    return render_template('similarity.html')


# AJAX function. (Similarity Page)
@app.route("/_similarity")
def _similarity():
    result = None
    action = request.args.get("action")
    if action == "most_similar":
        word1 = request.args.get("word1")
        word2 = request.args.get("word2")
        word3 = request.args.get("word3")
        result = most_similar(word1, word2, word3)[0][0]
    elif action == "similarity_by_word":
        word = request.args.get("word")
        result = similar_by_word(word)[0][0]
    elif action == "doesnt_match":
        phrase = request.args.get("phrase")
        result = doesnt_match(phrase)
    return jsonify(result=result)


# SETTINGS PAGE
@app.route('/settings/')
def settings():
    return render_template('settings.html')


# HELP PAGE
@app.route('/help/')
def help_page():
    return render_template('help.html')


# ERROR HANDLER (404)
@app.errorhandler(404)
def page_not_found(e):
    make_dir(config.ERROR_DIR)
    filename = os.path.join(config.ERROR_DIR, config.FOUR_OH_FOUR_ERR_FILE)
    create_file(filename)
    errmsg = '{} – {}\n\t{}\n'.format(request.remote_addr, request.path, e)
    append_to_file(filename, errmsg)
    return render_template('error/404.html', error=e)