PEF / app.py
allenchienxxx's picture
Update app.py
0128604
raw
history blame
694 Bytes
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!")
# 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()