Update article_generator.py

article_generator.py

@@ -133,6 +133,17 @@ def setup_plan_and_execute_agent():
     print("PlanAndExecute agent setup complete.")
     return agent
 
+# GPT-4を使用してテキストを生成するヘルパー関数
+def generate_text_with_gpt4(prompt):
+    response = openai.ChatCompletion.create(
+        model="gpt-4-turbo",
+        messages=[{"role": "system", "content": "以下について詳細な情報を生成してください。"},
+                  {"role": "user", "content": prompt}],
+        temperature=0.7,
+        max_tokens=2000
+    )
+    return response.choices[0]["message"]["content"].strip()
+
 # 記事のセクションをGPT-4で拡張する関数
 def expand_section_with_gpt4(h2_text, h3_texts, preloaded_data):
     prompts = []
@@ -146,22 +157,17 @@ def expand_section_with_gpt4(h2_text, h3_texts, preloaded_data):
             prompts.append(f"このテーマ「{h3_text}」に関する詳細情報を加えてください。")
     
     expanded_texts = []
-    for prompt in prompts:
-        try:
-            print(f"Generating text for prompt: {prompt}")
-            response = openai.ChatCompletion.create(
-                model="gpt-4-turbo",
-                messages=[{"role": "system", "content": "以下について詳細な情報を生成してください。"},
-                          {"role": "user", "content": prompt}],
-                temperature=0.7,
-                max_tokens=2000
-            )
-            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)}"
-            print(error_message)  # エラーメッセージを詳細に出力
-            expanded_texts.append("Error in text generation.")
+    with ThreadPoolExecutor(max_workers=len(prompts)) as executor:
+        future_to_prompt = {executor.submit(generate_text_with_gpt4, prompt): prompt for prompt in prompts}
+        for future in as_completed(future_to_prompt):
+            prompt = future_to_prompt[future]
+            try:
+                expanded_text = future.result()
+                expanded_texts.append(expanded_text)
+            except Exception as e:
+                error_message = f"Error in generating text for {prompt}: {str(e)}"
+                print(error_message)
+                expanded_texts.append("Error in text generation.")
     
     return expanded_texts