Nayan Ghosh commited on
Commit
f7ec320
·
1 Parent(s): be5e8ad
Files changed (1) hide show
  1. app.py +9 -1
app.py CHANGED
@@ -65,7 +65,7 @@ _db_dir = os.path.dirname(DB)
65
  if _db_dir and not os.path.exists(_db_dir):
66
  os.makedirs(_db_dir, exist_ok=True)
67
 
68
- def _key_ok(k): return bool(k and len(k) > 10)
69
 
70
  log.info(f"VT API: {'loaded' if _key_ok(VT_API_KEY) else 'missing'}")
71
  log.info(f"GSB API: {'loaded' if _key_ok(GSB_API_KEY) else 'missing'}")
@@ -1056,6 +1056,14 @@ def reload_models():
1056
  return jsonify({"rf":rf_model is not None,"nn":nn_model is not None})
1057
 
1058
 
 
 
 
 
 
 
 
 
1059
  @app.route("/retrain/status", methods=["GET"])
1060
  def retrain_status():
1061
  return jsonify(retrain_watcher.get_status())
 
65
  if _db_dir and not os.path.exists(_db_dir):
66
  os.makedirs(_db_dir, exist_ok=True)
67
 
68
+ def _key_ok(k): return bool(k and len(k.strip()) >= 4)
69
 
70
  log.info(f"VT API: {'loaded' if _key_ok(VT_API_KEY) else 'missing'}")
71
  log.info(f"GSB API: {'loaded' if _key_ok(GSB_API_KEY) else 'missing'}")
 
1056
  return jsonify({"rf":rf_model is not None,"nn":nn_model is not None})
1057
 
1058
 
1059
+ @app.route("/admin/verify", methods=["POST"])
1060
+ def admin_verify():
1061
+ """Let the extension verify if a given admin key is valid."""
1062
+ err = _require_admin()
1063
+ if err: return err
1064
+ return jsonify({"success": True, "message": "Admin key verified ✓"})
1065
+
1066
+
1067
  @app.route("/retrain/status", methods=["GET"])
1068
  def retrain_status():
1069
  return jsonify(retrain_watcher.get_status())