Corin1998 commited on
Commit
91dc69f
·
verified ·
1 Parent(s): 3f322c7

Update app/main.py

Browse files
Files changed (1) hide show
  1. 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":