Update app.py

app.py

@@ -1,13 +1,20 @@
 from analze import *
+import streamlit as st
+
+def is_valid_filetype(file):
+    valid_extensions = ['.eml', '.txt']
+    return any(file.name.endswith(ext) for ext in valid_extensions)
+
+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:
+        if is_valid_filetype(uploaded_file):
+            st.success("File uploaded successfully!")
+            # Process the file here
+        else:
+            st.error("Invalid file type. Please upload an EML or TXT file.")
 
-app = Flask(__name__)
-
-@app.route('/')
-def home():
-    return render_template('home.html')
-
-
-@app.route('/upload', methods=['GET', 'POST'])
 def upload_file():
     if request.method == 'POST':
         # Check if a file was uploaded
@@ -29,6 +36,6 @@ def upload_file():
     return render_template('home.html')
 
 
-
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=8000)
+    main()
+