Update app.py

app.py

@@ -1,13 +1,25 @@
 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 display_feature():
+    features = get_features(filepath)
+
+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
+            save_file(uploaded_file)
+            
+        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 +41,6 @@ def upload_file():
     return render_template('home.html')
 
 
-
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=8000)
+    main()
+