Update article_generator.py

article_generator.py

@@ -26,43 +26,85 @@ class EnhancedTavilySearchTool:
         else:
             raise Exception("Failed to fetch data from Tavily API")
 
-# PlanAndExecute エージェントのセットアップ
-def setup_agent():
-    tavily_search_tool = EnhancedTavilySearchTool()
-    tools = [Tool(name="TavilySearch", func=tavily_search_tool.search, description="Enhanced search tool using Tavily API")]
-    llm = ChatOpenAI(model_name="gpt-3.5-turbo-1106", temperature=0, max_tokens=1000)
+# 実行された指示を追跡するリスト
+executed_instructions = []
+# 調査結果を保存するリスト
+research_results = []
+
+async def main(editable_output2, keyword_id):
+    tavily_search_tool = Tool(
+        name="TavilySearch",
+        func=EnhancedTavilySearchTool().search,
+        description="Enhanced search tool using Tavily API"
+    )
+
+    tools = [tavily_search_tool]
+
+    # PlannerとExecutorの拡張定義
+    model_name = "gpt-3.5-turbo-1106"
+    llm = ChatOpenAI(model_name=model_name, temperature=0, max_tokens=1000)
     planner = load_chat_planner(llm)
     executor = load_agent_executor(llm, tools, verbose=True)
-    agent = PlanAndExecute(planner=planner, executor=executor, verbose=True)
-    return agent
 
-agent = setup_agent()
+    agent = PlanAndExecute(planner=planner, executor=executor, verbose=True)
 
-# 記事本文の生成
-def generate_article(editable_output2):
+    # HTML解析
     soup = BeautifulSoup(editable_output2, 'html.parser')
     h1_text = soup.find('h1').get_text()
     h2_texts = [h2.get_text() for h2 in soup.find_all('h2')]
     h3_texts = [h3.get_text() for h3 in soup.find_all('h3')]
 
+    purpose = f"about {h1_text}, focusing particularly on {' and '.join(h2_texts)} and {' and '.join(h3_texts)}, to investigate the latest information and details"
+
+    # 特定情報の指定
+    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("商品" 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("イベント" 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."
+
+    if instruction not in executed_instructions:
+        raw_output = agent.run(instruction)
+        executed_instructions.append(instruction)
+        response_content = raw_output
+        research_results.append(response_content)
+    else:
+        index = executed_instructions.index(instruction)
+        response_content = research_results[index]
+
     system_message = {
         "role": "system",
         "content": "あなたはプロのライターです。すべての回答を日本語でお願いします。"
     }
 
-    instructions = [
-        f"<h1>{h1_text}</h1> \"{h1_text}\"に関する導入文を日本語で作成してください。直接的なコピーまたは近いフレーズを避けて、オリジナルな内容にしてください。"
-    ]
+    research_summary = "\n".join(research_results)
+    instructions = []
+
+    instructions.append(f"""
+    <h1>{h1_text}</h1> 
+    "{h1_text}"に関する導入文を日本語で作成してください。直接的なコピーまたは近いフレーズを避けて、オリジナルな内容にしてください。""")
+
+    sentences = research_summary.split('。')
 
     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> \"{h2_text}\"に関する導入文を日本語で作成してください。この導入文は、以下の小見出しの内容を考慮してください：{'、'.join(h3_for_this_h2)}。直接的なコピーまたは近いフレーズを避けて、オリジナルな内容にしてください。"
-        )
+        instructions.append(f"""
+            <h2>{h2_text}</h2>
+            "{h2_text}"に関する導入文を日本語で作成してください。この導入文は、以下の小見出しの内容を考慮してください：{"、".join(h3_for_this_h2)}。直接的なコピーまたは近いフレーズを避けて、オリジナルな内容にしてください。""")
         for h3 in h3_for_this_h2:
-            instructions.append(
-                f"<h3>{h3}</h3> \"{h3}\"に関する詳細な内容を日本語で記述してください。オリジナルな内容を心がけてください。"
-            )
+            related_sentences = [sentence for sentence in sentences if h3 in sentence]
+            if related_sentences:
+                content_for_h3 = "。".join(related_sentences) + "。"
+                instructions.append(f"""
+                    <h3>{h3}</h3>
+                    "{h3}"に関する詳細な内容として、以下の情報を日本語で記述してください：{content_for_h3} ここでも、オリジナルな内容を心がけてください。""")
+            else:
+                instructions.append(f"""
+                    <h3>{h3}</h3>
+                    "{h3}"に関する詳細な内容を日本語で記述してください。オリジナルな内容を心がけてください。""")
 
     user_message = {
         "role": "user",
@@ -72,7 +114,7 @@ def generate_article(editable_output2):
     response = openai.ChatCompletion.create(
         model="gpt-4-0125-preview",
         messages=[system_message, user_message],
-        temperature=0.7
+        temperature=0.7,
     )
     result = response.choices[0]["message"]["content"]