from flask import Flask, render_template_string, request, jsonify, session, redirect, url_for import os import fitz # PyMuPDF import re import json from werkzeug.utils import secure_filename from datetime import datetime app = Flask(__name__) app.config['SECRET_KEY'] = 'your-secret-key-here' app.config['MAX_CONTENT_LENGTH'] = 100 * 1024 * 1024 # 100MB app.config['UPLOAD_FOLDER'] = 'uploads' # uploads 폴더 생성 os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True) class MmpiExtractor: def extract_name(self, page): try: text = page.get_text("text") match = re.search(r"(?:이름|성명)\s*:\s*(\S+)", text) if match: return match.group(1) blocks = page.get_text("blocks") if blocks: return blocks[0][4].strip().split('\n')[0] except Exception: pass return "이름모름" def reformat_numeric_blocks(self, text): lines = text.split('\n') if not lines: return "" new_lines = [lines[0]] for i in range(1, len(lines)): line, stripped_line = lines[i], lines[i].strip() is_numeric = False if stripped_line and ' ' not in stripped_line: try: float(stripped_line) is_numeric = True except ValueError: is_numeric = False if is_numeric and new_lines and new_lines[-1].strip(): new_lines[-1] += f" {stripped_line}" else: new_lines.append(line) return "\n".join(new_lines) def process_files(self, file_paths): results = {} pages_to_extract = [1, 2, 3, 4] for i, file_path in enumerate(file_paths): try: doc = fitz.open(file_path) if doc.page_count < 5: continue name = self.extract_name(doc.load_page(0)) if name in results: name = f"{name}_{i+1}" combined_raw_text = "" for page_num in pages_to_extract: combined_raw_text += doc.load_page(page_num).get_text("text") + "\n" lines = combined_raw_text.split('\n') unique_lines = [] seen_lines = set() for line in lines: stripped_line = line.strip() if stripped_line and stripped_line not in seen_lines: seen_lines.add(stripped_line) unique_lines.append(line) elif not stripped_line: unique_lines.append(line) deduplicated_text = "\n".join(unique_lines) formatted_text = self.reformat_numeric_blocks(deduplicated_text) results[name] = formatted_text doc.close() except Exception as e: print(f"Error processing {file_path}: {e}") continue return results # HTML 템플릿들 MAIN_TEMPLATE = ''' MMPI-2 종합 분석 시스템

MMPI-2 종합 분석 시스템

심리검사 데이터 추출부터 유형 분석까지 원스톱 서비스

PDF 데이터 추출

MMPI-2 PDF 보고서에서 텍스트 데이터를 자동 추출합니다.

PDF 업로드

직접 데이터 입력

이름과 MMPI 점수를 직접 입력해서 분석을 진행합니다.

직접 입력

유형 분석

20-30대 미혼 집단 대상 20가지 유형으로 분류 분석합니다.

유형 분석
시스템 특징
보안
개인정보는 처리 후 즉시 삭제
빠른 처리
수 초 내 분석 결과 제공
CBTTR 기반
20년+ 검증된 치료 기법
전문성
20-30대 특화 분석
''' MANUAL_INPUT_TEMPLATE = ''' MMPI 데이터 직접 입력

MMPI 데이터 직접 입력

메인으로
기본 정보
타당도 척도
{% for scale, desc in [ ('L', '거짓말 척도'), ('F', 'F척도'), ('K', 'K척도'), ('VRIN', '비일관성'), ('TRIN', '응답 편향') ] %}
{% endfor %}
기본 임상척도
{% for scale, desc in [ ('Hs', '건강염려증'), ('D', '우울증'), ('Hy', '히스테리'), ('Pd', '정신병질적 이탈'), ('Mf', '남성성-여성성'), ('Pa', '편집증'), ('Pt', '강박증'), ('Sc', '조현병'), ('Ma', '경조증'), ('Si', '사회적 내향성') ] %}
{% endfor %}
주요 호소사항
''' PDF_UPLOAD_TEMPLATE = ''' PDF 업로드

PDF 파일 업로드

메인으로

PDF 파일을 드래그하거나 클릭해서 선택하세요

여러 개의 MMPI-2 PDF 보고서를 동시에 업로드할 수 있습니다.

