Yasu777 commited on
Commit
e90c01d
·
verified ·
1 Parent(s): 54b92ec

Update article_generator.py

Browse files
Files changed (1) hide show
  1. article_generator.py +11 -6
article_generator.py CHANGED
@@ -152,13 +152,18 @@ def expand_section_with_gpt4(h2_text, h3_texts, preloaded_data):
152
  key = f"{h2_text} {h3_text}"
153
  if key in preloaded_data:
154
  context = preloaded_data[key]
155
- prompt = f"「{h3_text}」をそのまま記述し、「{h3_text}」に続く本文も生成してください。その本文は「{h3_text}」についての(初期の情報)から得られた詳細な情報を加えた本文としてオリジナルの内容にしてください。以下は初期の情報です:\n{context}"
156
  prompts.append(prompt)
157
  else:
158
- prompts.append(f"「{h3_text}」をそのまま記述し、「{h3_text}」に続く本文も生成してください。その本文は「{h3_text}」についての(初期の情報)から得られた詳細な情報を加えた本文としてオリジナルの内容にしてください。")
159
-
 
 
 
 
160
  expanded_texts = []
161
- with ThreadPoolExecutor(max_workers=len(prompts)) as executor:
 
162
  future_to_prompt = {executor.submit(generate_text_with_gpt4, prompt): prompt for prompt in prompts}
163
  for future in as_completed(future_to_prompt):
164
  prompt = future_to_prompt[future]
@@ -167,9 +172,9 @@ def expand_section_with_gpt4(h2_text, h3_texts, preloaded_data):
167
  expanded_texts.append(expanded_text)
168
  except Exception as e:
169
  error_message = f"Error in generating text for {prompt}: {str(e)}"
170
- print(error_message) # エラーログの出力
171
  expanded_texts.append("Error in text generation.")
172
-
173
  return expanded_texts
174
 
175
  # 記事を拡張する関数
 
152
  key = f"{h2_text} {h3_text}"
153
  if key in preloaded_data:
154
  context = preloaded_data[key]
155
+ prompt = f"「{h3_text}」をそのまま記述し、「{h3_text}」に続く本文も生成してください。その本文は「{h3_text}」についての(初期の情報)から得られた詳細な情報を加えた本文としてください。以下は初期の情報です:\n{context}"
156
  prompts.append(prompt)
157
  else:
158
+ prompts.append(f"「{h3_text}」をそのまま記述し、「{h3_text}」に続く本文も生成してください。その本文は「{h3_text}」についての(初期の情報)から得られた詳細な情報を加えた本文としてください。")
159
+
160
+ if not prompts: # promptsが空の場合
161
+ print("No prompts to process.")
162
+ return []
163
+
164
  expanded_texts = []
165
+ # ThreadPoolExecutormax_workersに最小値を設定
166
+ with ThreadPoolExecutor(max_workers=max(1, len(prompts))) as executor:
167
  future_to_prompt = {executor.submit(generate_text_with_gpt4, prompt): prompt for prompt in prompts}
168
  for future in as_completed(future_to_prompt):
169
  prompt = future_to_prompt[future]
 
172
  expanded_texts.append(expanded_text)
173
  except Exception as e:
174
  error_message = f"Error in generating text for {prompt}: {str(e)}"
175
+ print(error_message)
176
  expanded_texts.append("Error in text generation.")
177
+
178
  return expanded_texts
179
 
180
  # 記事を拡張する関数