Update app.py

app.py

@@ -1,11 +1,16 @@
 import streamlit as st
 from PIL import Image
 from fpdf import FPDF
+import os
 
 # Function to process a single image with user-provided sequence number
 def process_image(image_file, sequence_number):
-    image = Image.open(image_file)
-    return image, sequence_number
+    try:
+        image = Image.open(image_file)
+        return image, sequence_number
+    except Exception as e:
+        st.error(f"Failed to open image: {e}")
+        return None, None
 
 # Function to generate the PDF using fpdf
 def generate_pdf(images_with_sequence):
@@ -14,12 +19,16 @@ def generate_pdf(images_with_sequence):
     images_with_sequence.sort(key=lambda x: x[1])  # Sort by sequence number
 
     pdf = FPDF()
-    for image, _ in images_with_sequence:
-        image.save("temp_image.jpg")
+    for i, (image, _) in enumerate(images_with_sequence):
+        image.save(f"temp_image_{i}.jpg")
         pdf.add_page()
-        pdf.image("temp_image.jpg", x=10, y=10, w=190)
+        pdf.image(f"temp_image_{i}.jpg", x=10, y=10, w=190)
     
-    pdf.output(pdf_name)
+    try:
+        pdf.output(pdf_name)
+    except Exception as e:
+        st.error(f"Failed to generate PDF: {e}")
+        return None
     return pdf_name
 
 st.title("Image to PDF Converter (Multiple Images)")
@@ -37,7 +46,8 @@ if uploaded_images:
             default_index = i  # Set default selection of sequence as the order of upload
             sequence_number = st.selectbox(f"Sequence Number for Image {i+1}", sequence_options, index=default_index)
             processed_image, processed_sequence = process_image(image_file, int(sequence_number))
-            images_with_sequence.append((processed_image, processed_sequence))
+            if processed_image:
+                images_with_sequence.append((processed_image, processed_sequence))
 
         if st.button("Generate PDF"):
             pdf_name = generate_pdf(images_with_sequence)
@@ -45,6 +55,9 @@ if uploaded_images:
                 st.success(f"PDF generated! Download: {pdf_name}")
                 with open(pdf_name, "rb") as pdf_file:
                     st.download_button("Download PDF", pdf_file, file_name=pdf_name)
+                for i in range(len(images_with_sequence)):
+                    if os.path.exists(f"temp_image_{i}.jpg"):
+                        os.remove(f"temp_image_{i}.jpg")
                 os.remove(pdf_name)  # Remove the file after download
 else:
-    st.write("Please upload images to continue")
+    st.write("Please upload images to continue")