Update article_generator.py

article_generator.py

@@ -147,7 +147,7 @@ def expand_section_with_gpt4(h2_text, h3_texts, preloaded_data):
                 temperature=0.7,
                 max_tokens=2000
             )
-            expanded_text = response.choices[0].message["content"].strip()
+            expanded_text = response.choices[0]["message"]["content"].strip()
             expanded_texts.append(expanded_text)
         except Exception as e:
             error_message = f"Error in generating text for {prompt}: {str(e)}"
@@ -163,7 +163,7 @@ def generate_expanded_article(article_html):
     preloaded_data = load_preloaded_tavily_data()
     
     for h2 in h2_elements:
-        h3_texts = [h3.get_text() for h3 in h2.find_next_siblings('h3', limit=2)]
+        h3_texts = [h3.get_text() for h3 in h2.find_next_siblings('h3')]
         expanded_texts = expand_section_with_gpt4(h2.get_text(), h3_texts, preloaded_data)
         
         for h3, expanded_text in zip(h3_texts, expanded_texts):
@@ -344,13 +344,16 @@ def continue_generate_article():
 
     final_result = "\n".join(results)
 
+    # 生成された初期記事を拡張
+    expanded_article = generate_expanded_article(final_result)
+
     with open("output3.txt", "w", encoding="utf-8") as f:
-        f.write(final_result)
+        f.write(expanded_article)
 
-    print(final_result)  # ログに最終結果を出力
+    print(expanded_article)  # ログに最終結果を出力
 
     # 生成が完了したら状態ファイルを削除
     if os.path.exists("state.json"):
         os.remove("state.json")
 
-    return final_result
+    return expanded_article