Update app.py

app.py

@@ -1,4 +1,6 @@
 import gradio as gr
+import weasyprint
+import base64  # PDFをData URLに変換するために必要
 
 def display_html(html_code):
     return f"""
@@ -7,6 +9,8 @@ def display_html(html_code):
         </div>
     """
 
+# (テーマとCSSは前の回答のものをそのまま使用)
+# ... (テーマ設定, css変数) ...
 # カスタムテーマ (フォント、テキストスタイルを大幅改良)
 theme = gr.themes.Default(
     primary_hue=gr.themes.colors.indigo,
@@ -83,23 +87,40 @@ a:hover {
 """
 
 
+def generate_pdf(html_code):
+    """
+    HTMLコードからPDFを生成し、Data URLとして返す。
+    """
+    html = weasyprint.HTML(string=html_code)
+    pdf_bytes = html.write_pdf()  # PDFをバイト列として生成
+
+    # バイト列を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
+
+
 with gr.Blocks(theme=theme, title="HTML Viewer", css=css) as demo:
     gr.Markdown(
     """
     # HTML Viewer
 
     入力されたHTMLコードをプレビュー表示します。
-    """)  # 説明を簡潔に
+    """)
     with gr.Row():
         with gr.Column(scale=1):
             input_textbox = gr.Textbox(
-                label="HTMLコード",  # ラベルもシンプルに
+                label="HTMLコード",
                 placeholder="ここにHTMLコードを貼り付けてください...",
                 lines=10
                 )
-            submit_button = gr.Button("表示", variant="primary")
+            with gr.Row():
+                 submit_button = gr.Button("表示", variant="primary")
+                 pdf_button = gr.Button("PDFで保存", variant="secondary") #PDFボタン追加
+
         with gr.Column(scale=3):
-            output_html = gr.HTML(label="プレビュー",)
+            output_html = gr.HTML(label="プレビュー")
+
 
     submit_button.click(
         display_html,
@@ -107,6 +128,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にする
+    )
     #サンプル
     gr.Examples(
         [["<h1>Hello world</h1>"],["""