Update article_generator.py

article_generator.py

@@ -37,7 +37,7 @@ class EnhancedTavilySearchTool:
         params = {
             'api_key': tavily_api_key,
             'query': combined_query,
-            'max_results': 50,  # 必要に応じて結果の数を調整
+            'max_results': 20,  # 必要に応じて結果の数を調整
             'detail_level': 'high',
             'search_depth': 'advanced'
         }
@@ -70,9 +70,9 @@ def expand_h3_sections(soup, preloaded_data):
 
         if key in preloaded_data:
             context = preloaded_data[key]
-            prompt = f"「{h3_text}」に続ける文章を生成してください。こちらが背景情報です：\n{context}"
+            prompt = f"「{h3_text}」の内容を踏まえて、「{h3_text}」に続く、オリジナルの文章を生成してください。あなたの返答は必要なく、そのままテキストを生成してください。こちらが背景情報です：\n{context}"
         else:
-            prompt = f"「{h3_text}」に続ける文章を生成してください。"
+            prompt = f"「{h3_text}」の内容を踏まえて、「{h3_text}」に続く、オリジナルの文章を生成してください。あなたの返答は必要なく、そのままテキストを生成してください。"
         
         expanded_text = generate_text_with_gpt4(prompt)
         new_paragraph = soup.new_tag('p')
@@ -94,11 +94,11 @@ def expand_section_with_gpt4(h2_text, h3_texts, preloaded_data):
         key = f"{h2_text} {h3_text}"
         if key in preloaded_data:
             context = preloaded_data[key]
-            prompt = f"「{h3_text}」について詳しく説明してください。こちらが背景情報です：\n{context}"
+            prompt = f"「{h3_text}」の内容を踏まえて、「{h3_text}」に続く、オリジナルの文章を生成してください。あなたの返答は必要なく、そのまま文章やテキストを生成してください。こちらが背景情報です：\n{context}"
             prompts.append(prompt)
             h3_to_text[h3_text] = prompt  # プロンプトではなく後で置き換えるテキストを格納するための準備
         else:
-            prompt = f"「{h3_text}」について詳しく説明してください。"
+            prompt = f"「{h3_text}」の内容を踏まえて、「{h3_text}」に続く、オリジナルの文章を生成してください。あなたの返答は必要なく、そのまま文章やテキストを生成してください。"
             prompts.append(prompt)
             h3_to_text[h3_text] = prompt
 
@@ -178,7 +178,9 @@ def generate_expanded_article(article_html, h3_to_text, cached_responses):
 
     process_summary_section(soup, cached_responses)  # まとめセクションを特別処理し、キャッシュされたレスポンスを渡す
 
-    return str(soup)
+    final_html = str(soup)
+    latest_content.set(final_html)  # 最新のHTMLコンテンツを状態に保存
+    return final_html
 
 # PlanAndExecuteエージェントをセットアップする関数
 def setup_plan_and_execute_agent():
@@ -203,7 +205,7 @@ def setup_plan_and_execute_agent():
 def generate_text_with_gpt4(prompt):
     response = openai.ChatCompletion.create(
         model="gpt-4o",
-        messages=[{"role": "system", "content": "以下についての詳細な情報をまとめ、適宜箇所書き、もしくは表を使ってオリジナルの内容にしてください。"},
+        messages=[{"role": "system", "content": "以下についての詳細な情報をまとめ、適宜箇所書き、表もしくはグラフを使ってオリジナルの内容にしてください。"},
                   {"role": "user", "content": prompt}],
         temperature=0.7,
         max_tokens=500
@@ -379,4 +381,61 @@ def generate_article(editable_output2):
         f.write(final_html)
 
     print("Article generation complete. Output saved to output3.txt.")
-    return final_html
+    return final_html
+
+# HTMLをMarkdownに変換する関数
+def custom_html_to_markdown(html):
+    soup = BeautifulSoup(html, 'html.parser')
+
+    # 不要なタグの除去
+    for tag in soup(['html', 'body', 'head', 'div']):
+        tag.decompose()
+
+    # タグごとの処理
+    for h in soup.find_all('h1'):
+        h.replace_with(f"# {h.get_text().strip()}\n\n")
+    for h in soup.find_all('h2'):
+        h.replace_with(f"## {h.get_text().strip()}\n\n")
+    for h in soup.find_all('h3'):
+        h.replace_with(f"### {h.get_text().strip()}\n\n")
+    for p in soup.find_all('p'):
+        p.replace_with(f"{p.get_text().strip()}\n\n")
+
+    # 最終的なMarkdownテキストの取得
+    return soup.get_text()
+
+# コンテンツを表示する関数
+def display_content(content, format):
+    if format == "Markdown":
+        return custom_html_to_markdown(content)
+    return content
+
+# Gradioアプリの設定
+def setup_gradio_interface():
+    with gr.Blocks(css='''
+        .gr-markdown h1, .gr-markdown h2, .gr-markdown h3 {
+            word-wrap: break-word;
+            overflow-wrap: break-word;
+        }
+    ''') as app:
+        with gr.Row():
+            format_selector = gr.Radio(choices=["Markdown", "HTML"], label="Display Format", value="Markdown")
+            content_display = gr.Markdown(label="Content", value="")
+            
+            # 初期のHTMLコンテンツを想定
+            initial_html_content = "<h1>Welcome</h1><p>This is a sample paragraph in HTML.</p>"
+            initial_markdown_content = display_content(initial_html_content, "Markdown")
+            
+            # 最新のコンテンツを保持するための状態
+            latest_content = gr.Variable(value=initial_html_content)
+            
+            # 最初のコンテンツを設定
+            content_display.value = initial_markdown_content
+            
+            # フォーマット選択器の変更イベントを設定
+            def update_content(format_choice):
+                # 最新のコンテンツに基づいて表示を更新
+                updated_content = display_content(latest_content.get(), format_choice)
+                return updated_content
+
+            format_selector.change(update_content, inputs=[format_selector], outputs=[content_display])