Update app.py

app.py

@@ -3,35 +3,33 @@ from pdf2image import convert_from_bytes
 from PIL import Image
 import io
 import zipfile
+import sys
+import os
 
-def convert_pdf_to_images(pdf_byte_data, dpi_value):
+def convert_pdf_to_images(pdf_byte_data, dpi_value, poppler_path_value):
     """
     Convert a PDF file in bytes to a list of PIL images using pdf2image.
-    This function requires pdf2image and poppler installed on the system.
-
     Args:
         pdf_byte_data (bytes): PDF file in memory
         dpi_value (int): DPI setting for the conversion
-
+        poppler_path_value (str): poppler path (Windows user only)
     Returns:
         list: A list of PIL Image objects
     """
-    images_list = convert_from_bytes(pdf_byte_data, dpi=dpi_value)
+    images_list = convert_from_bytes(
+        pdf_byte_data,
+        dpi=dpi_value,
+        poppler_path=poppler_path_value if poppler_path_value else None
+    )
     return images_list
 
 def create_zip_from_images(images_list):
     """
     Create an in-memory ZIP file from a list of PIL images.
     Each image is saved as a PNG in the ZIP file.
-
-    Args:
-        images_list (list): A list of PIL Image objects
-
-    Returns:
-        BytesIO: A BytesIO object representing the in-memory ZIP file
     """
     zip_buf = io.BytesIO()
-    with zipfile.ZipFile(zip_buf, 'w', zipfile.ZIP_DEFLATED) as zf:
+    with zipfile.ZipFile(zip_buf, "w", zipfile.ZIP_DEFLATED) as zf:
         for idx, single_image in enumerate(images_list):
             img_buf = io.BytesIO()
             single_image.save(img_buf, format='PNG')
@@ -41,29 +39,43 @@ def create_zip_from_images(images_list):
     return zip_buf
 
 # Streamlit App
-st.title("PDFを高画質PNG画像に変換 (Hugging Face 版)")
+st.title("PDFを高画質PNG画像に変換")
 st.write(
     """
-    このアプリケーションでは、アップロードしたPDFファイルを
-    指定したDPIでPNG画像に変換し、まとめてZIPファイルとしてダウンロードできます。
+    このアプリケーションでは、アップロードしたPDFファイルを高解像度（例：1000 dpi）のPNG画像に変換します。
+    変換後、各ページの画像をダウンロード可能なZIPファイルとして提供します。
     """
 )
 
 uploaded_file = st.file_uploader("PDFファイルをアップロードしてください", type=["pdf"])
 
 if uploaded_file is not None:
-    dpi = st.slider("解像度（dpi）を選択してください", min_value=100, max_value=3000, value=300, step=100)
+    # DPIの設定
+    dpi = st.slider("解像度（dpi）を選択してください", min_value=100, max_value=3000, value=1000, step=100)
+
+    # Windows向けのPopplerパス入力 (Mac/LinuxなどPOPPLERがPATHに通っている環境なら不要)
+    poppler_path_input = None
+    if sys.platform.startswith("win"):
+        poppler_path_input = st.text_input(
+            "Popplerのパスを入力してください（Windowsユーザーのみ）",
+            value=""  # 例: "C:\\poppler-xx\\bin"
+        )
+        if poppler_path_input and not os.path.exists(poppler_path_input):
+            st.warning("指定されたPopplerのパスが存在しません。正しいパスを入力してください。")
 
     if st.button("変換開始"):
         try:
             with st.spinner("PDFを画像に変換中..."):
-                # Convert PDF -> list of images
-                images_converted = convert_pdf_to_images(uploaded_file.read(), dpi)
+                # PDF→画像へ変換
+                images_converted = convert_pdf_to_images(
+                    uploaded_file.read(),
+                    dpi_value=dpi,
+                    poppler_path_value=poppler_path_input
+                )
 
-                # Create zip in memory
+                # ZIPファイル作成
                 zip_file_bytes = create_zip_from_images(images_converted)
 
-                # Download button
                 st.success("変換が完了しました！")
                 st.download_button(
                     label="ZIPファイルをダウンロード",
@@ -72,10 +84,10 @@ if uploaded_file is not None:
                     mime="application/zip"
                 )
 
-                # Show preview of each page
-                st.write("変換された画像プレビュー:")
-                for idx, single_image in enumerate(images_converted):
-                    st.image(single_image, caption=f"ページ {idx+1}", use_column_width=True)
+                # 各ページの画像を表示（オプション）
+                st.write("変換された画像:")
+                for i, image in enumerate(images_converted):
+                    st.image(image, caption=f"ページ {i+1}", use_column_width=True)
 
         except Exception as e:
             st.error(f"変換中にエラーが発生しました: {e}")