Update server.py

server.py

@@ -24,7 +24,7 @@ app = Flask(__name__)
 app.secret_key = os.getenv("FLASK_SECRET", "changeme123")
 limiter = Limiter(get_remote_address, app=app)
 
-# --- Memory store for OTPs (optional if not using SQLite)
+# --- Memory store for OTPs
 otp_store = {}
 
 # ---------------- ROUTES ---------------- #
@@ -40,10 +40,16 @@ def admin():
         pwd = request.form.get("password")
         if user == ADMIN_USER and pwd == ADMIN_PASS:
             session["logged_in"] = True
-            return redirect(url_for("generate_otp"))
+            return redirect(url_for("dashboard"))
         flash("Invalid credentials", "error")
     return render_template("login.html")
 
+@app.route("/dashboard")
+def dashboard():
+    if not session.get("logged_in"):
+        return redirect(url_for("admin"))
+    return render_template("dashboard.html")
+
 @app.route("/generate_otp")
 def generate_otp():
     if not session.get("logged_in"):
@@ -53,11 +59,15 @@ def generate_otp():
     expiry = datetime.utcnow() + timedelta(minutes=OTP_EXPIRY_MINUTES)
     otp_store[otp] = {"expiry": expiry, "used": False}
 
-    # Save to dataset
     save_otp_to_dataset(otp, expiry.strftime("%Y-%m-%d %H:%M:%S UTC"))
 
     return render_template("otp_result.html", otp=otp, expiry=expiry.strftime("%Y-%m-%d %H:%M UTC"))
 
+@app.route("/logout")
+def logout():
+    session.clear()
+    return redirect(url_for("admin"))
+
 # ---------------- HELPERS ---------------- #
 
 def save_otp_to_dataset(otp: str, expiry: str):
@@ -65,32 +75,24 @@ def save_otp_to_dataset(otp: str, expiry: str):
         if not HF_TOKEN:
             print("❌ HF_TOKEN not set.")
             return
-        
+
         HfFolder.save_token(HF_TOKEN)
         api = HfApi()
 
-        # Load existing CSV or create new
         local_file = "otp_temp.csv"
-        df = pd.read_csv(local_file) if os.path.exists(local_file) else pd.DataFrame(columns=["otp", "expiry"])
-        
-        # Append new row
+        if not os.path.exists(local_file):
+            pd.DataFrame(columns=["otp", "expiry"]).to_csv(local_file, index=False)
+
+        df = pd.read_csv(local_file)
         df = pd.concat([df, pd.DataFrame([{"otp": otp, "expiry": expiry}])], ignore_index=True)
         df.to_csv(local_file, index=False)
 
-        # Push to HF dataset
         dataset = HFDataset.from_pandas(df)
         dataset.push_to_hub(REPO_ID, token=HF_TOKEN)
 
-        print(f"✅ OTP {otp} saved to HF dataset.")
+        print(f"✅ OTP {otp} pushed to dataset.")
     except Exception as e:
-        print("❌ Failed to push to dataset:", e)
-
-# ---------------- TEMPLATES ---------------- #
-
-@app.route("/logout")
-def logout():
-    session.clear()
-    return redirect(url_for("admin"))
+        print("❌ Dataset push failed:", e)
 
 @app.after_request
 def allow_iframe(response):