Corin1998 commited on
Commit
8e209de
·
verified ·
1 Parent(s): a7e95cc

Delete app/main.py

Browse files
Files changed (1) hide show
  1. app/main.py +0 -59
app/main.py DELETED
@@ -1,59 +0,0 @@
1
- from fastapi import FastAPI, HTTPException, Response
2
-
3
- # --- ここが重要:二段構えの import で確実に解決 ---
4
- try:
5
- # 通常はこれで通る(/code が PYTHONPATH にある想定)
6
- from app.models import IngestRequest, GenerateRequest
7
- except ModuleNotFoundError:
8
- # 念のため、相対 import にフォールバック(app パッケージ配下で実行される場合)
9
- from .models import IngestRequest, GenerateRequest
10
-
11
- from rag.ingest import ingest_edinet_for_company, download_edinet_pdf
12
- from generators.summary import make_summary
13
- from generators.qa import make_qa
14
- from export.ppt import build_deck, save_pptx
15
- from export.qa_csv import save_qa_csv
16
-
17
- app = FastAPI(title="IR/PR Co-Pilot Pro", version="0.1.0")
18
-
19
- @app.get("/")
20
- def root():
21
- return {"ok": True, "service": "IR/PR Co-Pilot Pro"}
22
-
23
- @app.post("/ingest/edinet")
24
- def ingest_edinet(req: IngestRequest):
25
- n = ingest_edinet_for_company(req.edinet_code, req.date)
26
- return {"ingested_chunks": n}
27
-
28
- @app.post("/generate/all")
29
- def generate_all(req: GenerateRequest):
30
- summary_text, links = make_summary(req.query)
31
- sections = {
32
- "highlights": _extract_section(summary_text, "業績ハイライト"),
33
- "outlook": _extract_section(summary_text, "見通し"),
34
- "segments": _extract_section(summary_text, "セグメント"),
35
- "finance": _extract_section(summary_text, "財務"),
36
- "shareholder":_extract_section(summary_text, "株主還元"),
37
- "esg": _extract_section(summary_text, "ESG"),
38
- "risks": _extract_section(summary_text, "リスク"),
39
- }
40
- qa_list, links2 = make_qa(req.query, 30)
41
- prs = build_deck(sections, links)
42
- ppt_path = save_pptx(prs, "data/ir_summary.pptx")
43
- csv_path = save_qa_csv(qa_list, links2, "data/qa_30.csv")
44
- return {"pptx": ppt_path, "qa_csv": csv_path, "links": links}
45
-
46
- def _extract_section(text: str, head: str):
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
- @app.get("/proxy/edinet/{doc_id}")
53
- def proxy_edinet(doc_id: str, type: str = "pdf"):
54
- if type != "pdf":
55
- raise HTTPException(400, "unsupported type")
56
- pdf = download_edinet_pdf(doc_id)
57
- if not pdf:
58
- raise HTTPException(404, "file not found")
59
- return Response(content=pdf, media_type="application/pdf")