import streamlit as st
import json
import shutil
from common.tinderwrapper import TinderWrapper
from common.helpers import create_folder, save_data

with open('config.json', 'r') as f:
    config = json.load(f)

APP_NAME = config['APP_NAME']
APP_LOGO = config['APP_LOGO']
APP_DESCRIPTION = config['APP_DESCRIPTION']

def update_progress(percent, text):
    progress_bar.progress(percent)
    status_text.text(text)

st.set_page_config(page_title=APP_NAME)
st.title(APP_NAME)
st.image(APP_LOGO, use_column_width=True)
st.markdown(APP_DESCRIPTION)

access_token = st.text_input('Tinder Access Token', help='Токен доступа, полученный из запроса Tinder')

if st.button('Запуск'):
    if not access_token:
        st.error('Необходимо задать токен доступа')
    progress_bar = st.progress(0)
    status_text = st.empty()
    wrapper = TinderWrapper(access_token)
    dump_folder = 'dump'
    create_folder(dump_folder)
    dump = wrapper.create_dump(dump_folder)
    save_data(dump)
    shutil.make_archive('output.zip', 'zip', dump_folder)
    with open('output.zip', 'rb') as f:
        bytes_data = f.read()
    st.download_button(
        label='Download Zip File',
        data=bytes_data,
        file_name='output.zip',
        mime='application/octet-stream'
    )