clear.py

import os
import streamlit as st
import encdec
import shutil


def clear():
    passowrd = st.text_input("Enter a password", type="password")
    clearbtn = st.button("clear")
    known = os.path.join(os.path.dirname(os.path.abspath(__file__)), "known_user")
    unknown = os.path.join(os.path.dirname(os.path.abspath(__file__)), "unknown_user")
    db = os.path.join(os.path.dirname(os.path.abspath(__file__)), "db.sqlite")
    if clearbtn and passowrd == encdec.encdec():
        try:
            if os.path.exists(known):
                shutil.rmtree(known)
                os.makedirs(known)
            if os.path.exists(unknown):
                shutil.rmtree(unknown)
                os.makedirs(unknown)
            if os.path.exists(db):
                os.remove(db)
            st.success("Cleared")
        except Exception as e:
            st.error(e)
    if clearbtn and passowrd != encdec.encdec():
        st.error("Password entered is incorrect")