Spaces:
Runtime error
Runtime error
Update article_generator.py
Browse files- 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}」についての(初期の情報)から得られた詳細な情報を加えた本文として
|
| 156 |
prompts.append(prompt)
|
| 157 |
else:
|
| 158 |
-
prompts.append(f"「{h3_text}」をそのまま記述し、「{h3_text}」に続く本文も生成してください。その本文は「{h3_text}」についての(初期の情報)から得られた詳細な情報を加えた本文として
|
| 159 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 160 |
expanded_texts = []
|
| 161 |
-
|
|
|
|
| 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 |
+
# ThreadPoolExecutorのmax_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 |
# 記事を拡張する関数
|