Spaces:
Sleeping
Sleeping
Update app/main.py
Browse files- app/main.py +0 -9
app/main.py
CHANGED
|
@@ -5,9 +5,6 @@ from generators.summary import make_summary
|
|
| 5 |
from generators.qa import make_qa
|
| 6 |
from export.ppt import build_deck, save_pptx
|
| 7 |
from export.qa_csv import save_qa_csv
|
| 8 |
-
from app.config import settings
|
| 9 |
-
from jinja2 import Template
|
| 10 |
-
import os
|
| 11 |
|
| 12 |
app = FastAPI(title="IR/PR Co-Pilot Pro", version="0.1.0")
|
| 13 |
|
|
@@ -22,9 +19,7 @@ def ingest_edinet(req: IngestRequest):
|
|
| 22 |
|
| 23 |
@app.post("/generate/all")
|
| 24 |
def generate_all(req: GenerateRequest):
|
| 25 |
-
# 1) 要約
|
| 26 |
summary_text, links = make_summary(req.query)
|
| 27 |
-
# Jinja整形(ここは簡易; 実運用は専用パーサを)
|
| 28 |
sections = {
|
| 29 |
"highlights": _extract_section(summary_text, "業績ハイライト"),
|
| 30 |
"outlook": _extract_section(summary_text, "見通し"),
|
|
@@ -34,22 +29,18 @@ def generate_all(req: GenerateRequest):
|
|
| 34 |
"esg": _extract_section(summary_text, "ESG"),
|
| 35 |
"risks": _extract_section(summary_text, "リスク"),
|
| 36 |
}
|
| 37 |
-
# 2) Q&A
|
| 38 |
qa_list, links2 = make_qa(req.query, 30)
|
| 39 |
-
# 3) PPT/CSV 出力
|
| 40 |
prs = build_deck(sections, links)
|
| 41 |
ppt_path = save_pptx(prs, "data/ir_summary.pptx")
|
| 42 |
csv_path = save_qa_csv(qa_list, links2, "data/qa_30.csv")
|
| 43 |
return {"pptx": ppt_path, "qa_csv": csv_path, "links": links}
|
| 44 |
|
| 45 |
def _extract_section(text: str, head: str):
|
| 46 |
-
# " - 業績ハイライト:" の行以降〜次セクション直前までを抜く程度の簡易版
|
| 47 |
import re
|
| 48 |
pat = rf"{head}[::]\s*(.*?)(?:\n-\s*\S+[::]|\Z)"
|
| 49 |
m = re.search(pat, text, re.S)
|
| 50 |
return (m.group(1).strip() if m else "").strip()
|
| 51 |
|
| 52 |
-
# --- プロキシ: PPT内の根拠リンクを外へ連携する安全な方法 ---
|
| 53 |
@app.get("/proxy/edinet/{doc_id}")
|
| 54 |
def proxy_edinet(doc_id: str, type: str = "pdf"):
|
| 55 |
if type != "pdf":
|
|
|
|
| 5 |
from generators.qa import make_qa
|
| 6 |
from export.ppt import build_deck, save_pptx
|
| 7 |
from export.qa_csv import save_qa_csv
|
|
|
|
|
|
|
|
|
|
| 8 |
|
| 9 |
app = FastAPI(title="IR/PR Co-Pilot Pro", version="0.1.0")
|
| 10 |
|
|
|
|
| 19 |
|
| 20 |
@app.post("/generate/all")
|
| 21 |
def generate_all(req: GenerateRequest):
|
|
|
|
| 22 |
summary_text, links = make_summary(req.query)
|
|
|
|
| 23 |
sections = {
|
| 24 |
"highlights": _extract_section(summary_text, "業績ハイライト"),
|
| 25 |
"outlook": _extract_section(summary_text, "見通し"),
|
|
|
|
| 29 |
"esg": _extract_section(summary_text, "ESG"),
|
| 30 |
"risks": _extract_section(summary_text, "リスク"),
|
| 31 |
}
|
|
|
|
| 32 |
qa_list, links2 = make_qa(req.query, 30)
|
|
|
|
| 33 |
prs = build_deck(sections, links)
|
| 34 |
ppt_path = save_pptx(prs, "data/ir_summary.pptx")
|
| 35 |
csv_path = save_qa_csv(qa_list, links2, "data/qa_30.csv")
|
| 36 |
return {"pptx": ppt_path, "qa_csv": csv_path, "links": links}
|
| 37 |
|
| 38 |
def _extract_section(text: str, head: str):
|
|
|
|
| 39 |
import re
|
| 40 |
pat = rf"{head}[::]\s*(.*?)(?:\n-\s*\S+[::]|\Z)"
|
| 41 |
m = re.search(pat, text, re.S)
|
| 42 |
return (m.group(1).strip() if m else "").strip()
|
| 43 |
|
|
|
|
| 44 |
@app.get("/proxy/edinet/{doc_id}")
|
| 45 |
def proxy_edinet(doc_id: str, type: str = "pdf"):
|
| 46 |
if type != "pdf":
|