Update irpr/main.py

irpr/main.py

@@ -1,22 +1,66 @@
-from fastapi import FastAPI, HTTPException, Response
+from fastapi import FastAPI, HTTPException, Response, UploadFile, File, Request
+from fastapi.responses import HTMLResponse
+from fastapi.staticfiles import StaticFiles
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.templating import Jinja2Templates
 from irpr.models import IngestRequest, GenerateRequest
+import os
 
-app = FastAPI(title="IR/PR Co-Pilot Pro", version="0.1.0")
+app = FastAPI(title="IR/PR Co-Pilot Pro", version="0.2.0")
 
-@app.get("/")
-def root():
+# --- CORS（同一オリジン運用想定だが念のため全許可） ---
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"], allow_credentials=True, allow_methods=["*"], allow_headers=["*"],
+)
+
+# --- 静的 & 生成物の公開マウント ---
+os.makedirs("static", exist_ok=True)
+os.makedirs("templates", exist_ok=True)
+os.makedirs("data", exist_ok=True)
+os.makedirs("data/uploads", exist_ok=True)
+
+app.mount("/static", StaticFiles(directory="static"), name="static")
+app.mount("/files", StaticFiles(directory="data"), name="files")
+templates = Jinja2Templates(directory="templates")
+
+# ---------------- UI ----------------
+@app.get("/", response_class=HTMLResponse)
+def ui(request: Request):
+    return templates.TemplateResponse("index.html", {"request": request})
+
+@app.get("/api/health")
+def health():
     return {"ok": True, "service": "IR/PR Co-Pilot Pro"}
 
+# ------------- API: 取込 -------------
 @app.post("/ingest/edinet")
 def ingest_edinet(req: IngestRequest):
-    # ← ここで初めて読み込む（起動時に重い依存を踏まない）
     from rag.ingest import ingest_edinet_for_company
     n = ingest_edinet_for_company(req.edinet_code, req.date)
     return {"ingested_chunks": n}
 
+@app.post("/ingest/upload")
+async def ingest_upload(files: list[UploadFile] = File(...)):
+    from rag.ingest import ingest_pdf_bytes
+    total = 0
+    saved = []
+    for f in files:
+        if not f.filename.lower().endswith(".pdf"):
+            continue
+        blob = await f.read()
+        # 保存してダウンロード可能に
+        save_path = os.path.join("data", "uploads", f.filename)
+        with open(save_path, "wb") as w:
+            w.write(blob)
+        source_url = f"/files/uploads/{f.filename}"
+        total += ingest_pdf_bytes(title=f.filename, source_url=source_url, pdf_bytes=blob)
+        saved.append(source_url)
+    return {"ingested_chunks": total, "saved": saved}
+
+# ------------- API: 生成 -------------
 @app.post("/generate/all")
 def generate_all(req: GenerateRequest):
-    # 必要なものだけ都度 import（遅延）
     from generators.summary import make_summary
     from generators.qa import make_qa
     from export.ppt import build_deck, save_pptx
@@ -36,21 +80,26 @@ def generate_all(req: GenerateRequest):
     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()
+    return {
+        "pptx": ppt_path.replace("data/", "/files/"),
+        "qa_csv": csv_path.replace("data/", "/files/"),
+        "links": links
+    }
 
+# ------------- API: EDINET PDF 直配 -------------
 @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")
+
+# ------------- helper -------------
+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()