Update article_generator.py

article_generator.py

@@ -8,6 +8,7 @@ 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
 from langchain.agents.tools import Tool
+import gradio as gr
 
 # APIキーの設定
 openai.api_key = os.getenv("OPENAI_API_KEY")
@@ -76,7 +77,7 @@ def clear_state():
     return "状態がクリアされました"
 
 # 見出しを処理する関数
-def process_heading(agent, h2_text, h3_for_this_h2, executed_instructions, research_results, cached_responses):
+def process_heading(agent, h2_text, h3_for_this_h2, cached_responses):
     query = f"{h2_text} {' '.join(h3_for_this_h2)}"
     if query in cached_responses:
         return (query, cached_responses[query])
@@ -149,7 +150,7 @@ def generate_article(editable_output2):
         futures = []
         for h2_text in h2_texts:
             h3_for_this_h2 = [h3 for h3 in h3_texts if h3.startswith(f"{h2_texts.index(h2_text)+1}-")]
-            futures.append(executor.submit(process_heading, agent, h2_text, h3_for_this_h2, executed_instructions, research_results, cached_responses))
+            futures.append(executor.submit(process_heading, agent, h2_text, h3_for_this_h2, cached_responses))
 
         for future in as_completed(futures):
             purpose, response = future.result()
@@ -300,5 +301,4 @@ def continue_generate_article():
     if os.path.exists("state.json"):
         os.remove("state.json")
 
-    return final_result
-
+    return final_result