Update article_generator.py

article_generator.py

@@ -1,17 +1,17 @@
-import os
-import openai
 import gradio as gr
 import requests
 from bs4 import BeautifulSoup
-from langchain.llms import OpenAI
+import openai
 from langchain.chat_models import ChatOpenAI
 from langchain_experimental.plan_and_execute import PlanAndExecute, load_agent_executor, load_chat_planner
 from langchain.agents.tools import Tool
+import os
 
-# 環境変数からAPIキーを設定
+# APIキーの設定
 openai.api_key = os.getenv("OPENAI_API_KEY")
 tavily_api_key = os.getenv('TAVILY_API_KEY')
 
+# Tavily APIのカスタムツールを定義
 class EnhancedTavilySearchTool:
     def search(self, query):
         params = {
@@ -26,37 +26,65 @@ class EnhancedTavilySearchTool:
         else:
             raise Exception("Failed to fetch data from Tavily API")
 
-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]
-
+# 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)
     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()
+
+# 記事本文の生成
+def generate_article(editable_output2):
     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')]
+
+    system_message = {
+        "role": "system",
+        "content": "あなたはプロのライターです。すべての回答を日本語でお願いします。"
+    }
+
+    instructions = [
+        f"<h1>{h1_text}</h1> \"{h1_text}\"に関する導入文を日本語で作成してください。直接的なコピーまたは近いフレーズを避けて、オリジナルな内容にしてください。"
+    ]
+
+    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)}。直接的なコピーまたは近いフレーズを避けて、オリジナルな内容にしてください。"
+        )
+        for h3 in h3_for_this_h2:
+            instructions.append(
+                f"<h3>{h3}</h3> \"{h3}\"に関する詳細な内容を日本語で記述してください。オリジナルな内容を心がけてください。"
+            )
+
+    user_message = {
+        "role": "user",
+        "content": "\n".join(instructions)
+    }
+
+    response = openai.ChatCompletion.create(
+        model="gpt-4-0125-preview",
+        messages=[system_message, user_message],
+        temperature=0.7
+    )
+    result = response.choices[0]["message"]["content"]
 
-    # 実行と結果の取得
-    result = agent.run(f"Research about {h1_text}")
     return result
 
-# Gradioインターフェースの設定
 with gr.Blocks() as app:
-    with gr.Column():
-        input_html = gr.Textbox(label="Editable HTML Output", placeholder="HTML content here...", lines=10)
-        keyword_id_input = gr.Textbox(label="Keyword ID", placeholder="Enter Keyword ID...")
-        generate_button = gr.Button("Generate")
-        output_article = gr.Textbox(label="Generated Article", lines=20)
+    editable_output2 = gr.Textbox(label="編集可能な記事構成", placeholder="HTML content here...", lines=10)
+    final_article = gr.Textbox(label="最終的な記事本文", lines=20, placeholder="Generated article content will appear here.")
+    generate_button = gr.Button("記事を生成")
 
     generate_button.click(
-        fn=main,
-        inputs=[input_html, keyword_id_input],
-        outputs=output_article
+        fn=generate_article,
+        inputs=[editable_output2],
+        outputs=[final_article]
     )