Upload 3 files

README.md

@@ -1,12 +1,28 @@
----
-title: Agent Studio
-emoji: 🏆
-colorFrom: pink
-colorTo: pink
-sdk: gradio
-sdk_version: 5.46.1
-app_file: app.py
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# 営業自動化 Agent Studio (HF Spaces)
+
+## セットアップ
+1. 本リポジトリ構成でファイルを用意（`app.py` がエントリ）。
+2. Hugging Face Spaces で **Python** Space を作成し、上記ファイルをアップロード。
+3. **Variables/Secrets** に `.env.example` の内容を登録（少なくとも `PUBLIC_BASE_URL`, `TRACKING_SECRET` 推奨）。
+4. 必要であれば SMTP 設定を投入（メール送信を使う場合）。
+
+## できること
+- 企業スコア → RAG（URL/ファイル）→ 提案ドラフト生成（LLM）→ DOCX/PPTXエクスポート
+- メール本文の生成（トラッキングリンク挿入）
+- `/t/{token}` へのアクセスでクリックログを `data/events.jsonl` に蓄積
+
+## LLM
+- 既定は `google/flan-t5-base`（軽量）。必要なら `HF_LOCAL_MODEL_ID` を `mistralai/Mistral-7B-Instruct-v0.3` などに変更。
+- Inference API を使う場合は `USE_HF_INFERENCE_API=true` と `HF_API_TOKEN` を設定。
+
+## 使い方
+- UIに企業名・URL・（任意で）送信先メール、RAG用URLやファイルを指定→「ワークフロー実行」
+- `data/` に `proposal_*.docx` と `proposal_*.pptx` が出力されます
+- クリックログは `data/events.jsonl` をご確認ください
+
+## 送信
+- 実運用では `modules/emailer.py` の `send_email` を呼ぶAPIやUIボタンを追加してください（デモでは本文プレビューのみ）。
+
+## 注意
+- 無断クロールを避けるため、RAGのURL投入は対象サイトの規約に従ってください。
+- 機密ファイルの取り扱いには十分ご注意ください。

app.py

