Update export/ppt.py

export/ppt.py

@@ -1,43 +1,132 @@
 # export/ppt.py
-from pptx import Presentation
-from typing import Dict, List
-import os
-
-def _add_bullets(prs: Presentation, title: str, body: str):
-    layout = prs.slide_layouts[1]  # Title and Content
-    s = prs.slides.add_slide(layout)
-    s.shapes.title.text = title
-    tf = s.placeholders[1].text_frame
+from __future__ import annotations
+from typing import Dict, List, Tuple
+from datetime import datetime
+
+try:
+    from pptx import Presentation
+    from pptx.util import Inches, Pt
+    from pptx.enum.text import PP_ALIGN
+    from pptx.dml.color import RGBColor
+except ModuleNotFoundError as e:
+    # 呼び出し側でこのエラーをユーザに分かりやすく返すため、敢えてそのまま投げる
+    raise
+
+MAX_BULLETS = 10
+
+def _add_title(prs: Presentation, title: str, subtitle: str | None = None):
+    slide_layout = prs.slide_layouts[0]  # Title Slide
+    slide = prs.slides.add_slide(slide_layout)
+    slide.shapes.title.text = title
+    if subtitle is not None:
+        slide.placeholders[1].text = subtitle
+
+def _add_title_and_bullets(prs: Presentation, title: str, bullets: List[str]):
+    slide_layout = prs.slide_layouts[1]  # Title and Content
+    slide = prs.slides.add_slide(slide_layout)
+    slide.shapes.title.text = title
+    tx = slide.shapes.placeholders[1].text_frame
+    tx.clear()
+    # 先頭段落
+    if not bullets:
+        bullets = ["（該当なし）"]
+    bullets = bullets[:MAX_BULLETS]
+    p = tx.paragraphs[0]
+    p.text = bullets[0]
+    p.level = 0
+    for b in bullets[1:]:
+        rp = tx.add_paragraph()
+        rp.text = b
+        rp.level = 0
+
+def _add_sources(prs: Presentation, links: List[str]):
+    slide_layout = prs.slide_layouts[5]  # Title Only
+    slide = prs.slides.add_slide(slide_layout)
+    slide.shapes.title.text = "根拠リンク / 参考資料"
+    left = Inches(0.8); top = Inches(1.8); width = Inches(9); height = Inches(5)
+    box = slide.shapes.add_textbox(left, top, width, height)
+    tf = box.text_frame
+    tf.word_wrap = True
+    if not links:
+        tf.text = "（なし）"
+        return
     tf.clear()
-    lines = [l.strip("•- 　") for l in (body or "").splitlines() if l.strip()]
-    if not lines: lines = ["（なし）"]
-    first = True
-    for line in lines:
-        if first:
-            p = tf.paragraphs[0]; first = False
-        else:
-            p = tf.add_paragraph()
-        p.text = line
+    for i, url in enumerate(links, 1):
+        p = tf.add_paragraph() if i > 1 else tf.paragraphs[0]
+        p.text = f"{i}. {url}"
         p.level = 0
 
-def build_deck(sections: Dict[str, str], links: List[str]) -> Presentation:
+def _split_lines(s: str, sep: str = "\n") -> List[str]:
+    s = (s or "").strip()
+    if not s:
+        return []
+    out = [ln.strip(" \t\u3000") for ln in s.split(sep)]
+    return [ln for ln in out if ln]
+
+def build_deck(sections: Dict[str, str | List[str]], links: List[str] | None = None) -> Presentation:
+    """
+    sections 例:
+    {
+      "highlights": "• 売上+10%\n• 営業益+5%\n...",
+      "outlook": "...",
+      "segments": "...",
+      "finance": "...",
+      "shareholder": "...",
+      "esg": "...",
+      "risks": "..."
+    }
+    """
     prs = Presentation()
-    title_slide_layout = prs.slide_layouts[0]
-    s0 = prs.slides.add_slide(title_slide_layout)
-    s0.shapes.title.text = "IR/PR サマリー"
-    s0.placeholders[1].text = "自動生成（RAG + OpenAI）"
-
-    _add_bullets(prs, "業績ハイライト", sections.get("highlights",""))
-    _add_bullets(prs, "見通し",       sections.get("outlook",""))
-    _add_bullets(prs, "セグメント",   sections.get("segments",""))
-    _add_bullets(prs, "財務",         sections.get("finance",""))
-    _add_bullets(prs, "株主還元",     sections.get("shareholder",""))
-    _add_bullets(prs, "ESG",          sections.get("esg",""))
-    _add_bullets(prs, "リスク",       sections.get("risks",""))
-    _add_bullets(prs, "参考リンク", "\n".join(links or []))
+
+    # 1) タイトル
+    today = datetime.now().strftime("%Y-%m-%d")
+    _add_title(prs, "IR/PR Co-Pilot Pro 自動生成スライド", f"作成日: {today}")
+
+    # 2) 目次
+    toc = [
+        "業績ハイライト",
+        "見通し",
+        "セグメント",
+        "財務",
+        "株主還元",
+        "ESG",
+        "リスク",
+        "想定Q&A（抜粋）",
+        "参考リンク",
+    ]
+    _add_title_and_bullets(prs, "目次", toc)
+
+    # 3) 各セクション
+    def bullets_of(key: str) -> List[str]:
+        v = sections.get(key, "") if sections else ""
+        if isinstance(v, list):
+            return [str(x) for x in v if str(x).strip()]
+        return _split_lines(str(v or ""))
+
+    _add_title_and_bullets(prs, "業績ハイライト", bullets_of("highlights"))
+    _add_title_and_bullets(prs, "見通し", bullets_of("outlook"))
+    _add_title_and_bullets(prs, "セグメント", bullets_of("segments"))
+    _add_title_and_bullets(prs, "財務", bullets_of("finance"))
+    _add_title_and_bullets(prs, "株主還元", bullets_of("shareholder"))
+    _add_title_and_bullets(prs, "ESG", bullets_of("esg"))
+    _add_title_and_bullets(prs, "リスク", bullets_of("risks"))
+
+    # 4) 想定Q&A（別モジュール側で上位N件だけ詰めて来てもOK）
+    qas = sections.get("qa_top", [])
+    if isinstance(qas, str):
+        qas = _split_lines(qas)
+    qa_bullets = []
+    if qas:
+        # "Q: ... / A: ..." 形式を1行ずつ箇条書き
+        for qa in qas[:MAX_BULLETS]:
+            qa_bullets.append(qa)
+    _add_title_and_bullets(prs, "想定Q&A（抜粋）", qa_bullets or ["（別添CSV参照）"])
+
+    # 5) 参考リンク
+    _add_sources(prs, list(dict.fromkeys(links or [])))
     return prs
 
-def save_pptx(prs: Presentation, path: str) -> str:
+def save_pptx(prs: Presentation, path: str) -> None:
+    import os
     os.makedirs(os.path.dirname(path), exist_ok=True)
     prs.save(path)
-    return path