Cleanup: Deleting 03.Code/업로드용/handlers/briefing/processor.py
This commit is contained in:
@@ -1,263 +0,0 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
기획서 (briefing) 처리 로직
|
||||
- 1~2매 핵심 보고서
|
||||
- Navy 스타일 기본 적용"""
|
||||
|
||||
def __init__(self):
|
||||
self.prompts_dir = Path(__file__).parent / 'prompts'
|
||||
|
||||
def _load_prompt(self, filename: str) -> str:
|
||||
"""프롬프트 파일 로드"""
|
||||
return load_prompt(str(self.prompts_dir), filename)
|
||||
|
||||
def _get_step1_prompt(self) -> str:
|
||||
"""1단계: 구조 추출 프롬프트"""
|
||||
prompt = self._load_prompt('step1_extract.txt')
|
||||
if prompt:
|
||||
return prompt
|
||||
return """HTML 문서를 분석하여 JSON 구조로 추출하세요.
|
||||
원문 텍스트는 그대로 보존하고, 구조만 정리해주십시오."""
|
||||
|
||||
def _get_step1_5_prompt(self) -> str:
|
||||
"""1.5단계: 배치 계획 프롬프트"""
|
||||
prompt = self._load_prompt('step1_5_plan.txt')
|
||||
if prompt:
|
||||
return prompt
|
||||
return """JSON 구조를 분석하여 페이지 배치 계획을 수립하세요."""
|
||||
|
||||
def _get_step2_prompt(self) -> str:
|
||||
"""2단계: HTML 생성 프롬프트"""
|
||||
prompt = self._load_prompt('step2_generate.txt')
|
||||
if prompt:
|
||||
return prompt
|
||||
return """JSON 구조를 가진 기획서용 HTML로 변환하세요.
|
||||
Navy 스타일, A4 크기, Noto Sans KR 폰트를 사용하세요."""
|
||||
|
||||
def _content_too_long(self, html: str, max_sections_per_page: int = 4) -> bool:
|
||||
"""페이지별 콘텐츠 양 체크"""
|
||||
from bs4 import BeautifulSoup
|
||||
soup = BeautifulSoup(html, 'html.parser')
|
||||
|
||||
sheets = soup.find_all('div', class_='sheet')
|
||||
for sheet in sheets:
|
||||
sections = sheet.find_all('div', class_='section')
|
||||
if len(sections) > max_sections_per_page:
|
||||
return True
|
||||
|
||||
all_li = sheet.find_all('li')
|
||||
if len(all_li) > 12:
|
||||
return True
|
||||
|
||||
steps = sheet.find_all('div', class_='process-step')
|
||||
if len(steps) > 6:
|
||||
return True
|
||||
|
||||
return False
|
||||
|
||||
def generate(self, content: str, options: dict) -> dict:
|
||||
"""기획서 생성"""
|
||||
try:
|
||||
if not content.strip():
|
||||
return {'error': '내용이 부족합니다. 파일을 업로드해주십시오.'}
|
||||
|
||||
page_option = options.get('page_option', '1')
|
||||
department = options.get('department', '기획실')
|
||||
additional_prompt = options.get('instruction', '')
|
||||
|
||||
# ============== 1단계: 구조 추출 ==============
|
||||
step1_prompt = self._get_step1_prompt()
|
||||
step1_message = f"""다음 HTML 문서의 구조를 분석하여 JSON으로 추출해주십시오.
|
||||
|
||||
## 원본 HTML
|
||||
{content}
|
||||
|
||||
---
|
||||
위 문서를 분석하여 JSON 구조로 출력하세요. 설명 없이 출력."""
|
||||
|
||||
step1_response = call_claude(step1_prompt, step1_message, max_tokens=4000)
|
||||
structure_json = extract_json(step1_response)
|
||||
|
||||
if not structure_json:
|
||||
structure_json = {"raw_content": content, "parse_failed": True}
|
||||
|
||||
# ============== 1.5단계: 배치 계획 ==============
|
||||
step1_5_prompt = self._get_step1_5_prompt()
|
||||
step1_5_message = f"""다음 JSON 구조를 분석하여 페이지 배치 계획을 수립해주십시오.
|
||||
|
||||
## 문서 구조 (JSON)
|
||||
{json.dumps(structure_json, ensure_ascii=False, indent=2)}
|
||||
|
||||
## 페이지 수
|
||||
{page_option}페이지
|
||||
|
||||
---
|
||||
배치 계획 JSON만 출력하세요. 설명 없이."""
|
||||
|
||||
step1_5_response = call_claude(step1_5_prompt, step1_5_message, max_tokens=4000)
|
||||
page_plan = extract_json(step1_5_response)
|
||||
|
||||
if not page_plan:
|
||||
page_plan = {"page_plan": {}, "parse_failed": True}
|
||||
|
||||
# ============== 2단계: HTML 생성 ==============
|
||||
page_instructions = {
|
||||
'1': '1페이지로 모든 내용을 압축하여 설계하십시오.',
|
||||
'2': '2페이지로 설계하십시오. 1페이지 본문, 2페이지 [첨부]입니다.',
|
||||
'n': '여러 페이지로 설계하십시오. 1페이지 본문, 이후 [첨부] 페이지로 구성합니다.'
|
||||
}
|
||||
|
||||
step2_prompt = self._get_step2_prompt()
|
||||
step2_message = f"""다음 배치 계획과 문서 구조를 기반으로 기획서용 HTML을 생성해주십시오.
|
||||
|
||||
## 배치 계획
|
||||
{json.dumps(page_plan, ensure_ascii=False, indent=2)}
|
||||
|
||||
## 문서 구조 (JSON)
|
||||
{json.dumps(structure_json, ensure_ascii=False, indent=2)}
|
||||
|
||||
## 작성 부서
|
||||
{department}
|
||||
|
||||
## 추가 요청사항
|
||||
{additional_prompt if additional_prompt else '없음'}
|
||||
|
||||
---
|
||||
위 JSON을 바탕으로 완성된 HTML 문서를 설계하십시오.
|
||||
코드 블록 안에 </html>까지 전체 출력."""
|
||||
|
||||
step2_response = call_claude(step2_prompt, step2_message, max_tokens=8000)
|
||||
html_content = extract_html(step2_response)
|
||||
|
||||
# 내용이 너무 길면 압축
|
||||
if self._content_too_long(html_content):
|
||||
compress_message = f"""현재 생성된 HTML의 페이지별 콘텐츠가 너무 많습니다.
|
||||
각 페이지별 섹션 3~4개, 리스트 항목 8개 이하로 압축해주십시오.
|
||||
|
||||
{html_content}
|
||||
|
||||
코드 블록 안에 완성된 HTML만 출력하세요."""
|
||||
|
||||
compress_response = call_claude(step2_prompt, compress_message, max_tokens=8000)
|
||||
html_content = extract_html(compress_response)
|
||||
|
||||
# 세션 저장
|
||||
session['original_html'] = content
|
||||
session['current_html'] = html_content
|
||||
session['structure_json'] = json.dumps(structure_json, ensure_ascii=False)
|
||||
session['conversation'] = []
|
||||
|
||||
return {
|
||||
'success': True,
|
||||
'html': html_content,
|
||||
'structure': structure_json
|
||||
}
|
||||
|
||||
except Exception as e:
|
||||
import traceback
|
||||
return {'error': str(e), 'trace': traceback.format_exc()}
|
||||
|
||||
def refine(self, feedback: str, current_html: str, original_html: str = '') -> dict:
|
||||
"""피드백 반영"""
|
||||
try:
|
||||
if not feedback.strip():
|
||||
return {'error': '피드백 내용을 입력해주십시오.'}
|
||||
|
||||
if not current_html:
|
||||
return {'error': '수정할 HTML이 없습니다.'}
|
||||
|
||||
refine_prompt = f"""당신은 HTML 기획서 수정 전문가입니다.
|
||||
사용자 피드백을 반영하여 HTML을 수정합니다.
|
||||
|
||||
## 규칙
|
||||
1. 피드백에서 언급된 부분만 정확히 수정
|
||||
2. 전체 구조와 스타일 그대로 유지
|
||||
3. 완성된 HTML 문서로 출력 (<!DOCTYPE html> ~ </html>)
|
||||
4. 코드 블록에 출력
|
||||
5. 원본 문서 텍스트를 참조하여 누락 내용 복구 가능
|
||||
|
||||
## 원본 HTML (참조용)
|
||||
{original_html[:3000] if original_html else '없음'}...
|
||||
|
||||
## 현재 HTML
|
||||
{current_html}
|
||||
|
||||
## 사용자 피드백
|
||||
{feedback}
|
||||
|
||||
---
|
||||
피드백을 반영한 최종 수정된 HTML을 출력하세요."""
|
||||
|
||||
response = call_claude("", refine_prompt, max_tokens=8000)
|
||||
new_html = extract_html(response)
|
||||
|
||||
session['current_html'] = new_html
|
||||
|
||||
return {
|
||||
'success': True,
|
||||
'html': new_html
|
||||
}
|
||||
|
||||
except Exception as e:
|
||||
return {'error': str(e)}
|
||||
|
||||
def refine_selection(self, current_html: str, selected_text: str, user_request: str) -> dict:
|
||||
"""부분 수정"""
|
||||
try:
|
||||
if not current_html or not selected_text or not user_request:
|
||||
return {'error': '필수 데이터가 없습니다.'}
|
||||
|
||||
message = client.messages.create(
|
||||
model="claude-sonnet-4-20250514",
|
||||
max_tokens=8000,
|
||||
messages=[{
|
||||
"role": "user",
|
||||
"content": f"""HTML 문서에서 지정된 부분만 수정해주십시오.
|
||||
|
||||
## 전체 문서 (컨텍스트 파악용)
|
||||
{current_html}
|
||||
|
||||
## 수정 대상 텍스트
|
||||
"{selected_text}"
|
||||
|
||||
## 수정 요청
|
||||
{user_request}
|
||||
|
||||
## 규칙
|
||||
1. 요청을 분석하여 수정 유형을 판단:
|
||||
- TEXT: 텍스트 내용만 수정 (번역, 문체 변경, 단어 수정, 요약 등)
|
||||
- STRUCTURE: HTML 구조 변경 필요 (표 생성, 배치 추가, 태그 변경 등)
|
||||
|
||||
2. 반드시 다음 형식으로만 출력:
|
||||
|
||||
TYPE: (TEXT 또는 STRUCTURE)
|
||||
CONTENT:
|
||||
(수정 내용)
|
||||
|
||||
3. TEXT형: 수정된 텍스트만 출력 (HTML 태그 포함하지 않은 순수텍스트)
|
||||
4. STRUCTURE형: 해당 부분의 HTML 전체 출력 (기존 이미지 유지)
|
||||
5. 문체유지 (~함, ~임, ~함)
|
||||
"""
|
||||
}]
|
||||
)
|
||||
|
||||
result = message.content[0].text
|
||||
result = result.replace('```html', '').replace('```', '').strip()
|
||||
|
||||
edit_type = 'TEXT'
|
||||
content = result
|
||||
|
||||
if 'TYPE:' in result and 'CONTENT:' in result:
|
||||
type_line = result.split('CONTENT:')[0]
|
||||
if 'STRUCTURE' in type_line:
|
||||
edit_type = 'STRUCTURE'
|
||||
content = result.split('CONTENT:')[1].strip()
|
||||
|
||||
return {
|
||||
'success': True,
|
||||
'type': edit_type,
|
||||
'html': content
|
||||
}
|
||||
|
||||
except Exception as e:
|
||||
return {'error': str(e)}
|
||||
Reference in New Issue
Block a user