from flask import Flask, render_template, request, send_from_directory
import os
from model_loader import get_model
from utils import process_video

app = Flask(__name__)
UPLOAD_FOLDER = 'static/uploads'
RESULT_FOLDER = 'static/results'
os.makedirs(UPLOAD_FOLDER, exist_ok=True)
os.makedirs(RESULT_FOLDER, exist_ok=True)

num_classes = 21  
model, device = get_model(num_classes)

@app.route('/', methods=['GET', 'POST'])
def index():
    if request.method == 'POST':
        file = request.files['video']
        if file:
            filepath = os.path.join(UPLOAD_FOLDER, file.filename)
            file.save(filepath)

            output_path = os.path.join(RESULT_FOLDER, 'result_' + file.filename)
            result_path = process_video(filepath, model, output_path, device)

            return render_template('index.html', video_result=result_path)

    return render_template('index.html', video_result=None)

@app.route('/static/<path:path>')
def send_static(path):
    return send_from_directory('static', path)

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