diff --git a/backend/app/api/routes.py b/backend/app/api/routes.py index a24f709..1b8c337 100644 --- a/backend/app/api/routes.py +++ b/backend/app/api/routes.py @@ -1,5 +1,5 @@ from fastapi import APIRouter, Depends, HTTPException, BackgroundTasks -from fastapi.responses import HTMLResponse +from fastapi.responses import HTMLResponse, FileResponse from sqlalchemy.ext.asyncio import AsyncSession from sqlalchemy import select from sqlalchemy.orm import selectinload @@ -9,6 +9,11 @@ from app.models import Report, Setting from app.services import analysis_service import os import markdown +from weasyprint import HTML +import io +import tempfile +from urllib.parse import quote +from bs4 import BeautifulSoup router = APIRouter() @@ -134,6 +139,26 @@ async def get_report_html(report_id: int, db: AsyncSession = Depends(get_db)): {financial_html} + """ @@ -167,3 +192,332 @@ async def update_config(request: ConfigUpdateRequest, db: AsyncSession = Depends await db.commit() return {"status": "updated", "key": request.key} + +@router.get("/reports/{report_id}/pdf") +async def download_report_pdf(report_id: int, db: AsyncSession = Depends(get_db)): + result = await db.execute(select(Report).options(selectinload(Report.sections)).where(Report.id == report_id)) + report = result.scalar_one_or_none() + if not report: + raise HTTPException(status_code=404, detail="Report not found") + + # Get Financial HTML (Charts) + root_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../../")) + base_dir = os.path.join(root_dir, "data", report.market) + symbol_dir = os.path.join(base_dir, report.symbol) + + # Fuzzy match logic + financial_html = "" + try: + if not os.path.exists(symbol_dir) and os.path.exists(base_dir): + candidates = [d for d in os.listdir(base_dir) if d.startswith(report.symbol) and os.path.isdir(os.path.join(base_dir, d))] + if candidates: + symbol_dir = os.path.join(base_dir, candidates[0]) + + start_html_path = os.path.join(symbol_dir, "report.html") + if os.path.exists(start_html_path): + with open(start_html_path, 'r', encoding='utf-8') as f: + financial_html = f.read() + + # Parse and clean HTML - keep table styles but remove containers and backgrounds + soup = BeautifulSoup(financial_html, 'html.parser') + + # Modify style tags to remove background colors + for style_tag in soup.find_all('style'): + style_content = style_tag.string + if style_content: + # Remove CSS variables for backgrounds + style_content = '\n'.join([line for line in style_content.split('\n') + if not any(bg in line.lower() for bg in ['--bg:', '--card-bg:', '--header-bg:', '--section-bg:'])]) + # Remove background properties from body + style_content = style_content.replace('background: var(--bg);', 'background: white;') + style_content = style_content.replace('background: var(--card-bg);', '') + style_content = style_content.replace('background: var(--header-bg);', '') + style_tag.string = style_content + + # Remove container divs but keep content + for div in soup.find_all('div', class_='report-container'): + div.unwrap() + + # Remove inline styles from container divs only, preserve td/th styles (including background colors) + for div in soup.find_all('div'): + if div.get('style'): + del div['style'] + + # Limit table columns to first 6 + for table in soup.find_all('table'): + for row in table.find_all('tr'): + cells = row.find_all(['th', 'td']) + if len(cells) > 6: + for cell in cells[6:]: + cell.decompose() + + financial_html = str(soup) + else: + financial_html = "

财务图表尚未生成，数据获取可能仍在进行中。

" + except Exception as e: + financial_html = f"

加载财务图表时出错: {str(e)}

" + + # Build analysis sections HTML + sections_html = "" + section_names = { + 'company_profile': '公司简介', + 'fundamental_analysis': '基本面分析', + 'insider_analysis': '内部人士分析', + 'bullish_analysis': '看涨分析', + 'bearish_analysis': '看跌分析' + } + + + import re + for section in sorted(report.sections, key=lambda s: list(section_names.keys()).index(s.section_name) if s.section_name in section_names else 999): + # Pre-process markdown to fix list formatting + content = section.content + + # 1. First, find cases where numbered lists are clumped in one line like "1. xxx 2. yyy" + # and split them into multiple lines. + content = re.sub(r'(\s)(\d+\.\s)', r'\n\2', content) + + # 2. Ensure list items have proper spacing + lines = content.split('\n') + fixed_lines = [] + in_list = False + + for i, line in enumerate(lines): + stripped = line.strip() + # Check if this line is a list item (bullet or numbered) + # Regex \d+\.\s matches "1. ", "2. ", etc. + is_list_item = stripped.startswith('* ') or stripped.startswith('- ') or re.match(r'^\d+\.\s', stripped) + + if is_list_item: + # Add blank line before first list item + if not in_list and fixed_lines and fixed_lines[-1].strip(): + fixed_lines.append('') + in_list = True + fixed_lines.append(line) + else: + # Add blank line after last list item + if in_list and stripped: + fixed_lines.append('') + in_list = False + fixed_lines.append(line) + + content = '\n'.join(fixed_lines) + + section_content = markdown.markdown(content, extensions=['tables', 'fenced_code']) + sections_html += f""" +
+
+ {section_content} +
+
+ """ + + # Complete PDF HTML + complete_html = f""" + + + + + {report.company_name} - 分析报告 + + + +
+

