Update app.py

app.py

@@ -1,9 +1,4 @@
 import gradio as gr
-import weasyprint
-import base64
-import os
-from bs4 import BeautifulSoup  # HTML解析用
-
 
 def display_html(html_code):
     return f"""
@@ -89,37 +84,6 @@ a:hover {
 }
 """
 
-def generate_pdf_and_link(html_code):
-    """
-    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()
-
-    # 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)
-
-    # 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(
     """
@@ -134,14 +98,11 @@ with gr.Blocks(theme=theme, title="HTML Viewer", css=css) as demo:
                 placeholder="ここにHTMLコードを貼り付けてください...",
                 lines=10
                 )
-            with gr.Row():
-                submit_button = gr.Button("表示", variant="secondary")  # スタイルを統一
-                pdf_button = gr.Button("PDFで保存", variant="secondary")
+            submit_button = gr.Button("表示", variant="primary") # スタイルを元に戻す
+
 
         with gr.Column(scale=3):
             output_html = gr.HTML(label="プレビュー")
-            output_pdf = gr.File(label="PDFダウンロード") # ファイル出力コンポーネント
-
 
     submit_button.click(
         display_html,
@@ -149,12 +110,6 @@ with gr.Blocks(theme=theme, title="HTML Viewer", css=css) as demo:
         outputs=output_html
     )
 
-    pdf_button.click(
-        generate_pdf_and_link,
-        inputs=input_textbox,
-        outputs=[output_html, output_pdf]  # プレビューとファイルを出力
-    )
-
     #サンプル
     gr.Examples(
         [["<h1>Hello world</h1>"],["""