Update app.py

app.py

@@ -472,6 +472,65 @@ def create_selflinking_pdf(pdf_file="SelfLinking.pdf"):
 
     return pdf_file
 
+def create_pdf_with_images(source_pdf_bytes, output_pdf="ImageLinked.pdf"):
+    """Create a PDF with links to images on numbered items 1-12."""
+    # Get all PNG files in the project directory, sorted by name
+    image_files = sorted(glob.glob("*.png"))
+    
+    if not source_pdf_bytes:
+        st.error("No source PDF provided.")
+        return None
+
+    # Read the source PDF
+    reader = PdfReader(io.BytesIO(source_pdf_bytes))
+    writer = PdfWriter()
+
+    # Copy all pages from source PDF
+    for page in reader.pages:
+        writer.add_page(page)
+
+    # Only process the first page for links
+    page = writer.pages[0]
+    number_pattern = re.compile(r'^\d+\.\s')
+
+    # Find numbered items 1-12 on the page
+    y_positions = []
+    # Assuming the PDF was generated with numbered items at predictable y-coordinates
+    for i in range(1, 13):  # 1 to 12
+        y = 800 - (i * 20)  # Adjust based on create_pdf layout
+        y_positions.append(y)
+
+    # Add links to images
+    for idx, (y, image_file) in enumerate(zip(y_positions, image_files[:12])):
+        if os.path.exists(image_file):
+            # Add "link" text next to numbered item
+            buffer = io.BytesIO()
+            c = canvas.Canvas(buffer)
+            c.setFont("Helvetica", 8)
+            c.drawString(90, y - 5, "link")
+            c.showPage()
+            c.save()
+            buffer.seek(0)
+            text_pdf = PdfReader(buffer)
+            page.merge_page(text_pdf.pages[0])
+            
+            # Add link annotation
+            link = Link(
+                rect=(90, y - 10, 150, y + 10),
+                url=f"file://{os.path.abspath(image_file)}"
+            )
+            writer.add_annotation(page_number=0, annotation=link)
+            buffer.close()
+
+    # Save the modified PDF
+    output_buffer = io.BytesIO()
+    writer.write(output_buffer)
+    output_buffer.seek(0)
+    with open(output_pdf, "wb") as f:
+        f.write(output_buffer.getvalue())
+    
+    return output_buffer.getvalue()
+
 # Streamlit UI
 md_files = [f for f in glob.glob("*.md") if os.path.basename(f) != "README.md"]
 md_options = [os.path.splitext(os.path.basename(f))[0] for f in md_files]
@@ -600,11 +659,33 @@ with st.sidebar:
             mime="audio/mpeg"
         )
     
-    if st.button("📑 Create CrossFile PDFs"):
-        with st.spinner("Creating cross-file linked PDFs..."):
-            source_pdf, target_pdf = create_crossfile_pdfs()
-            st.success(f"Created {source_pdf} and {target_pdf}")
-            for pdf_file in [source_pdf, target_pdf]:
+    col1, col2 = st.columns(2)
+    with col1:
+        if st.button("📑 Create CrossFile PDFs"):
+            with st.spinner("Creating cross-file linked PDFs..."):
+                source_pdf, target_pdf = create_crossfile_pdfs()
+                st.success(f"Created {source_pdf} and {target_pdf}")
+                for pdf_file in [source_pdf, target_pdf]:
+                    with open(pdf_file, "rb") as f:
+                        st.download_button(
+                            label=f"💾 Download {pdf_file}",
+                            data=f.read(),
+                            file_name=pdf_file,
+                            mime="application/pdf"
+                        )
+    
+    with col2:
+        if st.button("🧪 Create SelfLinking PDF"):
+            with st.spinner("Generating self-linking PDF with TOC..."):
+                pdf_file = create_selflinking_pdf()
+                st.success(f"Generated {pdf_file}")
+                with open(pdf_file, "rb") as f:
+                    pdf_bytes = f.read()
+                images = pdf_to_image(pdf_bytes)
+                if images:
+                    st.subheader(f"Preview of {pdf_file}")
+                    for i, img in enumerate(images):
+                        st.image(img, caption=f"{pdf_file} Page {i+1}", use_container_width=True)
                 with open(pdf_file, "rb") as f:
                     st.download_button(
                         label=f"💾 Download {pdf_file}",
@@ -612,23 +693,21 @@ with st.sidebar:
                         file_name=pdf_file,
                         mime="application/pdf"
                     )
-    
-    if st.button("🧪 Create SelfLinking PDF"):
-        with st.spinner("Generating self-linking PDF with TOC..."):
-            pdf_file = create_selflinking_pdf()
-            st.success(f"Generated {pdf_file}")
-            with open(pdf_file, "rb") as f:
-                pdf_bytes = f.read()
-            images = pdf_to_image(pdf_bytes)
-            if images:
-                st.subheader(f"Preview of {pdf_file}")
-                for i, img in enumerate(images):
-                    st.image(img, caption=f"{pdf_file} Page {i+1}", use_container_width=True)
-            with open(pdf_file, "rb") as f:
+
+    if st.button("🖼️ Generate PDF With Images"):
+        with st.spinner("Generating PDF with image links..."):
+            linked_pdf_bytes = create_pdf_with_images(pdf_bytes)
+            if linked_pdf_bytes:
+                st.success("Generated PDF with image links")
+                images = pdf_to_image(linked_pdf_bytes)
+                if images:
+                    st.subheader("Preview of Image-Linked PDF")
+                    for i, img in enumerate(images):
+                        st.image(img, caption=f"Image-Linked PDF Page {i+1}", use_container_width=True)
                 st.download_button(
-                    label=f"💾 Download {pdf_file}",
-                    data=f.read(),
-                    file_name=pdf_file,
+                    label="💾 Download Image-Linked PDF",
+                    data=linked_pdf_bytes,
+                    file_name=f"{prefix} {selected_md}_image_linked.pdf" if selected_md else f"{prefix} image_linked.pdf",
                     mime="application/pdf"
                 )