feat: PAAI 피드백 루프 시스템 구현 (1-2단계)
- paai_feedback.py: 피드백 API + SQLite 저장
- PMR 화면: 👎 클릭 시 코멘트 입력 모달
- 카테고리: 약물상호작용/적응증/용법용량/기타
- 피드백 규칙 자동 정제 (기본 버전)
This commit is contained in:
thug0bin
225
backend/paai_feedback.py
Normal file
225
backend/paai_feedback.py
Normal file
@@ -0,0 +1,225 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
PAAI 피드백 루프 시스템
|
||||
- 피드백 수집, AI 정제, 프롬프트 인젝션
|
||||
"""
|
||||
|
||||
import sqlite3
|
||||
import os
|
||||
import json
|
||||
from datetime import datetime
|
||||
from flask import Blueprint, request, jsonify
|
||||
|
||||
paai_feedback_bp = Blueprint('paai_feedback', __name__)
|
||||
|
||||
# DB 경로
|
||||
DB_PATH = os.path.join(os.path.dirname(__file__), 'db', 'paai_feedback.db')
|
||||
|
||||
def get_db():
|
||||
"""DB 연결"""
|
||||
os.makedirs(os.path.dirname(DB_PATH), exist_ok=True)
|
||||
conn = sqlite3.connect(DB_PATH)
|
||||
conn.row_factory = sqlite3.Row
|
||||
return conn
|
||||
|
||||
def init_db():
|
||||
"""테이블 초기화"""
|
||||
conn = get_db()
|
||||
conn.execute('''
|
||||
CREATE TABLE IF NOT EXISTS paai_feedback (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||
|
||||
-- 컨텍스트
|
||||
prescription_id TEXT,
|
||||
patient_name TEXT,
|
||||
patient_context TEXT,
|
||||
|
||||
-- PAAI 응답
|
||||
paai_request TEXT,
|
||||
paai_response TEXT,
|
||||
|
||||
-- 피드백
|
||||
rating TEXT,
|
||||
category TEXT,
|
||||
pharmacist_comment TEXT,
|
||||
|
||||
-- AI 정제 결과
|
||||
refined_rule TEXT,
|
||||
confidence REAL,
|
||||
|
||||
-- 적용 상태
|
||||
applied_to_prompt INTEGER DEFAULT 0,
|
||||
applied_to_training INTEGER DEFAULT 0
|
||||
)
|
||||
''')
|
||||
conn.commit()
|
||||
conn.close()
|
||||
|
||||
# 앱 시작 시 테이블 생성
|
||||
init_db()
|
||||
|
||||
|
||||
@paai_feedback_bp.route('/api/paai/feedback', methods=['POST'])
|
||||
def submit_feedback():
|
||||
"""피드백 제출"""
|
||||
try:
|
||||
data = request.json
|
||||
|
||||
conn = get_db()
|
||||
cursor = conn.cursor()
|
||||
|
||||
cursor.execute('''
|
||||
INSERT INTO paai_feedback (
|
||||
prescription_id, patient_name, patient_context,
|
||||
paai_request, paai_response,
|
||||
rating, category, pharmacist_comment
|
||||
) VALUES (?, ?, ?, ?, ?, ?, ?, ?)
|
||||
''', (
|
||||
data.get('prescription_id'),
|
||||
data.get('patient_name'),
|
||||
json.dumps(data.get('patient_context', {}), ensure_ascii=False),
|
||||
data.get('paai_request'),
|
||||
data.get('paai_response'),
|
||||
data.get('rating'), # 'good' or 'bad'
|
||||
data.get('category'), # 'interaction', 'indication', 'dosage', 'other'
|
||||
data.get('pharmacist_comment')
|
||||
))
|
||||
|
||||
feedback_id = cursor.lastrowid
|
||||
conn.commit()
|
||||
|
||||
# bad 피드백이고 코멘트가 있으면 AI 정제 시도
|
||||
if data.get('rating') == 'bad' and data.get('pharmacist_comment'):
|
||||
refined = refine_feedback_async(feedback_id, data)
|
||||
if refined:
|
||||
cursor.execute('''
|
||||
UPDATE paai_feedback
|
||||
SET refined_rule = ?, confidence = ?
|
||||
WHERE id = ?
|
||||
''', (refined['rule'], refined['confidence'], feedback_id))
|
||||
conn.commit()
|
||||
|
||||
conn.close()
|
||||
|
||||
return jsonify({
|
||||
'success': True,
|
||||
'feedback_id': feedback_id,
|
||||
'message': '피드백이 저장되었습니다.'
|
||||
})
|
||||
|
||||
except Exception as e:
|
||||
return jsonify({'success': False, 'error': str(e)}), 500
|
||||
|
||||
|
||||
def refine_feedback_async(feedback_id, data):
|
||||
"""피드백을 규칙으로 정제 (동기 버전 - 나중에 비동기로 변경 가능)"""
|
||||
try:
|
||||
# TODO: AI 호출로 정제
|
||||
# 지금은 간단히 코멘트를 규칙 형태로 저장
|
||||
comment = data.get('pharmacist_comment', '')
|
||||
if not comment:
|
||||
return None
|
||||
|
||||
# 간단한 규칙 형태로 변환
|
||||
category = data.get('category', 'other')
|
||||
rule = f"[{category}] {comment}"
|
||||
|
||||
return {
|
||||
'rule': rule,
|
||||
'confidence': 0.8 # 기본 신뢰도
|
||||
}
|
||||
except:
|
||||
return None
|
||||
|
||||
|
||||
@paai_feedback_bp.route('/api/paai/feedback/rules', methods=['GET'])
|
||||
def get_feedback_rules():
|
||||
"""축적된 피드백 규칙 조회 (프롬프트 인젝션용)"""
|
||||
try:
|
||||
conn = get_db()
|
||||
cursor = conn.cursor()
|
||||
|
||||
# bad 피드백 중 정제된 규칙만
|
||||
cursor.execute('''
|
||||
SELECT refined_rule, category, created_at
|
||||
FROM paai_feedback
|
||||
WHERE rating = 'bad'
|
||||
AND refined_rule IS NOT NULL
|
||||
ORDER BY created_at DESC
|
||||
LIMIT 20
|
||||
''')
|
||||
|
||||
rules = []
|
||||
for row in cursor.fetchall():
|
||||
rules.append({
|
||||
'rule': row['refined_rule'],
|
||||
'category': row['category'],
|
||||
'created_at': row['created_at']
|
||||
})
|
||||
|
||||
conn.close()
|
||||
|
||||
return jsonify({
|
||||
'success': True,
|
||||
'rules': rules,
|
||||
'count': len(rules)
|
||||
})
|
||||
|
||||
except Exception as e:
|
||||
return jsonify({'success': False, 'error': str(e)}), 500
|
||||
|
||||
|
||||
@paai_feedback_bp.route('/api/paai/feedback/stats', methods=['GET'])
|
||||
def get_feedback_stats():
|
||||
"""피드백 통계"""
|
||||
try:
|
||||
conn = get_db()
|
||||
cursor = conn.cursor()
|
||||
|
||||
cursor.execute('''
|
||||
SELECT
|
||||
COUNT(*) as total,
|
||||
SUM(CASE WHEN rating = 'good' THEN 1 ELSE 0 END) as good,
|
||||
SUM(CASE WHEN rating = 'bad' THEN 1 ELSE 0 END) as bad
|
||||
FROM paai_feedback
|
||||
''')
|
||||
row = cursor.fetchone()
|
||||
|
||||
conn.close()
|
||||
|
||||
return jsonify({
|
||||
'success': True,
|
||||
'stats': {
|
||||
'total': row['total'] or 0,
|
||||
'good': row['good'] or 0,
|
||||
'bad': row['bad'] or 0
|
||||
}
|
||||
})
|
||||
|
||||
except Exception as e:
|
||||
return jsonify({'success': False, 'error': str(e)}), 500
|
||||
|
||||
|
||||
def get_rules_for_prompt(patient_context=None):
|
||||
"""프롬프트에 주입할 규칙 목록 반환"""
|
||||
try:
|
||||
conn = get_db()
|
||||
cursor = conn.cursor()
|
||||
|
||||
# 최근 규칙 20개
|
||||
cursor.execute('''
|
||||
SELECT refined_rule
|
||||
FROM paai_feedback
|
||||
WHERE rating = 'bad'
|
||||
AND refined_rule IS NOT NULL
|
||||
ORDER BY created_at DESC
|
||||
LIMIT 20
|
||||
''')
|
||||
|
||||
rules = [row['refined_rule'] for row in cursor.fetchall()]
|
||||
conn.close()
|
||||
|
||||
return rules
|
||||
except:
|
||||
return []
|
||||
Reference in New Issue
Block a user