Update article_generator.py

article_generator.py

@@ -9,7 +9,6 @@ from langchain_experimental.plan_and_execute import PlanAndExecute, load_agent_e
 from langchain.llms import OpenAI
 from langchain.agents.tools import Tool
 import gradio as gr
-import html2text
 
 # APIキーの設定
 openai.api_key = os.getenv("OPENAI_API_KEY")
@@ -383,17 +382,30 @@ def generate_article(editable_output2):
     return final_html
 
 # HTMLをMarkdownに変換する関数
-def html_to_markdown(html_content):
-    converter = html2text.HTML2Text()
-    converter.ignore_links = False
-    converter.ignore_images = False
-    markdown_content = converter.handle(html_content)
-    return markdown_content
+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 html_to_markdown(content)
+        return custom_html_to_markdown(content)
     return content
 
 # Gradioアプリの設定
@@ -411,9 +423,9 @@ def setup_gradio_interface():
         content_display.value = initial_markdown_content
 
         # フォーマット選択器の変更イベントを設定
-        def update_content(format_choice, content):
+        def update_content(format_choice):
             # フォーマットに基づいてコンテンツを更新
-            updated_content = display_content(content, format_choice)
-            return updated_content
+            updated_content = display_content(initial_html_content, format_choice)
+            content_display.update(value=updated_content)
 
-        format_selector.change(update_content, inputs=[format_selector, content_display], outputs=content_display)
+        format_selector.change(update_content, inputs=format_selector, outputs=content_display)