Update article_generator.py

article_generator.py

@@ -62,7 +62,10 @@ def expand_h3_sections(soup, preloaded_data):
     h3_elements = soup.find_all('h3')
     for h3 in h3_elements:
         h3_text = h3.get_text(strip=True)
-        section_id = h3['id']
+        section_id = h3.get('id', None)
+        if section_id is None:
+            print(f"Warning: h3 element '{h3_text}' has no ID.")
+            continue
         key = f"{h3_text} {section_id}"
 
         if key in preloaded_data:
@@ -108,7 +111,10 @@ def expand_section_with_gpt4(h2_text, h3_texts, preloaded_data):
     with ThreadPoolExecutor(max_workers=max(1, len(prompts))) as executor:
         future_to_prompt = {executor.submit(generate_text_with_gpt4, prompt): h3_text for prompt, h3_text in zip(prompts, h3_texts)}
         for future in as_completed(future_to_prompt):
-            h3_text = future_to_prompt[future]
+            h3_text = future_to_prompt.get(future)
+            if h3_text is None:
+                print("Error: Future not found in future_to_prompt")
+                continue
             try:
                 expanded_text = future.result()
                 expanded_texts.append(expanded_text)
@@ -196,7 +202,7 @@ def setup_plan_and_execute_agent():
 # GPT-4を使用してテキストを生成するヘルパー関数
 def generate_text_with_gpt4(prompt):
     response = openai.ChatCompletion.create(
-        model="gpt-4o",
+        model="gpt-4-turbo",
         messages=[{"role": "system", "content": "以下についての詳細な情報をまとめ、適宜箇所書き、もしくは表を使ってオリジナルの内容にしてください。"},
                   {"role": "user", "content": prompt}],
         temperature=0.7,