Update app.py

app.py

@@ -1,62 +1,54 @@
 import streamlit as st
-from pdf2image import convert_from_bytes
-from PIL import Image
+from wand.image import Image
+from wand.color import Color
 import io
 import zipfile
-import shutil
 
 # タイトルと説明
-st.title("PDFを高画質PNG画像に変換")
+st.title("PDFを高画質PNG画像に変換 (Wand)")
 st.write("""
-このアプリケーションでは、アップロードしたPDFファイルを高解像度（1000 dpi）のPNG画像に変換します。
+このアプリケーションでは、アップロードしたPDFファイルを高解像度（例：1000 dpi）のPNG画像に変換します。
 変換後、各ページの画像をダウンロード可能なZIPファイルとして提供します。
 """)
 
-# popplerのパスを取得
-poppler_path = shutil.which("pdftoppm")
-if poppler_path:
-    st.write(f"Popplerが見つかりました: `{poppler_path}`")
-else:
-    st.error("Popplerが見つかりません。`apt.txt`に`poppler-utils`を追加してください。")
-
 # ファイルアップロード
 uploaded_file = st.file_uploader("PDFファイルをアップロードしてください", type=["pdf"])
 
 if uploaded_file is not None:
     # DPIの設定
     dpi = st.slider("解像度（dpi）を選択してください", min_value=100, max_value=3000, value=1000, step=100)
-
+    
     if st.button("変換開始"):
-        if not poppler_path:
-            st.error("Popplerがインストールされていません。`apt.txt`に`poppler-utils`を追加し、再デプロイしてください。")
-        else:
-            try:
-                with st.spinner("PDFを画像に変換中..."):
-                    # PDFを画像に変換
-                    images = convert_from_bytes(uploaded_file.read(), dpi=dpi, poppler_path=poppler_path)
+        try:
+            with st.spinner("PDFを画像に変換中..."):
+                with Image(blob=uploaded_file.read(), resolution=dpi) as img:
+                    img.compression_quality = 100
+                    img.alpha_channel = False
+                    images = img.sequence
 
                     # ZIPファイルの作成
                     zip_buffer = io.BytesIO()
                     with zipfile.ZipFile(zip_buffer, "w") as zip_file:
-                        for i, image in enumerate(images):
-                            img_byte_arr = io.BytesIO()
-                            image.save(img_byte_arr, format='PNG')
-                            img_byte_arr.seek(0)
-                            zip_file.writestr(f"page_{i+1}.png", img_byte_arr.read())
-
-                    zip_buffer.seek(0)
-                    st.success("変換が完了しました！")
-                    st.download_button(
-                        label="ZIPファイルをダウンロード",
-                        data=zip_buffer,
-                        file_name="pdf_images.zip",
-                        mime="application/zip"
-                    )
-
-                    # 各ページの画像を表示（オプション）
-                    st.write("変換された画像:")
-                    for i, image in enumerate(images):
-                        st.image(image, caption=f"ページ {i+1}", use_column_width=True)
-
-            except Exception as e:
-                st.error(f"変換中にエラーが発生しました: {e}")
+                        for i, page in enumerate(images):
+                            with Image(page) as page_image:
+                                png_blob = page_image.make_blob('png')
+                                zip_file.writestr(f"page_{i+1}.png", png_blob)
+                
+                zip_buffer.seek(0)
+                st.success("変換が完了しました！")
+                st.download_button(
+                    label="ZIPファイルをダウンロード",
+                    data=zip_buffer,
+                    file_name="pdf_images.zip",
+                    mime="application/zip"
+                )
+
+                # 各ページの画像を表示（オプション）
+                st.write("変換された画像:")
+                for i, page in enumerate(images):
+                    with Image(page) as page_image:
+                        img = page_image.convert('png')
+                        st.image(img.make_blob('png'), caption=f"ページ {i+1}", use_column_width=True)
+
+        except Exception as e:
+            st.error(f"変換中にエラーが発生しました: {e}")