mirror of
https://github.com/azaion/detections.git
synced 2026-04-22 06:46:32 +00:00
Oleksandr Bezdieniezhnykh
90 lines
2.1 KiB
Python
90 lines
2.1 KiB
Python
import os
|
|
|
|
from flask import Flask, request
|
|
|
|
app = Flask(__name__)
|
|
|
|
_mode = "normal"
|
|
_annotations: list = []
|
|
|
|
|
|
def _fail():
|
|
return _mode == "error"
|
|
|
|
|
|
@app.route("/annotations", methods=["POST"])
|
|
def annotations():
|
|
if _fail():
|
|
return "", 503
|
|
_annotations.append(request.get_json(silent=True))
|
|
return "", 200
|
|
|
|
|
|
@app.route("/auth/refresh", methods=["POST"])
|
|
def auth_refresh():
|
|
if _fail():
|
|
return "", 503
|
|
return {"token": "refreshed-test-token"}
|
|
|
|
|
|
@app.route("/api/users/<user_id>/ai-settings", methods=["GET"])
|
|
def user_ai_settings(user_id):
|
|
if _fail():
|
|
return "", 503
|
|
return {
|
|
"frame_period_recognition": 4,
|
|
"frame_recognition_seconds": 2,
|
|
"probability_threshold": 0.25,
|
|
"tracking_distance_confidence": 0.1,
|
|
"tracking_probability_increase": 0.1,
|
|
"tracking_intersection_threshold": 0.6,
|
|
"model_batch_size": 8,
|
|
"big_image_tile_overlap_percent": 20,
|
|
"altitude": 400,
|
|
"focal_length": 24,
|
|
"sensor_width": 23.5,
|
|
}
|
|
|
|
|
|
@app.route("/api/media/<media_id>", methods=["GET"])
|
|
def media_path(media_id):
|
|
if _fail():
|
|
return "", 503
|
|
root = os.environ.get("MEDIA_DIR", "/media")
|
|
if media_id.startswith("sse-") or media_id.startswith("video-"):
|
|
return {"path": f"{root}/video_test01.mp4"}
|
|
return {"path": f"{root}/image_small.jpg"}
|
|
|
|
|
|
@app.route("/mock/config", methods=["POST"])
|
|
def mock_config():
|
|
global _mode
|
|
body = request.get_json(silent=True) or {}
|
|
mode = body.get("mode", "normal")
|
|
if mode not in ("normal", "error"):
|
|
return "", 400
|
|
_mode = mode
|
|
return "", 200
|
|
|
|
|
|
@app.route("/mock/reset", methods=["POST"])
|
|
def mock_reset():
|
|
global _mode, _annotations
|
|
_mode = "normal"
|
|
_annotations.clear()
|
|
return "", 200
|
|
|
|
|
|
@app.route("/mock/status", methods=["GET"])
|
|
def mock_status():
|
|
return {
|
|
"mode": _mode,
|
|
"annotation_count": len(_annotations),
|
|
"annotations": list(_annotations),
|
|
}
|
|
|
|
|
|
@app.route("/mock/annotations", methods=["GET"])
|
|
def mock_annotations_list():
|
|
return {"annotations": list(_annotations)}
|