headline_generator.py

import openai
import os

def generate_headlines_and_subheadings(new_keyword, ngram_text):
    try:
        response = openai.ChatCompletion.create(
            model="gpt-4o",
            messages=[
                {"role": "system", "content": "あなたはライティングの専門家です。PREP法（結論・理由・具体例・結論の流れ）に則って見出しを構成してください。テキストは必ず日本語で生成してください。テキスト内に似た文章を作らないでください。同じ単語は3回まで使っていいものとします。"},
                {"role": "user", "content": f'（キーワード）の感情分析を行った結果に基づいて、興味や関心を引きつける「（キーワード）を含めたタイトル」、興味や関心を引きつける「見出し」を生成してください。そしてそれぞれの見出しに沿って「小見出し」を適宜生成してください。最後に、記事を締めくくる「<h2>まとめ</h2>」のセクションを追加してください。タイトル、見出し、小見出しはPREP法に基づいて一貫性のあるものとし、記事の前半部分で（キーワード）の検索意図を満たす記事構成内容にし、（テキスト）が（キーワード）に関連している場合、（テキスト）に含まれる語句を使用して「見出し」、「小見出し」を生成してください。（テキスト）は（キーワード）を検索するユーザーの検索意図を深掘りした興味や関心です。（テキスト）で特定のサービスや製品名が引用符（「""」,「「」」,「（）」）で囲まれて言及されていて、それが（キーワード）に関連している場合は、積極的に「見出し」、「小見出し」の生成に使用してください。情報を次のように整理してください：タイトル、見出しおよび小見出しは順に「<h1>（キーワード）を含めたタイトル</h1>」「<h2>1.見出し</h2>」「<h3>1-1 小見出し とは</h3>、<h3>1-2 小見出し とは</h3>」「<h2>2.見出し</h2>」「<h3>2-1 小見出し とは</h3>、<h3>2-2 小見出し とは</h3>」というようにいくつかの見出しと小見出しで構成し、必ず、見出しは最大4つ、小見出しは各見出しに対して最大2つまでの構成とします。各見出し、小見出し、まとめに説明文は不要です。タイトル、見出し、そして小見出しは検索者が求める情報を整理して記載するようにしてください。見出しと小見出しの冗長性を避けてください。すべてのコンテンツは日本語で書かれている必要があり、検索者の意図を満たすことを目指してください。テキスト：\"{ngram_text}\" キーワード：\"{new_keyword}\"'}
            ],
            temperature=0.7,
            max_tokens=2000
        )
        result = response.choices[0]["message"]["content"].strip()
        print(result)
        return result
    except Exception as e:
        print(f"An error occurred while generating content: {str(e)}")
        return None

if __name__ == "__main__":
    import sys
    new_keyword = sys.argv[1] if len(sys.argv) > 1 else "default_keyword"
    output1_path = os.path.join(os.path.dirname(__file__), "output1.txt")
    
    try:
        with open(output1_path, "r", encoding="utf-8") as f:
            ngram_text = f.read().strip() or "No Ngram text available"
    except Exception as e:
        print(f"Error reading output1.txt: {e}") 
        ngram_text = "No Ngram text available"

    output = generate_headlines_and_subheadings(new_keyword, ngram_text)
    if output:
        with open("output2.txt", "w", encoding="utf-8") as file:
            file.write(output)