Update app.py

app.py

@@ -1,6 +1,9 @@
 import gradio as gr
 import weasyprint
-import base64  # PDFをData URLに変換するために必要
+import base64
+import os
+from bs4 import BeautifulSoup  # HTML解析用
+
 
 def display_html(html_code):
     return f"""
@@ -86,19 +89,36 @@ a:hover {
 }
 """
 
-
-def generate_pdf(html_code):
+def generate_pdf_and_link(html_code):
     """
-    HTMLコードからPDFを生成し、Data URLとして返す。
+    HTMLからPDFを生成し、プレビューとダウンロードリンクを返す。
     """
+    # 1. HTMLプレビューの生成 (display_htmlを再利用)
+    preview_html = display_html(html_code)
+
+    # 2. PDFファイル名の決定
+    soup = BeautifulSoup(html_code, 'html.parser')
+    title_tag = soup.find('title')
+    if title_tag:
+        file_name = title_tag.string.strip() + ".pdf"  # 空白文字を除去
+    else:
+        file_name = "output.pdf"
+
+    # 3. PDF生成
     html = weasyprint.HTML(string=html_code)
-    pdf_bytes = html.write_pdf()  # PDFをバイト列として生成
+    pdf_bytes = html.write_pdf()
+
+    # 4. PDFを一時ファイルとして保存 (Hugging Face Spaceでは/tmpが書き込み可能)
+    pdf_path = os.path.join('/tmp', file_name)
+    with open(pdf_path, 'wb') as f:
+        f.write(pdf_bytes)
 
-    # バイト列をBase64エンコードし、Data URLを作成
-    pdf_base64 = base64.b64encode(pdf_bytes).decode('utf-8')
-    pdf_data_url = f"data:application/pdf;base64,{pdf_base64}"
-    return pdf_data_url
+    # 5. ダウンロードリンクの作成
+    # GradioのFileコンポーネントを使ってダウンロードリンクを作る
+    download_link =  pdf_path
 
+    # 6. プレビューとリンクを返す
+    return preview_html, download_link
 
 with gr.Blocks(theme=theme, title="HTML Viewer", css=css) as demo:
     gr.Markdown(
@@ -115,11 +135,12 @@ with gr.Blocks(theme=theme, title="HTML Viewer", css=css) as demo:
                 lines=10
                 )
             with gr.Row():
-                 submit_button = gr.Button("表示", variant="primary")
-                 pdf_button = gr.Button("PDFで保存", variant="secondary") #PDFボタン追加
+                submit_button = gr.Button("表示", variant="secondary")  # スタイルを統一
+                pdf_button = gr.Button("PDFで保存", variant="secondary")
 
         with gr.Column(scale=3):
             output_html = gr.HTML(label="プレビュー")
+            output_pdf = gr.File(label="PDFダウンロード") # ファイル出力コンポーネント
 
 
     submit_button.click(
@@ -128,12 +149,12 @@ with gr.Blocks(theme=theme, title="HTML Viewer", css=css) as demo:
         outputs=output_html
     )
 
-    # PDFダウンロードボタンの処理
     pdf_button.click(
-        generate_pdf,
-        inputs=input_textbox, #input_textboxをPDFのインプットにする
-        outputs=output_html #出力先をgr.htmlにする
+        generate_pdf_and_link,
+        inputs=input_textbox,
+        outputs=[output_html, output_pdf]  # プレビューとファイルを出力
     )
+
     #サンプル
     gr.Examples(
         [["<h1>Hello world</h1>"],["""