# 主程序入口
import os
import signal
import threading
import torch
from log import logger

import global_data as gd
from server import socketio, app

gd._init()

gd.set_value('detection_thread', None)
gd.set_value('detection_active', False)
gd.set_value('stop_event', threading.Event())
gd.set_value('mqtt_client', None)
gd.set_value('latest_drone_data', None)
gd.set_value('mqtt_data_lock', threading.Lock())

if __name__ == '__main__':
    logger.info("启动优化版YOLOv8服务")
    # 优化2: 使用最新版本YOLO和PyTorch特性
    logger.info(f"PyTorch版本: {torch.__version__}, CUDA可用: {torch.cuda.is_available()}")
    # 退出服务
    def graceful_exit(signum, frame):
        logger.info("收到退出信号，停止服务...")
        detection_active = gd.get_value('detection_active')
        detection_thread = gd.get_value('detection_thread')
        if detection_active and detection_thread:
            detection_thread.stop()
            detection_thread.join(5.0)
        logger.info("服务已退出")
        os._exit(0)


    signal.signal(signal.SIGINT, graceful_exit)
    signal.signal(signal.SIGTERM, graceful_exit)

    # 启动服务 - 禁用开发服务器调试
    socketio.run(app,
                 host='0.0.0.0',
                 port=9309,
                 debug=True,  # 禁用调试模式
                 use_reloader=False,
                 allow_unsafe_werkzeug=True)