106 lines
3.3 KiB
Python
106 lines
3.3 KiB
Python
"""
|
|
分析结果图表 API - 根据文本分析结果生成图表
|
|
"""
|
|
from fastapi import APIRouter, HTTPException
|
|
from pydantic import BaseModel
|
|
from typing import Optional
|
|
import logging
|
|
|
|
from app.services.text_analysis_service import text_analysis_service
|
|
from app.services.chart_generator_service import chart_generator_service
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
router = APIRouter(prefix="/analysis", tags=["分析结果图表"])
|
|
|
|
|
|
class AnalysisChartRequest(BaseModel):
|
|
"""分析图表生成请求模型"""
|
|
analysis_text: str
|
|
original_filename: Optional[str] = ""
|
|
file_type: Optional[str] = "text"
|
|
|
|
|
|
@router.post("/extract-and-chart")
|
|
async def extract_and_generate_charts(request: AnalysisChartRequest):
|
|
"""
|
|
从 AI 分析结果中提取数据并生成图表
|
|
|
|
Args:
|
|
request: 包含分析文本的请求
|
|
|
|
Returns:
|
|
dict: 包含图表数据的结果
|
|
"""
|
|
if not request.analysis_text or not request.analysis_text.strip():
|
|
raise HTTPException(status_code=400, detail="分析文本不能为空")
|
|
|
|
try:
|
|
logger.info("开始从分析结果中提取结构化数据...")
|
|
|
|
# 1. 使用 LLM 提取结构化数据
|
|
extract_result = await text_analysis_service.extract_structured_data(
|
|
analysis_text=request.analysis_text,
|
|
original_filename=request.original_filename or "unknown",
|
|
file_type=request.file_type or "text"
|
|
)
|
|
|
|
if not extract_result.get("success"):
|
|
raise HTTPException(
|
|
status_code=500,
|
|
detail=f"提取结构化数据失败: {extract_result.get('error', '未知错误')}"
|
|
)
|
|
|
|
logger.info("结构化数据提取成功，开始生成图表...")
|
|
|
|
# 2. 根据提取的数据生成图表
|
|
chart_result = chart_generator_service.generate_charts_from_analysis(extract_result)
|
|
|
|
if not chart_result.get("success"):
|
|
raise HTTPException(
|
|
status_code=500,
|
|
detail=f"生成图表失败: {chart_result.get('error', '未知错误')}"
|
|
)
|
|
|
|
logger.info("图表生成成功")
|
|
|
|
return chart_result
|
|
|
|
except HTTPException:
|
|
raise
|
|
except Exception as e:
|
|
logger.error(f"分析结果图表生成失败: {str(e)}")
|
|
raise HTTPException(
|
|
status_code=500,
|
|
detail=f"图表生成失败: {str(e)}"
|
|
)
|
|
|
|
|
|
@router.post("/analyze-text")
|
|
async def analyze_text_only(request: AnalysisChartRequest):
|
|
"""
|
|
仅提取结构化数据（不生成图表），用于调试
|
|
|
|
Args:
|
|
request: 包含分析文本的请求
|
|
|
|
Returns:
|
|
dict: 提取的结构化数据
|
|
"""
|
|
if not request.analysis_text or not request.analysis_text.strip():
|
|
raise HTTPException(status_code=400, detail="分析文本不能为空")
|
|
|
|
try:
|
|
result = await text_analysis_service.extract_structured_data(
|
|
analysis_text=request.analysis_text,
|
|
original_filename=request.original_filename or "unknown",
|
|
file_type=request.file_type or "text"
|
|
)
|
|
return result
|
|
except Exception as e:
|
|
logger.error(f"文本分析失败: {str(e)}")
|
|
raise HTTPException(
|
|
status_code=500,
|
|
detail=f"文本分析失败: {str(e)}"
|
|
)
|