shuqianpinggu/app/routes/message_routes.py

from flask import Blueprint, jsonify, request
from ..models import Session, Message
from .. import db
from . import globals
import json,ast
# from .report_routes import consultation

message_routes = Blueprint('message', __name__)


def filter_data(data):
    return {
        "content": data.get("content"),
        "remark": data.get("remark"),
        "role": data.get("role"),
        "sessionId": data.get("sessionId")
    }


def validate_session_id(sessionId):
    """验证sessionId是否提供且存在于Session表中"""
    if not sessionId:
        return jsonify({"error": "sessionId is required"}), 400
    session = Session.query.filter_by(id=sessionId).first()
    if not session:
        return jsonify({"error": "Session not found"}), 404
    return None


@message_routes.route('/get-message/<int:sessionId>', methods=['GET'])
def get_message(sessionId):
    """获取当前用户的所有会话"""
    validation_result = validate_session_id(sessionId)
    if validation_result:
        return validation_result
    messages = Message.query.filter_by(sessionId=sessionId).all()
    return jsonify(messages=[message.to_dict() for message in messages]), 200


@message_routes.route('/add-message', methods=['POST'])
def add_message():
    """添加会话"""
    data = request.get_json()
    print("22222222request",request)
    result = filter_data(data)
    validation_result = validate_session_id(result["sessionId"])  # 修改为字典访问方式
    if validation_result:
        return validation_result
    if not result["role"]:  # 修改为字典访问方式
        return jsonify({"error": "role is required"}), 400

    new_message = Message(**result)
    db.session.add(new_message)
    db.session.commit()
    return jsonify({'message': 'message added successfully', 'data': new_message.to_dict()}), 201


@message_routes.route('/update-message/<int:messageId>/<int:sessionId>', methods=['PUT'])
def update_message(messageId, sessionId):
    """更新会话"""
    print("1111111111111111111request",request)
    data = request.get_json()
    remark = data.get("remark")  # 修改为字典访问方式
    validation_result = validate_session_id(sessionId)  # 修改为字典访问方式
    if validation_result:
        return validation_result  # 修改为字典访问方式
    message = Message.query.filter_by(id=messageId).first()  # 修改为字典访问方式
    if not message:  # 修改为字典访问方式
        return jsonify({"error": "message not found"}), 404  # 修改为字典访问方式
    if remark:
        message.remark = remark  # 修改为字典访问方式
    db.session.commit()  # 修改为字典访问方式
    return jsonify({'message': 'message updated successfully'})


@message_routes.route('/to-chat', methods=['POST'])
def to_chat():
    print("请求头:", dict(request.headers))  # 检查Content-Type
    print("原始字节数据:", request.get_data())  # 检查数据是否合法
    print("1111111111111111111request",request)
    # if not request.is_json:
    #     return jsonify({"error": "Content-Type must be application/json"}), 401

    # # 2. 获取外层JSON数据
    # try:
    #     data = request.json
    #     print("外层JSON数据:", data)
    # except Exception as e:
    #     return jsonify({"error": f"外层JSON解析失败: {str(e)}"}), 402

    # # 3. 解析嵌套的JSON字符串（msgList和patientInfo）
    # try:
    #     # 解析msgList（字符串 -> 列表）
    #     msg_list = json.loads(data["msgList"])
    #     print("解析后的msgList:", msg_list)

    #     # 解析patientInfo（字符串 -> 字典）
    #     patient_info = json.loads(data["patientInfo"])
    #     print("解析后的patientInfo:", patient_info)

    # except json.JSONDecodeError as e:
    #     return jsonify({"error": f"嵌套JSON解析失败: {str(e)}"}), 403
    # except KeyError as e:
    #     return jsonify({"error": f"缺少必要字段: {str(e)}"}), 404





    data = request.json
    print("1111111111111111111data",data)
    patientId = data['patientId']
    msgList =  data['msgList']
    patient_info = data['patientInfo']
    print("222222222patient_info",patient_info)
    if len(msgList) == 0:
        globals.consultation.init_session(patientId,case_data=patient_info)
        content = None
    else:
        content = data['msgList'][-1]['content']
    print("333333333content",type(content))
    value, option, system = None, None, None
    
    cur_card, answer = globals.consultation.qa_chat(patientId, content, msgList)
    
    if cur_card and cur_card[0] and cur_card[0]['status'] == 'success':
        value, option, system = cur_card[0]['option_value'], cur_card[1], cur_card[2]

    return jsonify({
        'answer': answer,
        'analysis': {
            'value': value, 'option': option, 'system': system
        }
    }), 201