from analze import *
import streamlit as st

def main():
    st.title("Phishing Email Filtering Service")
    uploaded_file = st.file_uploader("Upload a file", type=['eml', 'txt'])
    if uploaded_file is not None:
        st.success("File uploaded successfully!")
        
        Txtf = text_feature(uploaded_file)
        Htmlf = html_tags_feature(uploaded_file)
        Numf = num_feature(uploaded_file)
        Extraf = extra_feature(uploaded_file)
        
        feat = get_features(uploaded_file)
        feat.at[0,"text"] = Txtf.at[0,"text"]
        columns = feat.columns.tolist()
        columns = [columns[-1]] + columns[:-1]
        feat = feat[columns]
        st.write(uploaded_file.read())
        st.write(feat)
        # st.write(txt)
        st.write("Text Prediction: "+predict_content(Txtf))
        st.write("HTML Tags Prediction: "+predict_html(Htmlf))
        st.write("Numeric Features Prediction: "+predict_num(Numf))
        st.write("Extra Features Prediction: "+predict_extra(Extraf))


if __name__ == '__main__':
    main()