Update article_generator.py

article_generator.py

@@ -87,9 +87,9 @@ def generate_article(editable_output2):
     # 特定情報の指定
     if "人物" in h1_text or any("人物" in h2 for h2 in h2_texts) or any("人物" in h3 for h3 in h3_texts):
         purpose += " including the person's name and career"
-    elif "商品" in h1_text or any("商品"にh2 for h2 in h2_texts) or any("商品"にh3 for h3 in h3_texts):
+    elif "商品" in h1_text or any("商品" in h2 for h2 in h2_texts) or any("商品" in h3 for h3 in h3_texts):
         purpose += " including the brand name, product name, and price"
-    elif "イベント" in h1_text or any("イベント"にh2 for h2 in h2_texts) or any("イベント"にh3 for h3 in h3_texts):
+    elif "イベント" in h1_text or any("イベント" in h2 for h2 in h2_texts) or any("イベント" in h3 for h3 in h3_texts):
         purpose += " including the event's content, schedule, and venue"
 
     instruction = f"Can you research {purpose} and include specific details in your response? Please provide the information in Japanese."
@@ -130,7 +130,7 @@ def generate_article(editable_output2):
     # 質問の数を制限
     max_questions_per_h3 = 2
 
-    for idx, h2_text in h2_texts:
+    for idx, h2_text in enumerate(h2_texts):
         h3_for_this_h2 = [h3 for h3 in h3_texts if h3.startswith(f"{idx+1}-")]
         instructions.append(f"""
             <h2>{h2_text}</h2>
@@ -253,4 +253,4 @@ def continue_generate_article():
     if os.path.exists(state_file):
         os.remove(state_file)
 
-    return final_result
+    return final_result