app.py

from analze import *
import streamlit as st
from datasets import FileSystemDataset
import huggingface_hub as hf_hub

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!")
        # Process the file here
        # st.write(uploaded_file.read())
        st.write(get_features(uploaded_file))
        st.write(predict_content(text_feature(uploaded_file)))
        # st.write(predict_html(html_tags_feature(uploaded_file)))               
        # st.write(predict_num(num_feature(uploaded_file)))
        # st.write(predict_extra(extra_feature(uploaded_file)))


if __name__ == '__main__':
    main()