Update app.py

app.py

@@ -1,41 +1,32 @@
-from flask import Flask, render_template, jsonify, request, redirect, url_for
-from prediction import getAgeGender
-
-app = Flask(__name__)
-
-
-mediaDir = "./static/uploaded/"
-
-
-@app.route("/", methods = ['GET'])
-def index():
-    if request.method=='GET':
-        return render_template("index.html")
-    else:
-        print("Post request sent on Home Page.")
-        return render_template("index.html")
-
-
-@app.route("/predict", methods = ['GET', 'POST'])  
-def predict():
-    if request.method=='GET':
-        return redirect(url_for('index'))
-    else:
-        f = request.files['upload']
-        if f.filename != "":
-            image_path = mediaDir+f.filename
-            uploaded_file_path = "uploaded/"+f.filename
-            f.save(image_path)
-            age, gender = getAgeGender(image_path)
-            return render_template("prediction.html", AGE=age, GENDER=gender, FILEPATH=uploaded_file_path)
-        else:
-            return render_template("no_image.html")
-
-
-@app.errorhandler(404)
-def page_not_found(error):
-    return render_template('404.html'), 404
-
-
-if __name__ == '__main__':
-    app.run(host="0.0.0.0", port=80, use_reloader=True, debug=True)
+import streamlit as st
+from streamlit_option_menu import option_menu
+import prediction
+import home
+import contact
+
+
+st.write('### Age and Gender Prediction')
+st.write('##### This page is created by [Simran Bola]')
+st.markdown('---')
+
+
+selected = option_menu(None, ["Home", "Predict","Contact"], 
+    icons=['house', 'gear-fill', 'envelope-at-fill'], 
+    menu_icon="cast", default_index=0, orientation="vertical",
+    styles={
+        "icon": {"color": "yellow", "font-size": "20px"}, 
+        "nav-link": {"font-size": "20px", "text-align": "left", "margin":"1px", "--hover-color": "#eee"},
+        "nav-link-selected": {"background-color": "grey"},
+    }
+)
+    
+selected
+    
+
+if selected == 'Predict':
+    prediction.run()
+
+elif selected == 'Home':
+    home.run()
+else:
+    contact.run()