Create workflow.py

modules/workflow.py

@@ -0,0 +1,64 @@
+import os
+from modules.company_score import score_company
+from modules.rag_retriever import retrieve_contexts
+from modules.proposal_generator import make_proposal, suggest_next_actions
+from modules.exporters import export_docx, export_pptx
+from modules.emailer import build_tracking_url
+from modules.utils import truncate
+
+def _make_email(company_name: str, lead_email: str, proposal_md: str) -> dict:
+    traking = build_tracking_url(
+        identifier=f"{company_name}",
+        payload={
+            "company": company_name,
+            "redirect": os.getenv("PUBLIC_BASE_URL","/")
+        },
+    )
+    subject = f"{company_name}さま向けご提案(無料PoCのご案内)"
+    body = (
+        f"{company_name} ご担当者さま\n\n"
+        "突然のご連絡失礼致します。貴社向けに要点を1枚にまとめた提案書を作成しました。\n\n"
+        f"▼提案要旨\n{truncate(proposal_md, 800)}\n\n"
+        f"詳細はこちら(クリック計測用リンク) :\n{traking}\n\n"
+        f"30分ほどのオンラインミーティングの機会を頂戴できればと幸いです。\n"
+        "ご検討のほど、よろしくお願いいたします。\n"
+        "\n--\nAgent Studio"
+    )
+    return {"subject": subject, "body": body}
+
+def run_full_workflow(company_name: str,
+                      company_website: str,
+                      lead_email: str | None,
+                      objective: str,
+                      temperature: float=0.4) -> dict:
+    # 1) スコア
+    score = score_company(company_name, company_website)
+
+    # 2) RAG 検索
+    query = f"{company_name} {objective} {company_website}"
+    top_contexts = retrieve_contexts(query, k=5)
+
+    # 3) 提案生成
+    proposal_md = make_proposal(company_name, objective, top_contexts, temperature=temperature)
+
+    # 4) エクスポート
+    docx_path = export_docx(proposal_md, company_name)
+    pptx_path = export_pptx(proposal_md, company_name)
+
+    # 5) メール下書き
+    email = _make_email(company_name, lead_email or "", proposal_md)
+
+    # 6) 次アクション
+    next_actions = suggest_next_actions(proposal_md, score["overall"])
+
+    return {
+        "score": score,
+        "top_contexts": top_contexts,
+        "proposal_markdown": proposal_md,
+        "exports": {
+            "docx_path": docx_path,
+            "pptx_path": pptx_path,
+        },
+        "email": email,
+        "next_actions": next_actions
+    }