fix(server): 解决Windows平台兼容性问题并添加调试日志
- 修复SocketIO在Windows平台的初始化问题 - 添加平台检测逻辑以处理不同环境的配置差异 - 在任务创建接口中添加请求参数日志记录 - 移除设备配置的注释以确保正确的设备选择main
parent
c6de10280a
commit
525bd3d91c
22
server.py
22
server.py
|
|
@ -22,11 +22,21 @@ from mandatory_model_crypto import ModelEncryptionService, validate_models_befor
|
|||
# Flask初始化
|
||||
app = Flask(__name__, static_url_path='/static')
|
||||
CORS(app)
|
||||
socketio = SocketIO(app,
|
||||
cors_allowed_origins="*",
|
||||
async_mode='threading',
|
||||
allow_unsafe_werkzeug=True,
|
||||
max_http_buffer_size=5 * 1024 * 1024)
|
||||
|
||||
# 处理Windows平台兼容性问题
|
||||
import platform
|
||||
|
||||
if platform.system() == 'Windows':
|
||||
socketio = SocketIO(app,
|
||||
cors_allowed_origins="*",
|
||||
async_mode='threading',
|
||||
allow_unsafe_werkzeug=True, #
|
||||
max_http_buffer_size=5 * 1024 * 1024)
|
||||
else:
|
||||
socketio = SocketIO(app,
|
||||
cors_allowed_origins="*",
|
||||
async_mode='threading',
|
||||
max_http_buffer_size=5 * 1024 * 1024)
|
||||
|
||||
_initialized = False
|
||||
|
||||
|
|
@ -67,6 +77,7 @@ def create_task():
|
|||
return jsonify({"status": "error", "message": "请求体不能为空"}), 400
|
||||
|
||||
data = request.json
|
||||
logger.info(f"请求参数: {data}")
|
||||
logger.info(f"收到创建任务请求: {data.get('taskname', '未命名')}")
|
||||
|
||||
# 验证必须的参数
|
||||
|
|
@ -163,6 +174,7 @@ def create_task():
|
|||
'color': model_data.get('color'),
|
||||
'line_width': int(model_data.get('line_width', 1)),
|
||||
'device': model_data.get('device', 'cuda:0' if torch.cuda.is_available() else 'cpu'),
|
||||
# 'device': 'cpu',
|
||||
'half': model_data.get('half', True),
|
||||
'enabled': model_data.get('enabled', True),
|
||||
# 注意:不再需要 download_url,模型从本地加载
|
||||
|
|
|
|||
Loading…
Reference in New Issue