# coding=utf-8

from __future__ import division, print_function

import os

import numpy as np

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

from werkzeug.utils import secure_filename

from gevent.pywsgi import WSGIServer

from predict import *

from utils import *

from config import get_config

app = Flask(__name__)

app.config['MAX_CONTENT_LENGTH'] = 5 * 1024 * 1024

global classesM

classesM = ['N','V','L','R','Paced','A','F']#,'f','j','E','a','J','Q','e','S']

print('Check http://127.0.0.1:5002/')

def model_predict(img_path):

    data = uploadedData(img_path, csvbool = True)

    sr = data[0]

    data = data[1:]

    size = len(data)

    if size > 9001:

        size = 9001

        data = data[:size]

    div = size // 1000

    data, peaks = preprocess(data, config)

    return predictByPart(data, peaks)

@app.route('/', methods=['GET'])

def index():

    # Main page

    return render_template('index.html')

@app.route('/predict', methods=['GET', 'POST'])

def upload():

    if request.method == 'POST':

        # Get the file from post request

        f = request.files['file']

        if not f:

            return "No file!"

        basepath = os.path.dirname(__file__)

        file_path = os.path.join(basepath, 'uploads', secure_filename(f.filename))

        try:

            print("You already analyzed the data file. We delete it and re-analyze it!")

            os.remove(file_path)

        except:

            print("The file is new!")

        f.save(file_path)

        predicted, result = model_predict(file_path)

        length = len(predicted)

        result = str(length) +" parts of the divided data were estimated as the followings with paired probabilities. \n"+result

        return format_result(result)

    return None

def format_result(text):

    # Extract the summary part at the end

    summary_start = text.rfind(")") + 1

    summary = text[summary_start:].strip()

    data = text[:summary_start].strip()

    # Format the summary sentence

    summary_parts = summary.split(", ")

    formatted_summary = "The results show: "

    formatted_summary += ", ".join(f"{part.split('-')[1]} beats labeled as {part.split('-')[0]}" for part in summary_parts)

    # Combine the formatted parts

    formatted_text = f"{data}, {formatted_summary}."

    return formatted_text

if __name__ == '__main__':

    config = get_config()

    #app.run(port=5002, debug=True)

    # Serve the app with gevent

    http_server = WSGIServer(('', 5002), app)

    http_server.serve_forever()