Update third.py

third.py

@@ -2,6 +2,7 @@
 
 import os
 import openai
+import json
 from langchain.chat_models import ChatOpenAI
 from langchain.experimental.plan_and_execute import PlanAndExecute, load_agent_executor, load_chat_planner
 from langchain.llms import OpenAI
@@ -9,6 +10,7 @@ from langchain.utilities import GoogleSearchAPIWrapper
 from langchain.agents.tools import Tool
 from bs4 import BeautifulSoup
 import asyncio
+from datetime import timedelta
 
 # APIキーと検索エンジンIDの設定
 openai.api_key = os.getenv("OPENAI_API_KEY")
@@ -26,8 +28,8 @@ async def main(editable_output2, keyword_id):
 
     tools = [
         Tool(
-            name="Search",
-            func=lambda input_dict: search.run(input_dict) if isinstance(input_dict, str) else search.run(input_dict.get("query", input_dict.get("value"))),
+            name = "Search",
+            func=search.run,
             description="useful for when you need to answer questions about current events"
         ),
     ]
@@ -37,7 +39,11 @@ async def main(editable_output2, keyword_id):
     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, 
+
+    # ここでexecutorの内容を出力して確認
+    print("Executor:", executor)
+
+    agent = PlanAndExecute(planner=planner, executer=executor, verbose=True, 
     suffix='Answer should be in Japanese.')
 
     # editable_output2 is the text of the article structure (h1, h2, h3...).
@@ -57,10 +63,13 @@ async def main(editable_output2, keyword_id):
 
     # Specify the type of information you want to research.
     if "人物" in h1_text or any("人物" in h2_text for h2_text in h2_texts) or any("人物" in h3_text for h3_text in h3_texts):
+        # If the topic is about a person, specify that you want to research their name and career.
         purpose += " including the person's name and career"
     elif "商品" in h1_text or any("商品" in h2_text for h2_text in h2_texts) or any("商品" in h3_text for h3_text in h3_texts):
+        # If the topic is about a product, specify that you want to research the brand name, product name, and price.
         purpose += " including the brand name, product name, and price"
     elif "イベント" in h1_text or any("イベント" in h2_text for h2_text in h2_texts) or any("イベント" in h3_text for h3_text in h3_texts):
+        # If the topic is about an event, specify that you want to research the event's content, schedule, and venue.
         purpose += " including the event's content, schedule, and venue"
 
     # Convert the purpose into an instruction in the form of a question.
@@ -85,12 +94,10 @@ async def main(editable_output2, keyword_id):
     # Prepare the user message
     research_summary = "\n".join(research_results)
     instructions = []
-    formatted_text = []
 
     # 記事タイトルに関する指示
     instructions.append(f'１. "{h1_text}"に関する導入文を作成してください。')
-    formatted_text.append(f"<h1>{h1_text}</h1>")
-    
+
     # research_summaryを文単位に分割
     sentences = research_summary.split('。')
 
@@ -98,7 +105,6 @@ async def main(editable_output2, keyword_id):
     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'{idx+2}. "{h2_text}"に関する導入文を作成してください。この導入文は、以下の小見出しの内容を考慮してください：{"、".join(h3_for_this_h2)}。')
-        formatted_text.append(f"<h2>{h2_text}</h2>")
         for h3 in h3_for_this_h2:
             # h3のテキストをキーワードとして使用し、関連する文を探す
             related_sentences = [sentence for sentence in sentences if h3 in sentence]
@@ -107,11 +113,10 @@ async def main(editable_output2, keyword_id):
             if related_sentences:
                 content_for_h3 = "。".join(related_sentences) + "。"
                 instructions.append(f'次に、"{h3}"に関する詳細な内容として、以下の情報を記述してください：{content_for_h3}')
-                formatted_text.append(f"<h3>{h3}</h3>")
             else:
                 instructions.append(f'次に、"{h3}"に関する詳細な内容を記述してください。')
-                formatted_text.append(f"<h3>{h3}</h3>")
-                
+
+
     user_message = {
         "role": "user",
         "content": "\n".join(instructions)
@@ -125,16 +130,9 @@ async def main(editable_output2, keyword_id):
     )
     result = response.choices[0]["message"]["content"]
 
-    # Split the generated text into sections
-    sections = result.split("\n")
-
-    # Format the generated text with the HTML tags
-    for i, section in enumerate(sections):
-        formatted_text[i] += section
-    
     # Save the generated message to output3.txt
     with open('output3.txt', 'w', encoding='utf-8') as f:
-        f.write("\n".join(formatted_text))
+        f.write(result)
 
     # Print the generated message
-        print("\n".join(formatted_text))
+    print(result)