anomaly_detection_ecg/ad_app.py

import streamlit as st
import requests as re
import zipfile 
import json
from time import sleep
import yaml
import pandas as pd

from src.prepare_data import get_data

# title of web app

SERVICE_URL = "https://ecg-ad.herokuapp.com/predict"

st.markdown('# **Web App for Anomaly Detection in ECG signals**')
bar = st.progress(0)

st.sidebar.header('Upload pickle file')

def save_zip():
    list_files = ['ecg_data/pred.pickle']
    with zipfile.ZipFile('final.zip', 'w') as zipF:
      for file in list_files:
         zipF.write(file, compress_type=zipfile.ZIP_DEFLATED)
      zipF.close()

with st.sidebar.form(key='my_form'):
    uploaded_file = st.file_uploader("Choose a file", accept_multiple_files=True)
    if uploaded_file is not None:
        array = get_data(dir)
        df = pd.read_csv(uploaded_file)
        array = df.array
        st.write(array)
    submit_button = st.form_submit_button(label='Submit')

    if submit_button:

        res = re.post(SERVICE_URL,array)
        json_str = json.dumps(res.json())
        resp = json.loads(json_str)
        

        st.write(f"""### The predictions are: {resp[0]}$.""")
        import pickle
        f=open('ecg_data/pred.pickle','wb')
        pickle.dump(resp[0], f)
        f.close()

        save_zip()

        with open("final.zip", "rb") as zip_download:
            btn = st.download_button(
                label="Download",
                data=zip_download,
                file_name="final.zip",
                mime="application/zip"
            )