Spaces:
Running
Running
Nayan Ghosh commited on
Commit ·
f7ec320
1
Parent(s): be5e8ad
new
Browse files
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) >
|
| 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())
|