Spaces:
Sleeping
Sleeping
Delete app/main.py
Browse files- 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")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|