Update server.py

server.py

@@ -1,7 +1,6 @@
 import os
 import sqlite3
 import secrets
-import pathlib
 from datetime import datetime, timedelta
 from flask import Flask, request, redirect, url_for, render_template, session, flash
 from flask_limiter import Limiter
@@ -18,23 +17,14 @@ HF_REPO_ID = "0vergeared/otp-logs"
 
 # Paths
 DATABASE = "otp.db"
-TEMPLATES = "templates"
-STATIC = "static"
+EXPORT_CSV = "/tmp/otp_all.csv"
 
-# Setup
-app = Flask(__name__, template_folder=TEMPLATES, static_folder=STATIC)
+# Flask App
+app = Flask(__name__, template_folder="templates", static_folder="static")
 app.secret_key = FLASK_SECRET
-
-# Rate Limiting
 limiter = Limiter(get_remote_address, app=app)
 
-# Make sure Hugging Face Hub doesn't cache
-os.environ["HF_DATASETS_CACHE"] = "/tmp/hf_cache"
-
-# Make sure templates/static folders exist
-os.makedirs(TEMPLATES, exist_ok=True)
-os.makedirs(STATIC, exist_ok=True)
-
+# OTP utils
 def init_db():
     with sqlite3.connect(DATABASE) as conn:
         c = conn.cursor()
@@ -53,22 +43,21 @@ def generate_otp():
     return ''.join(secrets.choice("0123456789") for _ in range(6))
 
 def upload_to_hf(otp_row):
-    csv_path = "/tmp/otp_all.csv"
-    file_exists = os.path.exists(csv_path)
-
+    file_exists = os.path.exists(EXPORT_CSV)
     df_new = pd.DataFrame([otp_row], columns=["otp", "generated_at", "expires_at", "used_at", "ip", "status"])
+
     if file_exists:
-        df_old = pd.read_csv(csv_path)
+        df_old = pd.read_csv(EXPORT_CSV)
         df = pd.concat([df_old, df_new], ignore_index=True)
     else:
         df = df_new
 
-    df.to_csv(csv_path, index=False)
+    df.to_csv(EXPORT_CSV, index=False)
 
     try:
         api = HfApi()
         api.upload_file(
-            path_or_fileobj=csv_path,
+            path_or_fileobj=EXPORT_CSV,
             path_in_repo="otp_all.csv",
             repo_id=HF_REPO_ID,
             token=HF_TOKEN,
@@ -76,8 +65,9 @@ def upload_to_hf(otp_row):
         )
         print("✅ OTP log uploaded to HF.")
     except Exception as e:
-        print("⚠️ Failed to upload to HF:", e)
+        print("⚠️ Hugging Face upload failed:", e)
 
+# Routes
 @app.route("/", methods=["GET", "POST"])
 def index():
     if request.method == "POST":
@@ -101,14 +91,13 @@ def index():
                     c.execute("UPDATE otps SET used_at = ?, status = ? WHERE otp = ?", (used_time, "used", otp_input))
                     conn.commit()
 
-                    # Log to Hugging Face
                     otp_row = [otp_input, gen_at, exp_at, used_time, request.remote_addr, "used"]
                     upload_to_hf(otp_row)
 
-                    flash("OTP accepted. Installation can continue!", "success")
+                    flash("✅ OTP accepted. Installation can proceed!", "success")
                     return redirect(url_for("index"))
             else:
-                flash("Invalid OTP!", "danger")
+                flash("❌ Invalid OTP!", "danger")
 
     return render_template("index.html")
 
@@ -121,7 +110,7 @@ def admin():
             session["admin"] = True
             return redirect(url_for("dashboard"))
         else:
-            flash("Login failed", "danger")
+            flash("❌ Login failed", "danger")
     return render_template("admin_login.html")
 
 @app.route("/dashboard", methods=["GET", "POST"])
@@ -141,7 +130,6 @@ def dashboard():
                       (otp, now.isoformat(), expires.isoformat(), ip, "generated"))
             conn.commit()
 
-        # Log to HF
         otp_row = [otp, now.isoformat(), expires.isoformat(), None, ip, "generated"]
         upload_to_hf(otp_row)