app.py

import os
import re

import pandas as pd
from flask import abort, flash, Flask, jsonify, redirect, render_template, request, url_for
from flask_login import current_user, login_required, login_user, LoginManager, logout_user
from werkzeug.security import check_password_hash

from config import Config
from extensions import db
from forms import LoginForm, RegistrationForm, UploadForm
from models import User, UserSettings, Video, VideoSettings
from video_processing import process_video

app = Flask(__name__)
app.config.from_object(Config)
db.init_app(app)
with app.app_context():
    db.create_all()

login_manager = LoginManager()
login_manager.init_app(app)
login_manager.login_view = "login"


@login_manager.user_loader
def load_user(user_id):
    return User.query.get(int(user_id))


@app.route("/")
def home():
    return render_template("home.html")


@app.route("/register", methods=["GET", "POST"])
def register():
    form = RegistrationForm()
    if form.validate_on_submit():
        user = User(username=form.username.data, email=form.email.data)
        user.set_password(form.password.data)
        db.session.add(user)
        db.session.commit()

        # Create a UserSettings entry for the new user with default brightness and contrast values
        user_settings = UserSettings(brightness=1.0,
                                     contrast=1.0,
                                     saturate=1.0,
                                     hueRotate=0.0,
                                     user_id=user.id)
        db.session.add(user_settings)
        db.session.commit()

        flash("Account created successfully!", "success")
        return redirect(url_for("login"))
    return render_template("register.html", form=form)


@app.route('/login', methods=['GET', 'POST'])
def login():
    form = LoginForm()
    if form.validate_on_submit():
        user = User.query.filter_by(email=form.email.data).first()
        if user and check_password_hash(user.password_hash, form.password.data):
            login_user(user)
            flash('You have successfully logged in.', 'success')
            return redirect(url_for('upload'))
        else:
            flash('Login unsuccessful. Please check your username and password.', 'danger')
    return render_template('login.html', title='Login', form=form)


@app.route('/upload', methods=['GET', 'POST'])
@login_required
def upload():
    form = UploadForm(request.form)
    if request.method == 'POST' and request.form.get('submit_button') == 'Upload':
        video = request.files['video']
        if video.filename == '':
            return render_template('upload.html', form=form)

        # Get the user's settings or use default values
        user_settings = current_user.settings
        brightness = user_settings.brightness
        contrast = user_settings.contrast
        saturate = user_settings.saturate
        hueRotate = user_settings.hueRotate

        paths = process_video(video)  # Modify this line
        video_entry = Video(name=video.filename,
                            video_filename=os.path.basename(paths["video_path"]),
                            preview_filename=os.path.basename(paths["preview_path"]),
                            blur_timeline_filename=os.path.basename(paths["blur_timeline_path"]),
                            epilepsy_timeline_filename=os.path.basename(paths["epilepsy_timeline_path"]),
                            user_id=current_user.id)
        db.session.add(video_entry)
        db.session.commit()

        # Create VideoSettings entry with custom brightness and contrast values (if any)
        video_settings = VideoSettings(brightness=brightness,
                                       contrast=contrast,
                                       saturate=saturate,
                                       hueRotate=hueRotate,
                                       video_id=video_entry.id)
        db.session.add(video_settings)
        db.session.commit()

        flash('Video uploaded and processed successfully!', 'success')
        return redirect(url_for('player', video_id=video_entry.id))

    return render_template('upload.html', form=form)


@app.route('/edit_video_name/<int:video_id>', methods=['POST'])
def edit_video_name(video_id):
    new_video_name = request.form.get('new_video_name')
    video = Video.query.get_or_404(video_id)
    video.name = new_video_name
    db.session.commit()
    flash("Name changed successfully!", "success")
    return redirect(url_for('player', video_id=video.id))


@app.route("/video/<int:video_id>/clip_labels")
@login_required
def get_clip_labels(video_id):
    video = Video.query.get_or_404(video_id)

    # Change the path to your Feather file based on the video_id
    feather_file = os.path.join(Config.BLUR_TIMELINE_FOLDER, video.blur_timeline_filename)
    df = pd.read_feather(feather_file)
    labels = list(df['data'])
    return jsonify(labels)


@app.route("/video/<int:video_id>/epilepsy_labels")
@login_required
def get_epilepsy_labels(video_id):
    video = Video.query.get_or_404(video_id)

    # Change the path to your Feather file based on the video_id
    feather_file = os.path.join(Config.EPILEPSY_TIMELINE_FOLDER, video.epilepsy_timeline_filename)
    df = pd.read_feather(feather_file)
    labels = list(df['data'])
    return jsonify(labels)


@app.route("/profile")
@login_required
def profile():
    user = User.query.get_or_404(current_user.id)
    return render_template('profile.html', user_settings=user.settings)


@app.route("/youtube_player/<string:youtube_id>")
@login_required
def youtube_player(youtube_id):
    return render_template('youtube_player.html',
                           name="Just a Video",
                           youtube_id=youtube_id,
                           user_settings=current_user.settings)


@app.route("/profile_settings_update", methods=['POST'])
@login_required
def profile_settings_update():
    brightness = request.form.get('brightness', type=float)
    contrast = request.form.get('contrast', type=float)
    saturate = request.form.get('saturate', type=float)
    hueRotate = request.form.get('hueRotate', type=float)

    user = User.query.get_or_404(current_user.id)

    user.settings.brightness = brightness
    user.settings.contrast = contrast
    user.settings.saturate = saturate
    user.settings.hueRotate = hueRotate
    db.session.commit()

    return jsonify({"message": "User settings updated successfully"})


@app.route('/library', methods=['GET'])
@login_required
def library():
    videos = Video.query.filter_by(user_id=current_user.id).all()
    return render_template('library.html', title='My Videos', videos=videos)


@app.route("/video/<int:video_id>")
@login_required
def player(video_id):
    video = Video.query.get_or_404(video_id)
    return render_template("player.html", video=video, video_settings=video.video_settings)


@app.route('/delete_video/<int:video_id>', methods=['POST'])
@login_required
def delete_video(video_id):
    video = Video.query.get_or_404(video_id)
    if video.user_id != current_user.id:
        abort(403)
    db.session.delete(video.video_settings)
    db.session.delete(video)
    db.session.commit()
    os.remove(os.path.join(Config.VIDEO_UPLOAD_FOLDER, video.video_filename))
    os.remove(os.path.join(Config.PREVIEW_UPLOAD_FOLDER, video.preview_filename))
    flash('Your video has been deleted.', 'success')
    return redirect(url_for('library'))


@app.route('/logout')
@login_required
def logout():
    logout_user()
    flash('You have successfully logged out.', 'success')
    return redirect(url_for('home'))


@app.route('/update_video_settings/<int:video_id>', methods=['POST'])
@login_required
def update_video_settings(video_id):
    brightness = request.form.get('brightness', type=float)
    contrast = request.form.get('contrast', type=float)
    saturate = request.form.get('saturate', type=float)
    hueRotate = request.form.get('hueRotate', type=float)

    video = Video.query.get_or_404(video_id)
    if video.user_id != current_user.id:
        abort(403)

    video.video_settings.brightness = brightness
    video.video_settings.contrast = contrast
    video.video_settings.saturate = saturate
    video.video_settings.hueRotate = hueRotate
    db.session.commit()

    return jsonify({"message": "Video settings updated successfully"})


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