app.py

import os, tempfile, datetime
import gradio as gr
from dotenv import load_dotenv
from models import SeedInput, CompanyInfo
from llm_flow import generate_narrative
from exporters import get_jinja_env, render_all, make_zip_from_outputs, export_to_notion, export_to_google_docs
from ingest import fetch_url_text, extract_pdf_file

load_dotenv()  # optional for local dev

TEMPLATES_DIR = os.path.join(os.path.dirname(__file__), "templates")
env = get_jinja_env(TEMPLATES_DIR)

DESCRIPTION = """
### 企業PR/IR素材ワンパス生成器
1つの元ネタ（テキスト/URL/PDF）から、プレス/LP/SNSスレッド/メディア向けピッチ、ESG/IRテンプレを一括生成します。  
**LLM: OpenAI**（環境変数 `OPENAI_API_KEY` を設定してください）
"""

def _pdf_path_from_input(pdf_file):
    """gr.File(type='filepath') は str、場合により file-like を返すことがあるので両対応"""
    if pdf_file is None:
        return None
    if isinstance(pdf_file, str):
        return pdf_file
    return getattr(pdf_file, "name", None) or getattr(pdf_file, "path", None)

def choose_source_text(source_text, url, pdf_file):
    log = ""
    text = (source_text or "").strip()

    # 1) PDF優先
    pdf_path = _pdf_path_from_input(pdf_file)
    if pdf_path:
        try:
            text, lg = extract_pdf_file(pdf_path)
            log = f"[PDF] {lg}"
            return text, log
        except Exception as e:
            log = f"[PDF] Error: {e}"

    # 2) URL
    if url and url.strip():
        try:
            text, lg = fetch_url_text(url.strip())
            log = f"[URL] {lg}"
            return text, log
        except Exception as e:
            log = f"[URL] Error: {e}"

    # 3) テキスト貼り付け
    if text:
        return text, "[TEXT] Using pasted text."

    raise ValueError("No valid source provided. Paste text, set a URL, or upload a PDF.")

def run(company_name, brand_voice, region, language, tone, event_type, announce_date,
        goals_text, source_text, url, pdf_file, include_esg, include_ir, do_notion, do_gdocs):
    try:
        goals = [g.strip() for g in (goals_text or "").split("\n") if g.strip()]
        chosen_text, ingest_log = choose_source_text(source_text, url, pdf_file)

        # 入力長の安全上限
        MAX_CHARS = int(os.environ.get("MAX_SOURCE_CHARS", "20000"))
        chosen_text = chosen_text[:MAX_CHARS]

        seed = SeedInput(
            event_type=event_type,
            announce_date=announce_date or None,
            source_text=chosen_text,
            goals=goals,
            tone=tone,
            company=CompanyInfo(
                company_name=company_name,
                brand_voice=brand_voice or None,
                region=region or None,
                language=language
            ),
            include_esg=include_esg,
            include_ir=include_ir,
        )

        narrative = generate_narrative(seed)
        outputs = render_all(env, narrative, seed.company)

        ts = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
        zip_path = os.path.join(tempfile.gettempdir(), f"onepass_outputs_{ts}.zip")
        make_zip_from_outputs(outputs, zip_path)

        notion_result = export_to_notion(outputs) if do_notion else None
        gdocs_result = export_to_google_docs(outputs) if do_gdocs else None

        press = outputs.get("press_release.md", "")
        lp = outputs.get("lp_section.md", "")
        sns = outputs.get("sns_thread.md", "")
        pitch = outputs.get("media_pitch.md", "")
        esg = outputs.get("esg_template.md", "")
        ir = outputs.get("ir_template.md", "")

        export_msg = ""
        if ingest_log:
            export_msg += f"{ingest_log}\n"
        if notion_result:
            export_msg += f"Notion: {notion_result}\n"
        if gdocs_result:
            export_msg += f"Google Docs: {gdocs_result}\n"

        return press, lp, sns, pitch, esg, ir, zip_path, export_msg, chosen_text

    except Exception as e:
        # ここでスタックの要約を返して原因を掴みやすく
        err = f"Error: {type(e).__name__}: {e}"
        return (err,) + ("",)*5 + (None, err) + ("",)

with gr.Blocks(title="One-Pass PR/IR Generator") as demo:
    gr.Markdown(DESCRIPTION)

    with gr.Row():
        with gr.Column():
            company_name = gr.Textbox(label="会社名", placeholder="例）SFM株式会社", value="SFM株式会社")
            brand_voice = gr.Textbox(label="ブランドボイス（任意）", placeholder="信頼性 / 革新性 / 先進性 など")
            region = gr.Textbox(label="地域/市場（任意）", placeholder="JP / US / APAC など")
            language = gr.Dropdown(["ja", "en"], value="ja", label="出力言語")
            tone = gr.Dropdown(["formal", "friendly", "excited", "neutral"], value="formal", label="トーン")
            event_type = gr.Dropdown(["product_update", "earnings_summary"], value="product_update", label="イベント種別")
            announce_date = gr.Textbox(label="発表日（任意）", placeholder="YYYY-MM-DD")

        with gr.Column():
            gr.Markdown("#### 情報ソース（いずれか）")
            source_text = gr.Textbox(label="テキスト貼り付け（任意）", placeholder="製品更新・決算要旨を貼り付け", lines=8)
            url = gr.Textbox(label="URL（任意：WebページまたはPDF）", placeholder="https://...")
            pdf_file = gr.File(label="PDFアップロード（任意）", file_types=[".pdf"], type="filepath")
            goals_text = gr.Textbox(label="目的（KGI/KPIなど）", placeholder="1行につき1つ", lines=4)
            include_esg = gr.Checkbox(label="ESGテンプレを出力", value=True)
            include_ir = gr.Checkbox(label="IRテンプレを出力", value=True)
            do_notion = gr.Checkbox(label="Notionへエクスポート", value=False)
            do_gdocs = gr.Checkbox(label="Google Docsへエクスポート", value=False)

    run_btn = gr.Button("一括生成", variant="primary")

    with gr.Tab("プレスリリース"):
        press_md = gr.Markdown("")
    with gr.Tab("LPセクション"):
        lp_md = gr.Markdown("")
    with gr.Tab("SNSスレッド"):
        sns_md = gr.Markdown("")
    with gr.Tab("メディア向けピッチ"):
        pitch_md = gr.Markdown("")
    with gr.Tab("ESGテンプレ"):
        esg_md = gr.Markdown("")
    with gr.Tab("IRテンプレ"):
        ir_md = gr.Markdown("")

    zip_file = gr.File(label="生成結果ZIP", interactive=False)
    ingest_status = gr.Textbox(label="インポートログ / Export結果", interactive=False)
    preview_src = gr.Textbox(label="投入された元テキスト（プレビュー）", lines=10, interactive=False)

    run_btn.click(
        fn=run,
        inputs=[company_name, brand_voice, region, language, tone, event_type, announce_date,
                goals_text, source_text, url, pdf_file, include_esg, include_ir, do_notion, do_gdocs],
        outputs=[press_md, lp_md, sns_md, pitch_md, esg_md, ir_md, zip_file, ingest_status, preview_src],
    )

if __name__ == "__main__":
    demo.launch()