Spaces:
Runtime error
Runtime error
Update modules/workflow.py
Browse files- modules/workflow.py +14 -14
modules/workflow.py
CHANGED
|
@@ -1,8 +1,8 @@
|
|
| 1 |
import os
|
| 2 |
from modules.company_score import score_company
|
| 3 |
-
from
|
| 4 |
-
from
|
| 5 |
-
from
|
| 6 |
from modules.emailer import build_tracking_url
|
| 7 |
from modules.utils import truncate
|
| 8 |
|
|
@@ -14,17 +14,17 @@ def _make_email(company_name: str, lead_email: str, proposal_md: str) -> dict:
|
|
| 14 |
"redirect": os.getenv("PUBLIC_BASE_URL", "/")
|
| 15 |
},
|
| 16 |
)
|
| 17 |
-
subject = f"{company_name}
|
| 18 |
body = (
|
| 19 |
-
f"{company_name}
|
| 20 |
-
"
|
| 21 |
f"▼提案要旨\n{truncate(proposal_md, 800)}\n\n"
|
| 22 |
-
f"
|
| 23 |
-
"30
|
| 24 |
"ご検討のほど、よろしくお願いいたします。\n"
|
| 25 |
"\n--\nAgent Studio"
|
| 26 |
)
|
| 27 |
-
return{"subject": subject, "body": body}
|
| 28 |
|
| 29 |
def run_full_workflow(company_name: str,
|
| 30 |
company_website: str,
|
|
@@ -34,7 +34,7 @@ def run_full_workflow(company_name: str,
|
|
| 34 |
# 1) スコアリング
|
| 35 |
score = score_company(company_name, company_website)
|
| 36 |
|
| 37 |
-
# 2)
|
| 38 |
query = f"{company_name} {objective} {company_website}"
|
| 39 |
top_contexts = retrieve_contexts(query, k=5)
|
| 40 |
|
|
@@ -45,19 +45,19 @@ def run_full_workflow(company_name: str,
|
|
| 45 |
docx_path = export_docx(proposal_md, company_name)
|
| 46 |
pptx_path = export_pptx(proposal_md, company_name)
|
| 47 |
|
| 48 |
-
# 5) メール作成
|
| 49 |
email = _make_email(company_name, lead_email or "", proposal_md)
|
| 50 |
|
| 51 |
# 6) 次アクション提案
|
| 52 |
next_actions = suggest_next_actions(proposal_md, score["overall"])
|
| 53 |
|
| 54 |
-
return{
|
| 55 |
"score": score,
|
| 56 |
"top_contexts": top_contexts,
|
| 57 |
"proposal_markdown": proposal_md,
|
| 58 |
-
"exports":{
|
| 59 |
"docx_path": docx_path,
|
| 60 |
-
"pptx_path": pptx_path
|
| 61 |
},
|
| 62 |
"email": email,
|
| 63 |
"next_actions": next_actions
|
|
|
|
| 1 |
import os
|
| 2 |
from modules.company_score import score_company
|
| 3 |
+
from modules.rag_retriever import retrieve_contexts # ← ここを修正(modules)
|
| 4 |
+
from modules.proposal_generator import make_proposal, suggest_next_actions
|
| 5 |
+
from modules.exporters import export_docx, export_pptx
|
| 6 |
from modules.emailer import build_tracking_url
|
| 7 |
from modules.utils import truncate
|
| 8 |
|
|
|
|
| 14 |
"redirect": os.getenv("PUBLIC_BASE_URL", "/")
|
| 15 |
},
|
| 16 |
)
|
| 17 |
+
subject = f"{company_name}様向けご提案(無料PoCのご案内)"
|
| 18 |
body = (
|
| 19 |
+
f"{company_name} ご担当者様\n\n"
|
| 20 |
+
"突然のご連絡失礼いたします。貴社向けに要点を1枚にまとめた提案書を作成しました。\n\n"
|
| 21 |
f"▼提案要旨\n{truncate(proposal_md, 800)}\n\n"
|
| 22 |
+
f"詳細はこちら(クリック計測用リンク):\n{tracking}\n\n"
|
| 23 |
+
"30分ほどオンラインでディスカッションできれば嬉しいです。\n"
|
| 24 |
"ご検討のほど、よろしくお願いいたします。\n"
|
| 25 |
"\n--\nAgent Studio"
|
| 26 |
)
|
| 27 |
+
return {"subject": subject, "body": body}
|
| 28 |
|
| 29 |
def run_full_workflow(company_name: str,
|
| 30 |
company_website: str,
|
|
|
|
| 34 |
# 1) スコアリング
|
| 35 |
score = score_company(company_name, company_website)
|
| 36 |
|
| 37 |
+
# 2) コンテキスト検索(RAG)
|
| 38 |
query = f"{company_name} {objective} {company_website}"
|
| 39 |
top_contexts = retrieve_contexts(query, k=5)
|
| 40 |
|
|
|
|
| 45 |
docx_path = export_docx(proposal_md, company_name)
|
| 46 |
pptx_path = export_pptx(proposal_md, company_name)
|
| 47 |
|
| 48 |
+
# 5) メール作成
|
| 49 |
email = _make_email(company_name, lead_email or "", proposal_md)
|
| 50 |
|
| 51 |
# 6) 次アクション提案
|
| 52 |
next_actions = suggest_next_actions(proposal_md, score["overall"])
|
| 53 |
|
| 54 |
+
return {
|
| 55 |
"score": score,
|
| 56 |
"top_contexts": top_contexts,
|
| 57 |
"proposal_markdown": proposal_md,
|
| 58 |
+
"exports": {
|
| 59 |
"docx_path": docx_path,
|
| 60 |
+
"pptx_path": pptx_path
|
| 61 |
},
|
| 62 |
"email": email,
|
| 63 |
"next_actions": next_actions
|