Update third.py

third.py

@@ -85,10 +85,12 @@ 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('。')
 
@@ -96,6 +98,7 @@ 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]
@@ -104,9 +107,11 @@ 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)
@@ -120,11 +125,16 @@ 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(result)
+        f.write("\n".join(formatted_text))
 
     # Print the generated message
-    print(result)
-
-# 以下、main関数の呼び出しやその他の処理を追加することができます。
+        print("\n".join(formatted_text))