Update app.py

app.py

@@ -18,9 +18,9 @@ st.set_page_config(
     initial_sidebar_state="expanded",
 )
 
-st.title("🖼️ Image → PDF • Comic-Book Layout Generator")
+st.title("🖼️ Image → PDF • Full-Page Layout Generator")
 st.markdown(
-    "Upload images, choose aspect ratio, filter by shape, reorder visually, and generate a PDF."
+    "Upload images, filter by orientation, reorder visually, and generate a PDF with each image filling its own page."
 )
 
 # --- Sidebar: Page Settings -----------------------------
@@ -55,30 +55,32 @@ uploaded = st.file_uploader(
 
 ordered_files = []
 if uploaded:
-    # ➊ Collect metadata
     records = []
     for idx, f in enumerate(uploaded):
         im = Image.open(f)
         w, h = im.size
         ar = round(w / h, 2)
         orient = "Square" if 0.9 <= ar <= 1.1 else ("Landscape" if ar > 1.1 else "Portrait")
-        records.append({"filename": f.name, "width": w, "height": h,
-                        "aspect_ratio": ar, "orientation": orient, "order": idx})
+        records.append({
+            "filename": f.name,
+            "width": w,
+            "height": h,
+            "aspect_ratio": ar,
+            "orientation": orient,
+            "order": idx
+        })
     df = pd.DataFrame(records)
 
-    # ➋ Filter by orientation
     dims = st.sidebar.multiselect(
         "Include orientations:", options=["Landscape","Portrait","Square"],
         default=["Landscape","Portrait","Square"]
     )
     df = df[df["orientation"].isin(dims)].reset_index(drop=True)
 
-    # ➌ Show metadata table
     st.markdown("#### Image Metadata")
     st.dataframe(df.style.format({"aspect_ratio": "{:.2f}"}), use_container_width=True)
 
-    # ➍ Visual reorder via Data Editor
-    st.markdown("#### Reorder Panels (drag rows or adjust 'order')")
+    st.markdown("#### Reorder Panels")
     try:
         edited = st.experimental_data_editor(
             df, num_rows="fixed", use_container_width=True
@@ -97,7 +99,6 @@ if uploaded:
         )
         ordered_df = edited.sort_values("order").reset_index(drop=True)
 
-    # Map filenames to ordered files
     name2file = {f.name: f for f in uploaded}
     ordered_files = [name2file[n] for n in ordered_df["filename"] if n in name2file]
 
@@ -109,26 +110,24 @@ def top_n_words(names, n=5):
         words += re.findall(r"\w+", stem.lower())
     return [w for w,_ in Counter(words).most_common(n)]
 
-def make_comic_pdf(images, w_pt, h_pt):
+
+def make_fullpage_pdf(images, w_pt, h_pt):
     buf = io.BytesIO()
-    c = canvas.Canvas(buf, pagesize=(w_pt, h_pt))
-    N = len(images)
-    cols = int(math.ceil(math.sqrt(N)))
-    rows = int(math.ceil(N/cols))
-    pw, ph = w_pt/cols, h_pt/rows
-    for idx, f in enumerate(images):
+    c = canvas.Canvas(buf, pagesize=(w_pt, h_ht := h_pt))
+    for f in images:
         im = Image.open(f)
         iw, ih = im.size
-        tar = pw/ph; ar = iw/ih
-        if ar>tar:
-            nw = int(ih*tar); left=(iw-nw)//2; im=im.crop((left,0,left+nw,ih))
-        else:
-            nh = int(iw/tar); top=(ih-nh)//2; im=im.crop((0,top,iw,top+nh))
-        im = im.resize((int(pw),int(ph)), Image.LANCZOS)
-        col, row = idx%cols, idx//cols
-        x, y = col*pw, h_pt - (row+1)*ph
-        c.drawImage(ImageReader(im), x, y, pw, ph, preserveAspectRatio=False, mask='auto')
-    c.showPage(); c.save(); buf.seek(0)
+        # calculate scale to fit page
+        scale = min(w_pt/iw, h_pt/ih)
+        new_w = int(iw * scale)
+        new_h = int(ih * scale)
+        x = (w_pt - new_w)/2
+        y = (h_pt - new_h)/2
+        im = im.resize((new_w, new_h), Image.LANCZOS)
+        c.drawImage(ImageReader(im), x, y, new_w, new_h, preserveAspectRatio=False, mask='auto')
+        c.showPage()
+    c.save()
+    buf.seek(0)
     return buf.getvalue()
 
 # --- Generate & Download -------------------------------
@@ -141,7 +140,7 @@ if st.button("🎉 Generate PDF"):
         words = top_n_words([f.name for f in ordered_files])
         slug = "-".join(words)
         fname = f"{date_s}-{slug}.pdf"
-        pdf = make_comic_pdf(ordered_files, page_width, page_height)
+        pdf = make_fullpage_pdf(ordered_files, page_width, page_height)
         st.success(f"✅ PDF ready: **{fname}**")
         st.download_button("⬇️ Download PDF", data=pdf, file_name=fname, mime="application/pdf")
         st.markdown("#### Preview")