@@ -0,0 +1,106 @@
+import os
+import json
+from pathlib import Path
+from typing import Optional, List
+
+from fastapi import FastAPI, Request, HTTPException
+from fastapi.responses import PlainTextResponse, RedirectResponse
+import gradio as gr
+from dotenv import load_dotenv
+
+from modules.workflow import run_full_workflow
+from modules.rag_indexer import index_files_and_urls
+from modules.rag_retriever import retrieve_contexts
+from modules.emailer import build_tracking_url
+from modules.utils import ensure_dirs, verify_tracking_token, log_event
+
+load_dotenv()
+ensure_dirs()
+
+app = FastAPI(title="Agent Studio")
+
+### ===== Gradio UI =====
+def ui_company_score_and_proposal(company_name: str,
+                                  company_website: str,
+                                  lead_email: str,
+                                  urls_text: str,
+                                  files: List[gr.File] | None,
+                                  custom_objective: str,
+                                  temperature: float = 0.4):
+    urls = [u.strip() for u in urls_text.splitlines() if u.strip()] if urls_text else []
+    file_paths = [f.name for f in (files or [])]
+
+    # 1) インデックス更新（アップロードとURL）
+    index_report = index_files_and_urls(file_paths=file_paths, urls=urls)
+
+    # 2) ワークフロー実行（スコア→RAG→提案→次アクション）
+    result = run_full_workflow(
+        company_name=company_name,
+        company_website=company_website,
+        lead_email=lead_email,
+        objective=custom_objective,
+        temperature=temperature
+    )
+
+    # 3) 出力まとめ
+    outputs = []
+    outputs.append("### ✅ 企業スコア\n" + json.dumps(result["score"], ensure_ascii=False, indent=2))
+    outputs.append("### 🧠 抽出コンテキスト（上位）\n" + "\n\n".join([f"- {c[:300]}..." for c in result["top_contexts"]]))
+    outputs.append("### ✍️ 提案ドラフト\n" + result["proposal_markdown"])
+    outputs.append("### 📎 エクスポート\n" + "\n".join([
+        f"- DOCX: {result['exports']['docx_path']}",
+        f"- PPTX: {result['exports']['pptx_path']}"
+    ]))
+    if lead_email:
+        outputs.append("### ✉️ メール準備\n" + f"To: {lead_email}\nSubject: {result['email']['subject']}\n\n{result['email']['body']}")
+        outputs.append(f"（本文内のトラッキングリンク例）\n{build_tracking_url('preview-only', {'company':company_name})}")
+
+    outputs.append("### 🤖 次アクション提案\n" + result["next_actions"])
+    outputs.append("### 🧩 インデックス更新ログ\n" + index_report)
+
+    return "\n\n".join(outputs)
+
+with gr.Blocks(title="Agent Studio") as demo:
+    gr.Markdown("# 営業自動化 Agent Studio")
+    with gr.Row():
+        with gr.Column():
+            company_name = gr.Textbox(label="企業名", placeholder="例）Acme Corp")
+            company_website = gr.Textbox(label="企業サイトURL", placeholder="例）https://www.acme.com")
+            lead_email = gr.Textbox(label="送信先メール（任意）", placeholder="例）lead@acme.com")
+            custom_objective = gr.Textbox(label="提案の目的/狙い（任意）", placeholder="例）SaaS導入の無料PoC打診")
+            urls_text = gr.Textbox(label="RAG用URL（複数は改行）", lines=4)
+            files = gr.File(label="RAG用ファイル（複数可）", file_count="multiple")
+            temperature = gr.Slider(0.0, 1.0, value=0.4, step=0.05, label="生成温度")
+            run_btn = gr.Button("ワークフロー実行", variant="primary")
+        with gr.Column():
+            out = gr.Markdown(label="結果")
+
+    run_btn.click(
+        fn=ui_company_score_and_proposal,
+        inputs=[company_name, company_website, lead_email, urls_text, files, custom_objective, temperature],
+        outputs=[out]
+    )
+
+# HF Spacesは "app" 変数をエクスポートしている必要あり
+# FastAPIにGradioをマウント（ルートに表示）
+from fastapi.middleware.cors import CORSMiddleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"], allow_credentials=True, allow_methods=["*"], allow_headers=["*"]
+)
+app = gr.mount_gradio_app(app, demo, path="/")
+
+### ===== クリックトラッキング・エンドポイント =====
+@app.get("/t/{token}", response_class=PlainTextResponse)
+def track_click(token: str, request: Request):
+    # token署名検証 & ペイロード取得
+    payload = verify_tracking_token(token)
+    if payload is None:
+        raise HTTPException(status_code=400, detail="invalid token")
+    # ログ
+    ip = request.client.host if request.client else "unknown"
+    ua = request.headers.get("User-Agent", "")
+    log_event(event_type="click", payload=payload, meta={"ip": ip, "ua": ua})
+    # 遷移先URL（ペイロードにredirectがあればそこへ、なければUIトップへ）
+    redirect_to = payload.get("redirect") or os.getenv("PUBLIC_BASE_URL", "/")
+    return RedirectResponse(url=redirect_to)

requirements.txt

@@ -0,0 +1,19 @@
+fastapi==0.111.1
+uvicorn==0.30.6
+gradio==4.44.1
+
+transformers==4.44.2
+torch>=2.1.0
+sentence-transformers==3.0.1
+faiss-cpu==1.8.0.post1
+
+python-docx==1.1.2
+python-pptx==0.6.23
+
+beautifulspup4==4.12.3
+requests==2.32.3
+readability-lxml==0.8.1
+lxml==5.3.0
+tldextract==5.1.2
+python-dotenv==1.0.1
+pydantic==2.9.2