{report.company_name}

+
{report.market} {report.symbol}
+
分析日期: {report.created_at.strftime('%Y年%m月%d日')}
+ {f'
AI模型: {report.ai_model}
' if report.ai_model else ''} +
+ +
+
+ {financial_html} +
+
+ + {sections_html} + + + """ + + # Generate PDF + try: + # Create a temporary file for the PDF + with tempfile.NamedTemporaryFile(delete=False, suffix='.pdf') as tmp_file: + pdf_path = tmp_file.name + HTML(string=complete_html).write_pdf(pdf_path) + + # Return the PDF file + filename = f"{report.company_name}_{report.symbol}_分析报告.pdf" + # Use RFC 5987 encoding for non-ASCII filenames + filename_encoded = quote(filename) + + return FileResponse( + path=pdf_path, + media_type='application/pdf', + headers={ + 'Content-Disposition': f"attachment; filename*=UTF-8''{filename_encoded}" + } + ) + except Exception as e: + raise HTTPException(status_code=500, detail=f"生成PDF失败: {str(e)}") diff --git a/backend/requirements.txt b/backend/requirements.txt index d61fa32..abe1522 100644 --- a/backend/requirements.txt +++ b/backend/requirements.txt @@ -13,3 +13,4 @@ pandas numpy pytest httpx +weasyprint diff --git a/frontend/src/app/analysis/[id]/page.tsx b/frontend/src/app/analysis/[id]/page.tsx index 7500619..080c745 100644 --- a/frontend/src/app/analysis/[id]/page.tsx +++ b/frontend/src/app/analysis/[id]/page.tsx @@ -1,12 +1,12 @@ "use client" -import { useEffect, useState, use } from "react" +import { useEffect, useState, use, useRef } from "react" import { getReport } from "@/lib/api" import { Badge } from "@/components/ui/badge" import { Card, CardContent } from "@/components/ui/card" import { Tabs, TabsContent, TabsList, TabsTrigger } from "@/components/ui/tabs" import { MarkdownRenderer } from "@/components/markdown-renderer" -import { Loader2, RefreshCcw, ExternalLink } from "lucide-react" +import { Loader2, RefreshCcw, ExternalLink, Download } from "lucide-react" import { Button } from "@/components/ui/button" export default function AnalysisPage({ params }: { params: Promise<{ id: string }> }) { @@ -22,6 +22,8 @@ export default function AnalysisPage({ params }: { params: Promise<{ id: string const [report, setReport] = useState(null) const [loading, setLoading] = useState(true) + const iframeRef = useRef(null) + const [iframeHeight, setIframeHeight] = useState(800) // Polling logic useEffect(() => { @@ -49,6 +51,41 @@ export default function AnalysisPage({ params }: { params: Promise<{ id: string return () => clearTimeout(interval) }, [reportId]) + // Listen for height messages from iframe + useEffect(() => { + const handleMessage = (event: MessageEvent) => { + // Accept messages from localhost (for development) + if (event.data.type === 'resize' && typeof event.data.height === 'number') { + setIframeHeight(event.data.height + 20) // Add some padding + } + } + + window.addEventListener('message', handleMessage) + return () => window.removeEventListener('message', handleMessage) + }, []) + + // Handle PDF download + const handleDownloadPDF = async () => { + try { + const response = await fetch(`http://localhost:8000/api/reports/${report.id}/pdf`) + if (!response.ok) { + throw new Error('下载失败') + } + const blob = await response.blob() + const url = window.URL.createObjectURL(blob) + const a = document.createElement('a') + a.href = url + a.download = `${report.company_name}_${report.symbol}_分析报告.pdf` + document.body.appendChild(a) + a.click() + window.URL.revokeObjectURL(url) + document.body.removeChild(a) + } catch (error) { + console.error('PDF下载失败:', error) + alert('PDF下载失败，请稍后重试') + } + } + if (loading && !report) return
if (!report) return
未找到报告。
@@ -75,6 +112,12 @@ export default function AnalysisPage({ params }: { params: Promise<{ id: string ) : report.status === "completed" ? "已完成" : report.status === "failed" ? "失败" : report.status === "pending" ? "待处理" : report.status} + {report.status === "completed" && ( + + )} @@ -99,8 +142,9 @@ export default function AnalysisPage({ params }: { params: Promise<{ id: string - +