Update article_generator.py

article_generator.py

@@ -247,31 +247,31 @@ def perform_initial_tavily_search(h2_texts, h3_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}-")]
         if not h3_for_this_h2 and h2_text.strip() != "まとめ":
-            print(f"{h2_text}に対応するh3が見つかりませんでした: インデックス {idx+1}")
+            print(f"No matching h3 elements found for h2: {h2_text} at index {idx+1}")
             continue
         query = f"{h2_text} {' '.join(h3_for_this_h2)}"
         queries.append(query)
     
-    print("Tavily search executing with queries:", queries)
+    print("Executing Tavily search with queries:", queries)
     responses = tavily_search_tool.search(queries)
-
-    # レスポンスの形式を確認
+    if isinstance(responses, str):
+        try:
+            responses = json.loads(responses)  # 文字列をJSONとして解析
+        except json.JSONDecodeError:
+            print("Error: Response string is not valid JSON.")
+            return {}
     if not isinstance(responses, list):
-        print("Error: Response is not a list.")
+        print("Error: Response format is incorrect, expected a list.")
         return {}
 
-    # レスポンスをフィルタリング
     filtered_responses = filter_responses_by_domain(responses, allowed_domains)
     response_dict = {}
     for i, query in enumerate(queries):
-        if i < len(filtered_responses):
-            if isinstance(filtered_responses[i], dict):
-                response_dict[query] = filtered_responses[i].get('title', "No title received")
-            else:
-                print(f"Error: Expected dictionary but received {type(filtered_responses[i])}")
-                response_dict[query] = "Invalid response format"
+        response = filtered_responses[i] if i < len(filtered_responses) else None
+        if isinstance(response, dict):
+            response_dict[query] = response.get('title', 'No title received')
         else:
-            response_dict[query] = "No response received"
+            response_dict[query] = "Invalid response format or no response received"
     return response_dict
 
 def save_preloaded_tavily_data(data):