import os
import wfdb
import numpy as np
import gradio as gr
from models.inception import *


def load_data():
    cwd = os.getcwd()
    sample_data = f"{cwd}/sample_data/ath_001"
    ecg = wfdb.rdsamp(sample_data)
    return np.asarray(ecg)

def load_model(sample_frequency,recording_time, num_leads):
    cwd = os.getcwd()
    weights = f"{cwd}/models/weights/model_weights_leadI.h5"
    model = build_model((sample_frequency * recording_time, num_leads), 1)
    model.load_weights(weights)
    return model


def run(ecg):
    SAMPLE_FREQUENCY = 100
    TIME = 10
    NUM_LEADS = 1
    data = load_data()
    model = load_model(sample_frequency=SAMPLE_FREQUENCY,recording_time=TIME,num_leads=NUM_LEADS)
    predicion = model.predict(data)
    return predicion

iface = gr.Interface(fn=run, inputs="text", outputs="text")
iface.launch()