Create app.py

app.py

@@ -0,0 +1,28 @@
+from flask import Flask, render_template, request
+import joblib
+import numpy as np
+
+# Load trained model
+model = joblib.load("mental_wellness.pkl")
+
+app = Flask(__name__)
+
+@app.route("/", methods=["GET", "POST"])
+def index():
+    prediction = None
+    if request.method == "POST":
+        try:
+            screen_time = float(request.form["screen_time"])
+            sleep_hours = float(request.form["sleep_hours"])
+            work_screen = float(request.form["work_screen"])
+
+            features = np.array([[screen_time, sleep_hours, work_screen]])
+            prediction = model.predict(features)[0]
+            prediction = round(prediction, 2)
+        except Exception as e:
+            prediction = f"Error: {str(e)}"
+
+    return render_template("index.html", prediction=prediction)
+
+if __name__ == "__main__":
+    app.run(debug=True)