app.py

import os
from flask import request
from flask import jsonify
from flask import Flask, render_template
from flask import flash, redirect, url_for
from werkzeug.utils import secure_filename

from models.defines import CONFIG_DICT
from accident_detection import get_prediction

UPLOAD_FOLDER = 'uploads'
ALLOWED_EXTENSIONS = {'mp4', 'avi'}

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

def allowed_file(filename):
    return '.' in filename and \
           filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS

@app.route('/prediction', methods=['GET'])
def test_prediction():
    probability = 0.44444444
    return render_template('prediction.html', accidentProbability="%.2f" % probability, noAccidentProbability="%.2f" % (1 - probability))

@app.route('/prediction', methods=['POST'])
def form_submitted():
    # Check if the post request has the file part
    if 'file' not in request.files:
        flash('No file part')
        return redirect(request.url)
    file = request.files['file']
    # If the user does not select a file, the browser submits an
    # empty file without a filename.
    if file.filename == '':
        flash('No selected file')
        return redirect(request.url)
    if file and allowed_file(file.filename):
        backbone_name = request.form['backbone_name']
        backbone_name_lower = backbone_name.replace(" ", "").replace("-", "").lower()
        temporal_model = request.form['temporal_model']
        temporal_model_lower = temporal_model.lower()

        trained_models = [("Mobile-Net", "conv3d"), ("Mobile-Net", "convlstm"), 
                            ("Mobile-Net", "biconvlstm"), ("Mobile-Net", "taconv3d"),
                            ("Dense-Net 121", "conv3d"),("Dense-Net 121", "biconvlstm"),
                            ("Inception v3", "convlstm"), ("Inception v3", "conv3d")]
        
        if (backbone_name, temporal_model_lower) not in trained_models:
            return render_template('index.html', no_model=True)
        else:
            filename = secure_filename(file.filename)
            filename = os.path.join(app.config['UPLOAD_FOLDER'], filename)
            file.save(filename)
            start_timestamp = float(request.form['start'])
            end_timestamp = float(request.form['end'])
            probability = get_prediction(filename, start_timestamp, end_timestamp, backbone_name_lower, temporal_model_lower)[0][0]
            return render_template('prediction.html', accidentProbability="%.2f" % probability, noAccidentProbability="%.2f" % (1 - probability), backbone_name=backbone_name, temporal_model=temporal_model)
    return redirect(request.url)

    # probability = 0.44444
    # return render_template('prediction.html', accidentProbability="%.2f" % probability, noAccidentProbability="%.2f" % (1 - probability))

@app.route('/statistics')
def statistics():
    return render_template('statistics.html', backbone_info=CONFIG_DICT["backbones"]["info"])

@app.route('/about_us')
def about_us():
    return render_template('about_us.html')

@app.route('/')
def index():
    backbone_info = CONFIG_DICT["backbones"]["info"]
    backbone_info_name = [info.name for info in backbone_info]
    return render_template('index.html', backbone_info_name=backbone_info_name)

if __name__ == "__main__":
    app.run(debug=True)