Update article_generator.py

article_generator.py

@@ -70,9 +70,9 @@ def expand_h3_sections(soup, preloaded_data):
 
         if key in preloaded_data:
             context = preloaded_data[key]
-            prompt = f"「{h3_text}」について詳しく説明してください。こちらが背景情報です：\n{context}"
+            prompt = f"「{h3_text}」に続ける文章を生成してください。こちらが背景情報です：\n{context}"
         else:
-            prompt = f"「{h3_text}」について詳しく説明してください。"
+            prompt = f"「{h3_text}」に続ける文章を生成してください。"
         
         expanded_text = generate_text_with_gpt4(prompt)
         new_paragraph = soup.new_tag('p')
@@ -202,7 +202,7 @@ def setup_plan_and_execute_agent():
 # GPT-4を使用してテキストを生成するヘルパー関数
 def generate_text_with_gpt4(prompt):
     response = openai.ChatCompletion.create(
-        model="gpt-4-turbo",
+        model="gpt-4o",
         messages=[{"role": "system", "content": "以下についての詳細な情報をまとめ、適宜箇所書き、もしくは表を使ってオリジナルの内容にしてください。"},
                   {"role": "user", "content": prompt}],
         temperature=0.7,
@@ -215,12 +215,17 @@ def perform_initial_tavily_search(h2_texts, h3_texts):
     tavily_search_tool = EnhancedTavilySearchTool()
     queries = []
     
-    for idx, h2_text in enumerate(h2_texts):  # インデックスの取得方法を改善
+    for idx, h2_text in enumerate(h2_texts):
+        # プレフィックスにマッチする h3 を探します。
         h3_for_this_h2 = [h3 for h3 in h3_texts if h3.startswith(f"{idx+1}-")]
+        if not h3_for_this_h2:
+            print(f"No matching h3 elements found for h2: {h2_text} at index {idx+1}")
+            continue  # 該当する h3 がなければスキップ
+        
         query = f"{h2_text} {' '.join(h3_for_this_h2)}"
         queries.append(query)
     
-    print("Performing Tavily search with queries:", queries)  # デバッグ情報追加
+    print("Performing Tavily search with queries:", queries)
     response = tavily_search_tool.search(queries)
     return {query: response[i] for i, query in enumerate(queries)}