Update app.py

app.py

@@ -1,89 +1,86 @@
 from flask import Flask, render_template, request, redirect, url_for, session, flash
-import sqlite3
+from werkzeug.security import generate_password_hash, check_password_hash
+from supabase import create_client
 import joblib
 import numpy as np
-from werkzeug.security import generate_password_hash, check_password_hash
 
 app = Flask(__name__)
 app.secret_key = "supersecretkey"  # change in production
 
-MODEL_PATH = "rf_house_price_model.joblib"
-model = joblib.load(MODEL_PATH)
-
-# ---------- Database Setup ----------
-def init_db():
-    conn = sqlite3.connect("users.db")
-    c = conn.cursor()
-    c.execute('''CREATE TABLE IF NOT EXISTS users
-                 (id INTEGER PRIMARY KEY AUTOINCREMENT,
-                  username TEXT UNIQUE,
-                  password TEXT)''')
-    conn.commit()
-    conn.close()
+# -------- Supabase setup --------
+SUPABASE_URL = "https://YOUR_PROJECT_ID.supabase.co"
+SUPABASE_KEY = "YOUR_ANON_KEY"
+supabase = create_client(SUPABASE_URL, SUPABASE_KEY)
 
-init_db()
+# -------- Load ML model --------
+model = joblib.load("rf_house_price_model.joblib")
 
-# ---------- Routes ----------
+# -------- Routes --------
 @app.route("/")
 def index():
     if "username" in session:
         return redirect(url_for("home"))
     return redirect(url_for("login"))
 
+# Signup
 @app.route("/signup", methods=["GET", "POST"])
 def signup():
     if request.method == "POST":
         username = request.form["username"]
-        password = generate_password_hash(request.form["password"])
-        try:
-            conn = sqlite3.connect("users.db")
-            c = conn.cursor()
-            c.execute("INSERT INTO users (username, password) VALUES (?, ?)", (username, password))
-            conn.commit()
-            conn.close()
-            flash("Signup successful! Please log in.", "success")
-            return redirect(url_for("login"))
-        except:
+        password_hashed = generate_password_hash(request.form["password"])
+        
+        # Check if username exists
+        res = supabase.table("users").select("*").eq("username", username).execute()
+        if res.data:
             flash("Username already exists. Try another one.", "error")
-            return "Username already exists!"
+            return redirect(url_for("signup"))
+        
+        # Insert new user
+        supabase.table("users").insert({"username": username, "password": password_hashed}).execute()
+        flash("Signup successful! Please log in.", "success")
+        return redirect(url_for("login"))
+
     return render_template("signup.html")
 
+# Login
 @app.route("/login", methods=["GET", "POST"])
 def login():
     if request.method == "POST":
         username = request.form["username"]
         password = request.form["password"]
-        conn = sqlite3.connect("users.db")
-        c = conn.cursor()
-        c.execute("SELECT * FROM users WHERE username=?", (username,))
-        user = c.fetchone()
-        conn.close()
-        if user and check_password_hash(user[2], password):
+
+        res = supabase.table("users").select("*").eq("username", username).execute()
+        if res.data and check_password_hash(res.data[0]["password"], password):
             session["username"] = username
             flash("Login successful!", "success")
             return redirect(url_for("home"))
-        else:
-            flash("Invalid credentials. Try again.", "error")
-            return "Invalid credentials"
+        
+        flash("Invalid credentials. Try again.", "error")
+        return redirect(url_for("login"))
+
     return render_template("login.html")
 
+# Logout
 @app.route("/logout")
 def logout():
     session.pop("username", None)
     flash("You have been logged out.", "info")
     return redirect(url_for("login"))
 
+# Home page
 @app.route("/home")
 def home():
     if "username" not in session:
         return redirect(url_for("login"))
     return render_template("home.html", username=session["username"])
 
+# Predict page
 @app.route("/predict", methods=["GET", "POST"])
 def predict():
     if "username" not in session:
         return redirect(url_for("login"))
-    
+
+    prediction = None
     if request.method == "POST":
         features = [
             float(request.form["longitude"]),
@@ -96,10 +93,9 @@ def predict():
             float(request.form["median_income"]),
             float(request.form["rooms_per_household"])
         ]
-        prediction = model.predict([features])[0]
-        return render_template("predict.html", prediction=round(prediction, 2))
-    
-    return render_template("predict.html", prediction=None)
+        prediction = round(model.predict([features])[0], 2)
+
+    return render_template("predict.html", prediction=prediction)
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)