Cleanup: Deleting 03.Code/업로드용/app.py
This commit is contained in:
@@ -1,805 +0,0 @@
|
|||||||
# -*- coding: utf-8 -*-
|
|
||||||
from dotenv import load_dotenv
|
|
||||||
load_dotenv()
|
|
||||||
|
|
||||||
"""
|
|
||||||
湲踰 Light v2.0
|
|
||||||
Flask 고
|
|
||||||
+ 怨듯 湲곕
|
|
||||||
"""
|
|
||||||
import os
|
|
||||||
import io
|
|
||||||
import tempfile
|
|
||||||
import json
|
|
||||||
import shutil
|
|
||||||
from datetime import datetime
|
|
||||||
from flask import Flask, render_template, request, jsonify, Response, session, send_file
|
|
||||||
import queue
|
|
||||||
import threading
|
|
||||||
from handlers.template.template_manager import TemplateManager
|
|
||||||
from pathlib import Path
|
|
||||||
from domain_api import register_domain_routes
|
|
||||||
from db import init_db
|
|
||||||
|
|
||||||
|
|
||||||
# 臾몄
|
|
||||||
蹂
|
|
||||||
|
|
||||||
濡
|
|
||||||
몄
|
|
||||||
|
|
||||||
from handlers.template import TemplateProcessor
|
|
||||||
app = Flask(__name__)g import BriefingProcessor
|
|
||||||
app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024 # 16MB max
|
|
||||||
app.config['SECRET_KEY'] = os.environ.get('SECRET_KEY', 'geulbeot-light-secret-key-v2')
|
|
||||||
register_domain_routes(app)analyzer import DocTypeAnalyzer
|
|
||||||
init_db()
|
|
||||||
|
|
||||||
|
|
||||||
# processors
|
|
||||||
|
|
||||||
由ъ媛
|
|
||||||
template_mgr = TemplateManager()
|
|
||||||
processors = {
|
|
||||||
'briefing': BriefingProcessor(),
|
|
||||||
'report': ReportProcessor(),
|
|
||||||
'template': TemplateProcessor(),
|
|
||||||
'custom': CustomDocTypeProcessor()
|
|
||||||
}
|
|
||||||
|
|
||||||
DOC_TYPES_DEFAULT = Path('templates/default/doc_types')
|
|
||||||
DOC_TYPES_USER = Path('templates/user/doc_types')
|
|
||||||
|
|
||||||
|
|
||||||
# ============== 硫 ==============
|
|
||||||
@app.route('/')
|
|
||||||
def index():
|
|
||||||
"""硫"""
|
|
||||||
return render_template('index.html')
|
|
||||||
|
|
||||||
|
|
||||||
@app.route('/api/doc-types', methods=['GET'])
|
|
||||||
def get_doc_types():
|
|
||||||
"""臾몄
|
|
||||||
紐⑸"""
|
|
||||||
try:
|
|
||||||
doc_types = []
|
|
||||||
|
|
||||||
# default 대 ㅼ
|
|
||||||
if DOC_TYPES_DEFAULT.exists():
|
|
||||||
for folder in DOC_TYPES_DEFAULT.iterdir():
|
|
||||||
if folder.is_dir():
|
|
||||||
config_file = folder / 'config.json'
|
|
||||||
if config_file.exists():
|
|
||||||
with open(config_file, 'r', encoding='utf-8') as f:
|
|
||||||
doc_types.append(json.load(f))
|
|
||||||
|
|
||||||
# user 대 ㅼ
|
|
||||||
if DOC_TYPES_USER.exists():
|
|
||||||
for folder in DOC_TYPES_USER.iterdir():
|
|
||||||
if folder.is_dir():
|
|
||||||
config_file = folder / 'config.json'
|
|
||||||
if config_file.exists():
|
|
||||||
with open(config_file, 'r', encoding='utf-8') as f:
|
|
||||||
doc_types.append(json.load(f))
|
|
||||||
|
|
||||||
# order isDefault
|
|
||||||
doc_types.sort(key=lambda x: (x.get('order', 999), not x.get('isDefault', False)))
|
|
||||||
|
|
||||||
return jsonify(doc_types)
|
|
||||||
|
|
||||||
except Exception as e:
|
|
||||||
import traceback
|
|
||||||
return jsonify({'error': str(e), 'trace': traceback.format_exc()}), 500
|
|
||||||
|
|
||||||
|
|
||||||
@app.route('/api/doc-types', methods=['POST'])
|
|
||||||
def add_doc_type():
|
|
||||||
"""臾몄
|
|
||||||
異媛 (遺
|
|
||||||
|
|
||||||
)"""
|
|
||||||
try:
|
|
||||||
data = request.get_json()
|
|
||||||
|
|
||||||
if not data:
|
|
||||||
return jsonify({'error': 'JSON 곗
|
|
||||||
곌
|
|
||||||
⑸'}), 400
|
|
||||||
|
|
||||||
# user 대
|
|
||||||
|
|
||||||
DOC_TYPES_USER.mkdir(parents=True, exist_ok=True)
|
|
||||||
|
|
||||||
type_id = data.get('id')
|
|
||||||
if not type_id:
|
|
||||||
import time
|
|
||||||
type_id = f"user_{int(time.time())}"
|
|
||||||
data['id'] = type_id
|
|
||||||
|
|
||||||
folder_path = DOC_TYPES_USER / type_id
|
|
||||||
folder_path.mkdir(parents=True, exist_ok=True)
|
|
||||||
|
|
||||||
# config.json
|
|
||||||
with open(folder_path / 'config.json', 'w', encoding='utf-8') as f:
|
|
||||||
json.dump(data, f, ensure_ascii=False, indent=2)
|
|
||||||
|
|
||||||
return jsonify(data)
|
|
||||||
|
|
||||||
except Exception as e:
|
|
||||||
import traceback
|
|
||||||
return jsonify({'error': str(e), 'trace': traceback.format_exc()}), 500
|
|
||||||
|
|
||||||
|
|
||||||
@app.route('/api/doc-types/<type_id>', methods=['DELETE'])
|
|
||||||
def delete_doc_type(type_id):
|
|
||||||
"""臾몄
|
|
||||||
"""
|
|
||||||
try:
|
|
||||||
folder_path = DOC_TYPES_USER / type_id
|
|
||||||
|
|
||||||
if not folder_path.exists():
|
|
||||||
return jsonify({'error': '臾몄
|
|
||||||
|
|
||||||
李얠
|
|
||||||
듬'}), 404
|
|
||||||
|
|
||||||
shutil.rmtree(folder_path)
|
|
||||||
return jsonify({'success': True, 'deleted': type_id})
|
|
||||||
|
|
||||||
except Exception as e:
|
|
||||||
import traceback
|
|
||||||
return jsonify({'error': str(e), 'trace': traceback.format_exc()}), 500
|
|
||||||
|
|
||||||
# ==============
|
|
||||||
濡 API ==============
|
|
||||||
|
|
||||||
@app.route('/api/upload-files', methods=['POST'])
|
|
||||||
def upload_files():
|
|
||||||
"""
|
|
||||||
濡 /tmp/{session_id}/input/ """
|
|
||||||
try:
|
|
||||||
import uuid
|
|
||||||
|
|
||||||
files = request.files.getlist('files')
|
|
||||||
if not files:
|
|
||||||
return jsonify({'error': '듬'}), 400
|
|
||||||
|
|
||||||
#
|
|
||||||
몄
|
|
||||||
|
|
||||||
|
|
||||||
session_id = str(uuid.uuid4())
|
|
||||||
input_dir = f'/tmp/{session_id}/input'
|
|
||||||
os.makedirs(input_dir, exist_ok=True)
|
|
||||||
|
|
||||||
saved = []
|
|
||||||
for f in files:
|
|
||||||
if f.filename:
|
|
||||||
# 湲
|
|
||||||
|
|
||||||
|
|
||||||
쇰 蹂 蹂寃 紐⑤ - step1~2 異異
|
|
||||||
GPD
|
|
||||||
由우ㅽ"""
|
|
||||||
try:
|
|
||||||
data = request.get_json() or {}
|
|
||||||
session_id = data.get('session_id', '')
|
|
||||||
|
|
||||||
if not session_id:
|
|
||||||
return jsonify({'error': 'session_id媛 듬'}), 400
|
|
||||||
|
|
||||||
input_dir = f'/tmp/{session_id}/input'
|
|
||||||
if not os.path.exists(input_dir):
|
|
||||||
return jsonify({'error': '
|
|
||||||
濡
|
|
||||||
李얠
|
|
||||||
듬'}), 400
|
|
||||||
|
|
||||||
# step_format (洹)
|
|
||||||
from converters.pipeline.step_format import run_format_only
|
|
||||||
result = run_format_only(session_id, input_dir)
|
|
||||||
|
|
||||||
if not result.get('success'):
|
|
||||||
return jsonify(result), 500
|
|
||||||
|
|
||||||
return jsonify(result)
|
|
||||||
|
|
||||||
except Exception as e:
|
|
||||||
import traceback
|
|
||||||
return jsonify({'error': str(e), 'trace': traceback.format_exc()}), 500
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
# ==============
|
|
||||||
API ==============
|
|
||||||
|
|
||||||
@app.route('/generate', methods=['POST'])
|
|
||||||
def generate():
|
|
||||||
"""臾몄
|
|
||||||
|
|
||||||
API"""
|
|
||||||
try:
|
|
||||||
content = ""
|
|
||||||
if 'file' in request.files and request.files['file'].filename:
|
|
||||||
file = request.files['file']
|
|
||||||
content = file.read().decode('utf-8')
|
|
||||||
elif 'content' in request.form:
|
|
||||||
content = request.form.get('content', '')
|
|
||||||
|
|
||||||
doc_type = request.form.get('doc_type', 'briefing')
|
|
||||||
|
|
||||||
if doc_type.startswith('user_'):
|
|
||||||
options = {
|
|
||||||
'instruction': request.form.get('instruction', '')
|
|
||||||
}
|
|
||||||
result = processors['custom'].generate(content, doc_type, options)
|
|
||||||
else:
|
|
||||||
options = {
|
|
||||||
'page_option': request.form.get('page_option', '1'),
|
|
||||||
'department': request.form.get('department', ''),
|
|
||||||
'instruction': request.form.get('instruction', '')
|
|
||||||
}
|
|
||||||
|
|
||||||
processor = processors.get(doc_type, processors['briefing'])
|
|
||||||
result = processor.generate(content, options)
|
|
||||||
|
|
||||||
if 'error' in result:
|
|
||||||
return jsonify(result), 400 if 'trace' not in result else 500
|
|
||||||
return jsonify(result)
|
|
||||||
|
|
||||||
except Exception as e:
|
|
||||||
import traceback
|
|
||||||
return jsonify({'error': str(e), 'trace': traceback.format_exc()}), 500
|
|
||||||
|
|
||||||
|
|
||||||
@app.route('/generate-report', methods=['POST'])
|
|
||||||
def generate_report():
|
|
||||||
"""蹂닿
|
|
||||||
|
|
||||||
API"""
|
|
||||||
try:
|
|
||||||
data = request.get_json() or {}
|
|
||||||
content = data.get('content', '')
|
|
||||||
|
|
||||||
options = {
|
|
||||||
'folder_path': data.get('folder_path', ''),
|
|
||||||
'cover': data.get('cover', False),
|
|
||||||
'toc': data.get('toc', False),
|
|
||||||
'divider': data.get('divider', False),
|
|
||||||
'instruction': data.get('instruction', ''),
|
|
||||||
'template_id': data.get('template_id')
|
|
||||||
}
|
|
||||||
|
|
||||||
result = processors['report'].generate(content, options)
|
|
||||||
|
|
||||||
if 'error' in result:
|
|
||||||
return jsonify(result), 500
|
|
||||||
return jsonify(result)
|
|
||||||
|
|
||||||
except Exception as e:
|
|
||||||
import traceback
|
|
||||||
return jsonify({'error': str(e), 'trace': traceback.format_exc()}), 500
|
|
||||||
|
|
||||||
|
|
||||||
# ============== API ==============
|
|
||||||
|
|
||||||
@app.route('/refine', methods=['POST'])
|
|
||||||
def refine():
|
|
||||||
"""쇰諛 諛 API"""
|
|
||||||
try:
|
|
||||||
feedback = request.json.get('feedback', '')
|
|
||||||
current_html = request.json.get('current_html', '') or session.get('current_html', '')
|
|
||||||
original_html = session.get('original_html', '')
|
|
||||||
doc_type = request.json.get('doc_type', 'briefing')
|
|
||||||
|
|
||||||
processor = processors.get(doc_type, processors['briefing'])
|
|
||||||
result = processor.refine(feedback, current_html, original_html)
|
|
||||||
|
|
||||||
if 'error' in result:
|
|
||||||
return jsonify(result), 400
|
|
||||||
return jsonify(result)
|
|
||||||
|
|
||||||
except Exception as e:
|
|
||||||
return jsonify({'error': str(e)}), 500
|
|
||||||
|
|
||||||
|
|
||||||
@app.route('/refine-selection', methods=['POST'])
|
|
||||||
def refine_selection():
|
|
||||||
"""
|
|
||||||
遺
|
|
||||||
API"""
|
|
||||||
try:
|
|
||||||
data = request.json
|
|
||||||
current_html = data.get('current_html', '')
|
|
||||||
selected_text = data.get('selected_text', '')
|
|
||||||
user_request = data.get('request', '')
|
|
||||||
doc_type = data.get('doc_type', 'briefing')
|
|
||||||
|
|
||||||
processor = processors.get(doc_type, processors['briefing'])
|
|
||||||
result = processor.refine_selection(current_html, selected_text, user_request)
|
|
||||||
|
|
||||||
if 'error' in result:
|
|
||||||
return jsonify(result), 400
|
|
||||||
return jsonify(result)
|
|
||||||
|
|
||||||
except Exception as e:
|
|
||||||
return jsonify({'error': str(e)}), 500
|
|
||||||
|
|
||||||
|
|
||||||
# ============== ㅼ대 API ==============
|
|
||||||
|
|
||||||
@app.route('/download/html', methods=['POST'])
|
|
||||||
def download_html():
|
|
||||||
"""HTML ㅼ대"""
|
|
||||||
html_content = request.form.get('html', '')
|
|
||||||
if not html_content:
|
|
||||||
return "No content", 400
|
|
||||||
|
|
||||||
timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
|
|
||||||
filename = f'report_{timestamp}.html'
|
|
||||||
|
|
||||||
return Response(
|
|
||||||
html_content,
|
|
||||||
mimetype='text/html',
|
|
||||||
headers={'Content-Disposition': f'attachment; filename={filename}'}
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
@app.route('/download/pdf', methods=['POST'])
|
|
||||||
def download_pdf():
|
|
||||||
"""PDF ㅼ대"""
|
|
||||||
try:
|
|
||||||
from weasyprint import HTML
|
|
||||||
|
|
||||||
html_content = request.form.get('html', '')
|
|
||||||
if not html_content:
|
|
||||||
return "No content", 400
|
|
||||||
|
|
||||||
pdf_buffer = io.BytesIO()
|
|
||||||
HTML(string=html_content).write_pdf(pdf_buffer)
|
|
||||||
pdf_buffer.seek(0)
|
|
||||||
|
|
||||||
timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
|
|
||||||
filename = f'report_{timestamp}.pdf'
|
|
||||||
|
|
||||||
return Response(
|
|
||||||
pdf_buffer.getvalue(),
|
|
||||||
mimetype='application/pdf',
|
|
||||||
headers={'Content-Disposition': f'attachment; filename={filename}'}
|
|
||||||
)
|
|
||||||
except ImportError:
|
|
||||||
return jsonify({'error': 'PDF 蹂몄고 API ==============
|
|
||||||
|
|
||||||
@app.route('/assets/<path:filename>')
|
|
||||||
def serve_assets(filename):
|
|
||||||
"""濡而 assets 대
|
|
||||||
鍮"""
|
|
||||||
assets_dir = os.environ.get("ASSETS_BASE_PATH", "/tmp/assets")
|
|
||||||
return send_file(os.path.join(assets_dir, filename))
|
|
||||||
|
|
||||||
|
|
||||||
@app.route('/hwp-script')
|
|
||||||
def hwp_script():
|
|
||||||
"""HWP 蹂ㅽщ┰ """
|
|
||||||
return render_template('hwp_guide.html')
|
|
||||||
|
|
||||||
|
|
||||||
@app.route('/health')
|
|
||||||
def health():
|
|
||||||
"""ъ 泥댄"""
|
|
||||||
return jsonify({'status': 'healthy', 'version': '2.0.0'})
|
|
||||||
|
|
||||||
|
|
||||||
@app.route('/export-hwp', methods=['POST'])
|
|
||||||
def export_hwp():
|
|
||||||
"""HWP 蹂ㅽ 吏
|
|
||||||
댁⑹듬'}), 400
|
|
||||||
|
|
||||||
temp_dir = tempfile.gettempdir()
|
|
||||||
html_path = os.path.join(temp_dir, 'geulbeot_temp.html')
|
|
||||||
hwp_path = os.path.join(temp_dir, 'geulbeot_output.hwp')
|
|
||||||
|
|
||||||
with open(html_path, 'w', encoding='utf-8') as f:
|
|
||||||
f.write(html_content)
|
|
||||||
|
|
||||||
# 蹂
|
|
||||||
ㅽ ъ щ
|
|
||||||
if use_style_grouping:
|
|
||||||
final_path = converter.convert_with_styles(html_path, hwp_path)
|
|
||||||
# HWPX
|
|
||||||
|
|
||||||
return send_file(
|
|
||||||
final_path,
|
|
||||||
as_attachment=True,
|
|
||||||
download_name=f'report_{datetime.now().strftime("%Y%m%d_%H%M%S")}.hwpx',
|
|
||||||
mimetype='application/vnd.hancom.hwpx'
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
converter.convert(html_path, hwp_path)
|
|
||||||
return send_file(
|
|
||||||
hwp_path,
|
|
||||||
as_attachment=True,
|
|
||||||
download_name=f'report_{datetime.now().strftime("%Y%m%d_%H%M%S")}.hwp',
|
|
||||||
mimetype='application/x-hwp'
|
|
||||||
)
|
|
||||||
|
|
||||||
except ImportError as e:
|
|
||||||
return jsonify({'error': f'pyhwpx
|
|
||||||
: {str(e)}'}), 500
|
|
||||||
except Exception as e:
|
|
||||||
return jsonify({'error': str(e)}), 500
|
|
||||||
|
|
||||||
|
|
||||||
# 湲곗〈 add_doc_type 泥
|
|
||||||
@app.route('/api/doc-types/analyze', methods=['POST'])
|
|
||||||
def analyze_doc_type():
|
|
||||||
"""臾몄
|
|
||||||
遺
|
|
||||||
|
|
||||||
|
|
||||||
⑸"}), 400
|
|
||||||
|
|
||||||
file = request.files['file']
|
|
||||||
doc_name = request.form.get('name', ' 臾몄
|
|
||||||
')
|
|
||||||
|
|
||||||
#
|
|
||||||
|
|
||||||
import tempfile
|
|
||||||
temp_path = os.path.join(tempfile.gettempdir(), file.filename)
|
|
||||||
file.save(temp_path)
|
|
||||||
|
|
||||||
try:
|
|
||||||
analyzer = DocTypeAnalyzer()
|
|
||||||
result = analyzer.analyze(temp_path, doc_name)
|
|
||||||
|
|
||||||
return jsonify({
|
|
||||||
"success": True,
|
|
||||||
"config": result["config"],
|
|
||||||
"summary": {
|
|
||||||
"pageCount": result["structure"]["pageCount"],
|
|
||||||
"sections": len(result["toc"]),
|
|
||||||
except Exception as e:
|
|
||||||
return jsonify({"error": str(e)}), 500
|
|
||||||
finally:
|
|
||||||
os.remove(temp_path)
|
|
||||||
|
|
||||||
|
|
||||||
@app.route('/analyze-styles', methods=['POST'])
|
|
||||||
def analyze_styles():
|
|
||||||
"""HTML ㅽ
|
|
||||||
@app.route('/anayze-styles', methods=['POST'])
|
|
||||||
def analyze_style
|
|
||||||
try:
|
|
||||||
data = request.get_json()
|
|
||||||
html_content = data.get('html', '')
|
|
||||||
|
|
||||||
if not html_content:
|
|
||||||
return jsonify({'error': 'HTML 댁⑹듬'}), 400
|
|
||||||
|
|
||||||
from converters.style_analyzer import StyleAnalyzer
|
|
||||||
from converters.hwp_style_mapping import ROLE_TO_STYLE_NAME
|
|
||||||
|
|
||||||
analyzer = StyleAnalyzer()
|
|
||||||
elements = analyzer.analyze(html_content)
|
|
||||||
|
|
||||||
# 蹂
|
|
||||||
summary = analyzer.get_role_summary()
|
|
||||||
|
|
||||||
# elem in elements[:50]:
|
|
||||||
蹂 (泥 50媛留)
|
|
||||||
details = []
|
|
||||||
for elem in elements[:50]:
|
|
||||||
details.append({
|
|
||||||
'role': elem.role,
|
|
||||||
'hwp_style': ROLE_TO_STYLE_NAME.get(elem.role, '諛湲'),
|
|
||||||
'text': elem.text[:50] + ('...' if len(elem.text) > 50 else ''),
|
|
||||||
'section': elem.section
|
|
||||||
})
|
|
||||||
|
|
||||||
return jsonify({
|
|
||||||
'total_elements': len(elements),
|
|
||||||
'summary': summary,
|
|
||||||
'details': details
|
|
||||||
})
|
|
||||||
|
|
||||||
except Exception as e:
|
|
||||||
import traceback
|
|
||||||
return jsonify({'error': str(e), 'trace': traceback.format_exc()}), 500
|
|
||||||
|
|
||||||
@app.route('/templates', methods=['GET'])
|
|
||||||
def get_templates():
|
|
||||||
"""
|
|
||||||
由 紐⑸"""
|
|
||||||
try:
|
|
||||||
templates = template_mgr.list_templates()
|
|
||||||
return jsonify(templates)
|
|
||||||
except Exception as e:
|
|
||||||
return jsonify({'error': str(e)}), 500
|
|
||||||
|
|
||||||
@app.route('/api/templates', methods=['GET'])
|
|
||||||
def get_templates_api():
|
|
||||||
"""
|
|
||||||
由 紐⑸ (API 寃쎈)"""
|
|
||||||
try:
|
|
||||||
templates = template_mgr.list_templates()
|
|
||||||
return jsonify(templates)
|
|
||||||
except Exception as e:
|
|
||||||
return jsonify({'error': str(e)}), 500
|
|
||||||
|
|
||||||
|
|
||||||
@app.route('/analyze-template', methods=['POST'])
|
|
||||||
def analyze_template():
|
|
||||||
"""
|
|
||||||
由 異異 諛 (doc_template_analyzer template_manager)"""
|
|
||||||
try:
|
|
||||||
if 'file' not in request.files:
|
|
||||||
return jsonify({'error': '듬'}), 400
|
|
||||||
|
|
||||||
file = request.files['file']
|
|
||||||
name = request.form.get('name', '').strip()
|
|
||||||
|
|
||||||
if not name:
|
|
||||||
return jsonify({'error': '
|
|
||||||
由
|
|
||||||
|
|
||||||
|
|
||||||
ν댁＜
|
|
||||||
몄'}), 400
|
|
||||||
|
|
||||||
if not file.filename:
|
|
||||||
return jsonify({'error': '
|
|
||||||
|
|
||||||
댁＜
|
|
||||||
몄'}), 400
|
|
||||||
|
|
||||||
#
|
|
||||||
HWPX
|
|
||||||
由 異異
|
|
||||||
temp_dir = tempfile.gettempdir()
|
|
||||||
temp_path = os.path.join(temp_dir, file.filename)
|
|
||||||
file.save(temp_path)
|
|
||||||
|
|
||||||
try:
|
|
||||||
# v3
|
|
||||||
ъ (HWPX parsed dict)
|
|
||||||
from handlers.doc.doc_type_analyzer import DocTypeAnalyzer
|
|
||||||
parser = DocTypeAnalyzer()
|
|
||||||
parsed = parser._parse_hwpx(temp_path)
|
|
||||||
|
|
||||||
# template_manager濡 異異+
|
|
||||||
result = template_mgr.extract_and_save(
|
|
||||||
parsed, name,
|
|
||||||
source_file=file.filename
|
|
||||||
)
|
|
||||||
|
|
||||||
return jsonify(result)
|
|
||||||
finally:
|
|
||||||
try:
|
|
||||||
os.remove(temp_path)
|
|
||||||
except:
|
|
||||||
pass
|
|
||||||
|
|
||||||
except Exception as e:
|
|
||||||
import traceback
|
|
||||||
return jsonify({'error': str(e), 'trace': traceback.format_exc()}), 500
|
|
||||||
|
|
||||||
# ============== 臾몄
|
|
||||||
遺
|
|
||||||
|
|
||||||
|
|
||||||
遺
|
|
||||||
|
|
||||||
ㅽ몃━諛)
|
|
||||||
ㅼ媛
|
|
||||||
쇰 媛 ④
|
|
||||||
|
|
||||||
⑹
|
|
||||||
|
|
||||||
|
|
||||||
"""
|
|
||||||
import tempfile
|
|
||||||
|
|
||||||
# 곗
|
|
||||||
寃利듬'}), 400
|
|
||||||
|
|
||||||
file = request.files['file']
|
|
||||||
name = request.form.get('name', '').strip()
|
|
||||||
description = request.form.get('description', '').strip()
|
|
||||||
|
|
||||||
if not name:
|
|
||||||
return jsonify({'error': '臾몄
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
ν댁＜
|
|
||||||
몄'}), 400
|
|
||||||
|
|
||||||
if not file.filename:
|
|
||||||
return jsonify({'error': '
|
|
||||||
|
|
||||||
댁＜
|
|
||||||
몄'}), 400
|
|
||||||
|
|
||||||
#
|
|
||||||
|
|
||||||
temp_dir = tempfile.gettempdir()
|
|
||||||
temp_path = os.path.join(temp_dir, file.filename)
|
|
||||||
file.save(temp_path)
|
|
||||||
|
|
||||||
# 硫吏
|
|
||||||
|
|
||||||
message_queue = queue.Queue()
|
|
||||||
analysis_result = {"data": None, "error": None}
|
|
||||||
|
|
||||||
def progress_callback(step_id, status, message):
|
|
||||||
"""吏
|
|
||||||
肄諛 - 硫吏 媛"""
|
|
||||||
message_queue.put({
|
|
||||||
"type": "progress",
|
|
||||||
"step": step_id,
|
|
||||||
"status": status,
|
|
||||||
"message": message
|
|
||||||
})
|
|
||||||
|
|
||||||
def run_analysis():
|
|
||||||
"""遺
|
|
||||||
|
|
||||||
ㅽ (蹂
|
|
||||||
|
|
||||||
ㅻ)"""
|
|
||||||
try:
|
|
||||||
|
|
||||||
analyzer = DocTypeAnalyzer(progress_callback=progress_callback)
|
|
||||||
result = analyzer.analyze(temp_path, name, description)
|
|
||||||
|
|
||||||
#
|
|
||||||
save_path = analyzer.save_doc_type(result["config"], result.get("template", "") )
|
|
||||||
|
|
||||||
analysis_result["data"] = {
|
|
||||||
"success": True,
|
|
||||||
"config": result["config"],
|
|
||||||
"layout": result.get("layout", {}),
|
|
||||||
"context": result.get("context", {}),
|
|
||||||
"structure": result.get("structure", {}),
|
|
||||||
"template_generated": bool(result.get("template_id") or result.get("template")),
|
|
||||||
"template_id": result.get("template_id"), #
|
|
||||||
異媛
|
|
||||||
"saved_path": save_path
|
|
||||||
}
|
|
||||||
|
|
||||||
except Exception as e:
|
|
||||||
import traceback
|
|
||||||
analysis_result["error"] = {
|
|
||||||
"message": str(e),
|
|
||||||
"trace": traceback.format_exc()
|
|
||||||
}
|
|
||||||
finally:
|
|
||||||
#
|
|
||||||
猷
|
|
||||||
message_queue.put({"type": "complete"})
|
|
||||||
#
|
|
||||||
|
|
||||||
try:
|
|
||||||
os.remove(temp_path)
|
|
||||||
except:
|
|
||||||
pass
|
|
||||||
|
|
||||||
def generate_events():
|
|
||||||
"""SSE
|
|
||||||
깃린"""
|
|
||||||
# 遺
|
|
||||||
|
|
||||||
|
|
||||||
analysis_thread = threading.Thread(target=run_analysis)
|
|
||||||
analysis_thread.start()
|
|
||||||
|
|
||||||
# ㅽ몃━諛
|
|
||||||
while True:
|
|
||||||
try:
|
|
||||||
msg = message_queue.get(timeout=60) # 60珥
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
if msg["type"] == "complete":
|
|
||||||
# 遺
|
|
||||||
|
|
||||||
|
|
||||||
猷
|
|
||||||
if analysis_result["error"]:
|
|
||||||
yield f"data: {json.dumps({'type': 'error', 'error': analysis_result['error']}, ensure_ascii=False)}\n\n"
|
|
||||||
else:
|
|
||||||
yield f"data: {json.dumps({'type': 'result', 'data': analysis_result['data']}, ensure_ascii=False)}\n\n"
|
|
||||||
break
|
|
||||||
else:
|
|
||||||
# 吏
|
|
||||||
|
|
||||||
yield f"data: {json.dumps(msg, ensure_ascii=False)}\n\n"
|
|
||||||
|
|
||||||
except queue.Empty:
|
|
||||||
#
|
|
||||||
|
|
||||||
|
|
||||||
yield f"data: {json.dumps({'type': 'error', 'error': {'message': '遺
|
|
||||||
|
|
||||||
媛
|
|
||||||
珥怨'}}, ensure_ascii=False)}\n\n"
|
|
||||||
break
|
|
||||||
|
|
||||||
return Response(
|
|
||||||
generate_events(),
|
|
||||||
mimetype='text/event-stream',
|
|
||||||
headers={
|
|
||||||
'Cache-Control': 'no-cache',
|
|
||||||
'Connection': 'keep-alive',
|
|
||||||
'X-Accel-Buffering': 'no'
|
|
||||||
}
|
|
||||||
)
|
|
||||||
|
|
||||||
@app.route('/delete-template/<template_id>', methods=['DELETE'])
|
|
||||||
def delete_template(template_id):
|
|
||||||
"""
|
|
||||||
由 (嫄곗 뱀
|
|
||||||
由 議고"""
|
|
||||||
try:
|
|
||||||
result = template_mgr.load_template(tpl_id)
|
|
||||||
if 'error' in result:
|
|
||||||
return jsonify(result), 404
|
|
||||||
return jsonify(result)
|
|
||||||
except Exception as e:
|
|
||||||
return jsonify({'error': str(e)}), 500
|
|
||||||
|
|
||||||
|
|
||||||
@app.route('/api/templates/<tpl_id>', methods=['DELETE'])
|
|
||||||
def delete_template_new(tpl_id):
|
|
||||||
"""
|
|
||||||
由 """
|
|
||||||
try:
|
|
||||||
result = template_mgr.delete_template(tpl_id)
|
|
||||||
if 'error' in result:
|
|
||||||
return jsonify(result), 400
|
|
||||||
return jsonify(result)
|
|
||||||
except Exception as e:
|
|
||||||
return jsonify({'error': str(e)}), 500
|
|
||||||
|
|
||||||
|
|
||||||
@app.route('/api/doc-types/<type_id>/template', methods=['PUT'])
|
|
||||||
def change_doc_type_template(type_id):
|
|
||||||
"""臾몄
|
|
||||||
|
|
||||||
由 援"""
|
|
||||||
try:
|
|
||||||
data = request.get_json()
|
|
||||||
new_tpl_id = data.get('template_id')
|
|
||||||
|
|
||||||
if not new_tpl_id:
|
|
||||||
return jsonify({'error': 'template_id媛
|
|
||||||
⑸'}), 400
|
|
||||||
|
|
||||||
result = template_mgr.change_template(type_id, new_tpl_id)
|
|
||||||
if 'error' in result:
|
|
||||||
return jsonify(result), 400
|
|
||||||
return jsonify(result)
|
|
||||||
except Exception as e:
|
|
||||||
return jsonify({'error': str(e)}), 500
|
|
||||||
|
|
||||||
|
|
||||||
@app.route('/api/doc-types/<type_id>/template', methods=['GET'])
|
|
||||||
def get_doc_type_template(type_id):
|
|
||||||
"""臾몄
|
|
||||||
곌껐
|
|
||||||
由 議고"""
|
|
||||||
try:
|
|
||||||
result = template_mgr.get_template_for_doctype(type_id)
|
|
||||||
if 'error' in result:
|
|
||||||
return jsonify(result), 404
|
|
||||||
return jsonify(result)
|
|
||||||
except Exception as e:
|
|
||||||
return jsonify({'error': str(e)}), 500
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
|
||||||
port = int(os.environ.get('PORT', 5000))
|
|
||||||
debug = os.environ.get('FLASK_DEBUG', 'False').lower() == 'true'
|
|
||||||
app.run(host='0.0.0.0', port=port, debug=debug)
|
|
||||||
Reference in New Issue
Block a user