Update app.py

app.py

@@ -1,11 +1,11 @@
 import streamlit as st
-from wand.image import Image
-from wand.color import Color
+from pdf2image import convert_from_bytes
+from PIL import Image
 import io
 import zipfile
 
 # タイトルと説明
-st.title("PDFを高画質PNG画像に変換 (Wand)")
+st.title("PDFを高画質PNG画像に変換")
 st.write("""
 このアプリケーションでは、アップロードしたPDFファイルを高解像度（例：1000 dpi）のPNG画像に変換します。
 変換後、各ページの画像をダウンロード可能なZIPファイルとして提供します。
@@ -17,23 +17,22 @@ uploaded_file = st.file_uploader("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("変換開始"):
         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, 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)
-                
+                # PDFを画像に変換
+                images = convert_from_bytes(uploaded_file.read(), dpi=dpi)
+
+                # 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(
@@ -45,10 +44,8 @@ if uploaded_file is not None:
 
                 # 各ページの画像を表示（オプション）
                 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)
+                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}")