PEF / app.py
allenchienxxx's picture
Update app.py
7a5f23b
raw
history blame contribute delete
No virus
1.06 kB
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()