from time import sleep

import cv2
import threading
import time

import numpy as np
import requests
from flask import Flask, Response, render_template_string, request

import utils

app_flask = Flask(__name__)

# 全局变量，用于存储最新的帧
latest_frame:np.ndarray = None
lock = threading.Lock()
is_running = True

def update_latest_frame(n_bytes:np.ndarray):
    global latest_frame
    latest_frame=n_bytes

def generate_mjpeg():
    """生成MJPEG格式的视频流"""
    while is_running:
        with lock:
            if latest_frame is None:
                continue
            _,latest_frame_bytes = utils.frame_to_img(latest_frame)
            frame = latest_frame_bytes.tobytes()
            pass
        sleep(0.1)
        yield (b'--frame\r\n'
               b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n')


@app_flask.route('/video_flow')
def video_feed():
    """视频流路由"""
    return Response(
        generate_mjpeg(),
        mimetype='multipart/x-mixed-replace; boundary=frame'
    )

def run():
    port=2240
    print(f"推流服务启动,访问端口 127.0.0.1:{port}/video_flow")
    app_flask.run(host='0.0.0.0', port=port, threaded=True)

def video_server_run():
    thread = threading.Thread(target=run)
    thread.daemon = True  # 设置为守护线程，主线程退出时自动终止
    thread.start()
def stop():
    global is_running
    is_running=False

if __name__ == '__main__':
    run()