Create main.py

irpr/main.py

@@ -0,0 +1,52 @@
+from fastapi import FastAPI, HTTPException, Response
+from irpr.models import IngestRequest, GenerateRequest
+from rag.ingest import ingest_edinet_for_company, download_edinet_pdf
+from generators.summary import make_summary
+from generators.qa import make_qa
+from export.ppt import build_deck, save_pptx
+from export.qa_csv import save_qa_csv
+
+app = FastAPI(title="IR/PR Co-Pilot Pro", version="0.1.0")
+
+@app.get("/")
+def root():
+    return {"ok": True, "service": "IR/PR Co-Pilot Pro"}
+
+@app.post("/ingest/edinet")
+def ingest_edinet(req: IngestRequest):
+    n = ingest_edinet_for_company(req.edinet_code, req.date)
+    return {"ingested_chunks": n}
+
+@app.post("/generate/all")
+def generate_all(req: GenerateRequest):
+    summary_text, links = make_summary(req.query)
+    sections = {
+        "highlights": _extract_section(summary_text, "業績ハイライト"),
+        "outlook":    _extract_section(summary_text, "見通し"),
+        "segments":   _extract_section(summary_text, "セグメント"),
+        "finance":    _extract_section(summary_text, "財務"),
+        "shareholder":_extract_section(summary_text, "株主還元"),
+        "esg":        _extract_section(summary_text, "ESG"),
+        "risks":      _extract_section(summary_text, "リスク"),
+    }
+    qa_list, links2 = make_qa(req.query, 30)
+    prs = build_deck(sections, links)
+    ppt_path = save_pptx(prs, "data/ir_summary.pptx")
+    csv_path = save_qa_csv(qa_list, links2, "data/qa_30.csv")
+    return {"pptx": ppt_path, "qa_csv": csv_path, "links": links}
+
+def _extract_section(text: str, head: str):
+    import re
+    pat = rf"{head}[:：]\s*(.*?)(?:\n-\s*\S+[:：]|\Z)"
+    m = re.search(pat, text, re.S)
+    return (m.group(1).strip() if m else "").strip()
+
+@app.get("/proxy/edinet/{doc_id}")
+def proxy_edinet(doc_id: str, type: str = "pdf"):
+    if type != "pdf":
+        raise HTTPException(400, "unsupported type")
+    from rag.ingest import download_edinet_pdf
+    pdf = download_edinet_pdf(doc_id)
+    if not pdf:
+        raise HTTPException(404, "file not found")
+    return Response(content=pdf, media_type="application/pdf")