Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,40 +1,184 @@
-import altair as alt
-import numpy as np
-import pandas as pd
 import streamlit as st
+from pypdf import PdfWriter, PdfReader
+from PIL import Image
+import io
 
-"""
-# Welcome to Streamlit!
-
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
-
-In the meantime, below is an example of what you can do with just a few lines of code:
-"""
-
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
-
-indices = np.linspace(0, 1, num_points)
-theta = 2 * np.pi * num_turns * indices
-radius = indices
-
-x = radius * np.cos(theta)
-y = radius * np.sin(theta)
-
-df = pd.DataFrame({
-    "x": x,
-    "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
-})
-
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))
+# --- PAGE CONFIGURATION ---
+st.set_page_config(
+    page_title="PDF & Image Toolkit",
+    page_icon="📄",
+    layout="centered",
+)
+
+# --- HELPER FUNCTIONS ---
+def merge_pdfs(uploaded_files):
+    """Merges multiple PDF files into one."""
+    merger = PdfWriter()
+    for pdf_file in uploaded_files:
+        # pypdf needs a file object, so we wrap the uploaded bytes in BytesIO
+        merger.append(io.BytesIO(pdf_file.getvalue()))
+    
+    # Write the merged PDF to a BytesIO object in memory
+    output_buffer = io.BytesIO()
+    merger.write(output_buffer)
+    merger.close()
+    return output_buffer.getvalue()
+
+def compress_pdf(uploaded_file):
+    """Performs basic compression on a PDF file."""
+    reader = PdfReader(io.BytesIO(uploaded_file.getvalue()))
+    writer = PdfWriter()
+
+    for page in reader.pages:
+        # This is a basic form of compression
+        page.compress_content_streams()
+        writer.add_page(page)
+
+    # Write the compressed PDF to a BytesIO object
+    output_buffer = io.BytesIO()
+    writer.write(output_buffer)
+    writer.close()
+    return output_buffer.getvalue()
+
+def images_to_pdf(uploaded_files):
+    """Converts a list of images to a single PDF."""
+    images = []
+    for img_file in uploaded_files:
+        img = Image.open(io.BytesIO(img_file.getvalue()))
+        # Convert to RGB to avoid issues with different image modes (e.g., RGBA, P)
+        if img.mode == 'RGBA' or img.mode == 'P':
+            img = img.convert('RGB')
+        images.append(img)
+    
+    if not images:
+        return None
+
+    # Save the images as a PDF to a BytesIO object
+    output_buffer = io.BytesIO()
+    images[0].save(
+        output_buffer, 
+        "PDF" ,
+        resolution=100.0, 
+        save_all=True, 
+        append_images=images[1:]
+    )
+    return output_buffer.getvalue()
+
+
+# --- STREAMLIT UI ---
+
+st.title("📄 PDF & Image Toolkit")
+st.write("A simple web app to manage your documents. All processing is done in your browser and on the server, and your files are not saved.")
+
+# --- SIDEBAR FOR TOOL SELECTION ---
+with st.sidebar:
+    st.header("Tools")
+    tool_choice = st.radio(
+        "Select a tool:",
+        ("Merge PDFs", "Compress PDF", "Convert Images to PDF")
+    )
+
+# --- MAIN PAGE CONTENT ---
+
+if tool_choice == "Merge PDFs":
+    st.subheader("Merge Multiple PDF Files")
+    st.write("Upload two or more PDF files to combine them into a single document.")
+    
+    uploaded_pdfs = st.file_uploader(
+        "Upload PDF files", 
+        type="pdf", 
+        accept_multiple_files=True,
+        key="pdf_merger"
+    )
+
+    if st.button("Merge PDFs") and uploaded_pdfs:
+        if len(uploaded_pdfs) < 2:
+            st.warning("Please upload at least two PDF files to merge.")
+        else:
+            with st.spinner("Merging..."):
+                merged_pdf_bytes = merge_pdfs(uploaded_pdfs)
+                st.success("PDFs merged successfully!")
+                st.download_button(
+                    label="Download Merged PDF",
+                    data=merged_pdf_bytes,
+                    file_name="merged_document.pdf",
+                    mime="application/pdf",
+                )
+
+elif tool_choice == "Compress PDF":
+    st.subheader("Compress a PDF File")
+    st.info("This tool performs basic optimization. It may not significantly reduce the size of all PDFs, especially those that are already optimized or text-heavy.")
+
+    uploaded_pdf = st.file_uploader(
+        "Upload a PDF file", 
+        type="pdf", 
+        accept_multiple_files=False,
+        key="pdf_compressor"
+    )
+
+    if st.button("Compress PDF") and uploaded_pdf:
+        with st.spinner("Compressing..."):
+            compressed_pdf_bytes = compress_pdf(uploaded_pdf)
+            original_size = uploaded_pdf.size
+            compressed_size = len(compressed_pdf_bytes)
+            reduction = ((original_size - compressed_size) / original_size) * 100
+            
+            st.success(f"Compression complete! Size reduced by {reduction:.2f}%.")
+            st.download_button(
+                label="Download Compressed PDF",
+                data=compressed_pdf_bytes,
+                file_name="compressed_document.pdf",
+                mime="application/pdf",
+            )
+
+
+elif tool_choice == "Convert Images to PDF":
+    st.subheader("Convert Images to a Single PDF")
+    st.write("Upload one or more images (JPG, PNG) to combine them into a PDF.")
+
+    uploaded_images = st.file_uploader(
+        "Upload image files",
+        type=["png", "jpg", "jpeg"],
+        accept_multiple_files=True,
+        key="img_converter"
+    )
+
+    if st.button("Convert to PDF") and uploaded_images:
+        with st.spinner("Converting..."):
+            pdf_from_images_bytes = images_to_pdf(uploaded_images)
+            st.success("Images converted to PDF successfully!")
+            st.download_button(
+                label="Download PDF",
+                data=pdf_from_images_bytes,
+                file_name="images_converted.pdf",
+                mime="application/pdf",
+            )
+
+st.sidebar.markdown("---")
+st.sidebar.info("App built with [Streamlit](https://streamlit.io) and hosted on [Hugging Face Spaces](https://huggingface.co/spaces).")
+
+---
+
+### How to Deploy on Hugging Face Spaces (in 5 steps)
+
+1.  **Create a Hugging Face Account:** If you don't have one, sign up at [huggingface.co](https://huggingface.co).
+2.  **Create a New Space:**
+    *   Click on your profile picture and select "New Space".
+    *   Give it a **Space name** (e.g., `pdf-toolkit`).
+    *   Choose a license (e.g., `mit`).
+    *   Select **Streamlit** as the Space SDK.
+    *   Click "Create Space".
+
+3.  **Add the Files:**
+    *   You will be taken to your new Space repository. Click on the **"Files"** tab.
+    *   Click "Add file" -> "Create new file".
+    *   Name the file `requirements.txt`. Paste the content for `requirements.txt` from above and commit the new file.
+    *   You will see an `app.py` file was already created for you. Click on it, then click the pencil icon to **edit** it.
+    *   **Delete the existing placeholder content** and paste the entire `app.py` code from above. Commit your changes.
+
+4.  **Wait for it to Build:**
+    *   Go back to the "App" tab. You will see a "Building" status. Hugging Face is now installing Streamlit, PyPDF, and Pillow based on your `requirements.txt` file.
+    *   This usually takes 1-2 minutes.
+
+5.  **Use Your App!**
+    *   Once it's finished building, your application will be live and ready to use at its public URL (e.g., `https://huggingface.co/spaces/Abhisksks/Pdf_editor`).