''' @app.route('/') def index(): return render_template_string(MAIN_TEMPLATE) @app.route('/pdf-upload', methods=['GET', 'POST']) def pdf_upload(): if request.method == 'GET': return render_template_string(PDF_UPLOAD_TEMPLATE) try: files = request.files.getlist('files') if not files or all(f.filename == '' for f in files): return jsonify({'success': False, 'error': 'No files selected'}) file_paths = [] for file in files: if file and file.filename.lower().endswith('.pdf'): filename = secure_filename(file.filename) file_path = os.path.join(app.config['UPLOAD_FOLDER'], filename) file.save(file_path) file_paths.append(file_path) if not file_paths: return jsonify({'success': False, 'error': 'No valid PDF files'}) extractor = MmpiExtractor() results = extractor.process_files(file_paths) # 임시 파일들 삭제 for file_path in file_paths: try: os.remove(file_path) except: pass session['results'] = results return redirect(url_for('results')) except Exception as e: return jsonify({'success': False, 'error': str(e)}) @app.route('/manual-input', methods=['GET', 'POST']) def manual_input(): if request.method == 'GET': return render_template_string(MANUAL_INPUT_TEMPLATE) try: # 폼 데이터 수집 form_data = request.form.to_dict() # MMPI 점수만 추출 mmpi_scales = ['L', 'F', 'K', 'VRIN', 'TRIN', 'Hs', 'D', 'Hy', 'Pd', 'Mf', 'Pa', 'Pt', 'Sc', 'Ma', 'Si'] mmpi_scores = {} for scale in mmpi_scales: if scale in form_data and form_data[scale]: try: mmpi_scores[scale] = int(form_data[scale]) except ValueError: continue if not mmpi_scores: return render_template_string(MANUAL_INPUT_TEMPLATE + ''' ''') # 세션에 데이터 저장 session['manual_data'] = { 'name': form_data.get('name', '이름없음'), 'age': form_data.get('age', ''), 'gender': form_data.get('gender', ''), 'situation': form_data.get('situation', ''), 'complaints': form_data.get('complaints', ''), 'mmpi_scores': mmpi_scores } return redirect(url_for('analysis')) except Exception as e: return render_template_string(MANUAL_INPUT_TEMPLATE + f''' ''') @app.route('/analysis') def analysis(): manual_data = session.get('manual_data') if not manual_data: return redirect(url_for('index')) # 여기서 실제 분석 로직 구현 # 지금은 간단한 예시만 analysis_result = analyze_mmpi_data(manual_data) return render_template_string(ANALYSIS_RESULT_TEMPLATE, data=manual_data, result=analysis_result) def analyze_mmpi_data(data): """간단한 MMPI 분석 함수 (실제로는 더 복잡하게 구현)""" mmpi_scores = data['mmpi_scores'] # 간단한 분석 예시 elevated_scales = [] for scale, score in mmpi_scores.items(): if score > 65: elevated_scales.append(f"{scale}({score})") # 코드타입 간단 분석 clinical_scales = {k: v for k, v in mmpi_scores.items() if k in ['Hs', 'D', 'Hy', 'Pd', 'Mf', 'Pa', 'Pt', 'Sc', 'Ma', 'Si']} if clinical_scales: sorted_scales = sorted(clinical_scales.items(), key=lambda x: x[1], reverse=True) code_type = f"{sorted_scales[0][0]}-{sorted_scales[1][0] if len(sorted_scales) > 1 else '?'}" else: code_type = "분석불가" return { 'elevated_scales': elevated_scales, 'code_type': code_type, 'primary_concerns': ["우울감", "불안감", "사회적 위축"] if mmpi_scores.get('D', 0) > 65 else ["정상범위"], 'recommendations': ["전문가 상담 권고", "인지행동치료 고려", "정기적 모니터링"] } ANALYSIS_RESULT_TEMPLATE = ''' MMPI 분석 결과

MMPI-2 분석 결과

기본 정보
이름:{{ data.name }}
나이:{{ data.age }}세
성별:{{ '남성' if data.gender == 'M' else '여성' if data.gender == 'F' else '-' }}
상황:{{ data.situation }}
분석 요약

코드타입: {{ result.code_type }}

상승 척도: {{ ', '.join(result.elevated_scales) or '없음' }}

주요 특징: {{ ', '.join(result.primary_concerns) }}

척도 프로파일
주요 호소

{{ data.complaints or '기록된 호소사항이 없습니다.' }}

권고사항
    {% for rec in result.recommendations %}
  • {{ rec }}
    • {% endfor %}
새로운 분석
''' @app.route('/type-analysis') def type_analysis(): return render_template_string('''

20-30대 미혼 집단 유형 분석

개발 중입니다...

메인으로 돌아가기
''') @app.route('/results') def results(): results = session.get('results', {}) if not results: return redirect(url_for('index')) # 간단한 결과 페이지 return render_template_string('''

PDF 추출 결과

{% for name, content in results.items() %}
{{ name }}
{{ content }}
{% endfor %} 메인으로
''', results=results) if __name__ == '__main__': app.run(debug=True, host='0.0.0.0', port=5000)