Update modules/workflow.py

modules/workflow.py

@@ -1,64 +1,123 @@
+# modules/workflow.py
 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","/")
+from typing import Dict, Any, List
+
+# --- retriever は相対インポートを優先、失敗時は絶対インポートにフォールバック ---
+try:
+    from .rag_retriever import retrieve_contexts
+except Exception:
+    from modules.rag_retriever import retrieve_contexts  # type: ignore
+
+# exporters / utils も相対で
+try:
+    from .exporters import export_docx, export_pptx
+    from .utils import ensure_dirs, export_dir, make_tracking_token
+except Exception:
+    # 万一相対が失敗する環境でも動くように
+    from modules.exporters import export_docx, export_pptx  # type: ignore
+    from modules.utils import ensure_dirs, export_dir, make_tracking_token  # type: ignore
+
+
+def _score_company(company_name: str, website: str, contexts: List[str]) -> Dict[str, Any]:
+    """
+    デモ用の簡易スコア（実運用では独自ロジック/モデルと差し替え可）
+    """
+    score = 50
+    if website:
+        score += 10
+    if contexts:
+        score += min(40, len(contexts) * 5)
+    return {
+        "company": company_name,
+        "website": website,
+        "score": min(score, 95),
+        "factors": {
+            "has_website": bool(website),
+            "ctx_hits": len(contexts),
         },
-    )
-    subject = f"{company_name}さま向けご提案(無料PoCのご案内)"
+    }
+
+
+def _draft_proposal_md(company_name: str, objective: str, contexts: List[str]) -> str:
+    bullets = "\n".join([f"- {c[:200]}..." for c in contexts[:5]])
+    obj = objective or "商談化のための初回提案"
+    return f"""# 提案ドラフト（{company_name} 向け）
+
+## 目的
+{obj}
+
+## 背景インサイト（上位抜粋）
+{bullets if bullets else '- （まだインデックスが空です）'}
+
+## 提案骨子
+1. 現状の課題仮説を確認
+2. PoC スコープのすり合わせ
+3. 成果指標（KPI）とスケジュール
+4. 次のアクション
+"""
+
+
+def _draft_email(company_name: str) -> Dict[str, str]:
+    subject = f"{company_name}様向けご提案（ドラフト）"
     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"
+        f"{company_name} ご担当者様\n\n"
+        "お世話になっております。簡易の提案ドラフトを共有いたします。\n"
+        "ご確認の上、ご意見いただけますと幸いです。\n\n"
+        "何卒よろしくお願いいたします。"
     )
     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)
+def run_full_workflow(
+    company_name: str,
+    company_website: str,
+    lead_email: str,
+    objective: str,
+    temperature: float = 0.4,
+) -> Dict[str, Any]:
+    """
+    app.py から呼ばれるメイン処理。
+    - RAG で上位コンテキスト取得
+    - 簡易スコアリング
+    - 提案ドラフト生成
+    - DOCX/PPTX にエクスポート
+    - メール文面と次アクションの素案
+    """
+    ensure_dirs()
+
+    query = f"{company_name} {objective or ''}".strip()
+    top_contexts = retrieve_contexts(query, top_k=5)
 
-    # 3) 提案生成
-    proposal_md = make_proposal(company_name, objective, top_contexts, temperature=temperature)
+    score = _score_company(company_name, company_website, top_contexts)
+    proposal_md = _draft_proposal_md(company_name, objective, top_contexts)
 
-    # 4) エクスポート
-    docx_path = export_docx(proposal_md, company_name)
-    pptx_path = export_pptx(proposal_md, company_name)
+    # エクスポート先パス
+    out_dir = export_dir()
+    docx_path = os.path.join(out_dir, f"{company_name}_proposal.docx")
+    pptx_path = os.path.join(out_dir, f"{company_name}_proposal.pptx")
 
-    # 5) メール下書き
-    email = _make_email(company_name, lead_email or "", proposal_md)
+    # 出力
+    export_docx(proposal_md, docx_path)
+    export_pptx(proposal_md, pptx_path)
 
-    # 6) 次アクション
-    next_actions = suggest_next_actions(proposal_md, score["overall"])
+    email = _draft_email(company_name)
+
+    # 次アクション（ダミー）
+    next_actions = (
+        "- 先方の課題確認ミーティング候補日程を提案する\n"
+        "- PoC スコープの要件一覧を作成して共有\n"
+        "- 競合比較の要点を 3 点に要約\n"
+    )
+
+    # 計測リンクの例（app.py の /t/{token} で受ける想定）
+    tracking_token = make_tracking_token({"company": company_name, "redirect": "/"})
 
     return {
         "score": score,
         "top_contexts": top_contexts,
         "proposal_markdown": proposal_md,
-        "exports": {
-            "docx_path": docx_path,
-            "pptx_path": pptx_path,
-        },
+        "exports": {"docx_path": docx_path, "pptx_path": pptx_path},
         "email": email,
-        "next_actions": next_actions
+        "next_actions": next_actions,
+        "tracking_token": tracking_token,
     }