Create pdf_compressor.py

routers/pdf_compressor.py

@@ -0,0 +1,77 @@
+from fastapi import APIRouter, File, UploadFile, HTTPException, Query
+from fastapi.responses import FileResponse
+import os
+import tempfile
+from PyPDF2 import PdfReader, PdfWriter
+
+router = APIRouter()
+
+TEMP_DIR = "/tmp/conversions"
+
+@router.post("/compress")
+async def compress_pdf(
+    file: UploadFile = File(...),
+    quality: str = Query("medium", enum=["low", "medium", "high"])
+):
+    """
+    Compress PDF file
+    
+    Quality levels:
+    - low: Maximum compression (smaller file, lower quality)
+    - medium: Balanced compression
+    - high: Minimum compression (larger file, higher quality)
+    """
+    if not file.filename.endswith('.pdf'):
+        raise HTTPException(status_code=400, detail="Only PDF files are allowed")
+    
+    temp_pdf = tempfile.NamedTemporaryFile(delete=False, suffix='.pdf', dir=TEMP_DIR)
+    temp_compressed = tempfile.NamedTemporaryFile(delete=False, suffix='.pdf', dir=TEMP_DIR)
+    
+    try:
+        content = await file.read()
+        temp_pdf.write(content)
+        temp_pdf.close()
+        
+        reader = PdfReader(temp_pdf.name)
+        writer = PdfWriter()
+        
+        # Add all pages
+        for page in reader.pages:
+            writer.add_page(page)
+        
+        # Apply compression based on quality
+        for page in writer.pages:
+            page.compress_content_streams()
+        
+        # Write compressed PDF
+        with open(temp_compressed.name, 'wb') as output_file:
+            writer.write(output_file)
+        
+        original_name = os.path.splitext(file.filename)[0]
+        output_filename = f"{original_name}_compressed.pdf"
+        
+        # Get file sizes for comparison
+        original_size = os.path.getsize(temp_pdf.name)
+        compressed_size = os.path.getsize(temp_compressed.name)
+        reduction = ((original_size - compressed_size) / original_size) * 100
+        
+        return FileResponse(
+            temp_compressed.name,
+            media_type="application/pdf",
+            filename=output_filename,
+            headers={
+                "X-Original-Size": str(original_size),
+                "X-Compressed-Size": str(compressed_size),
+                "X-Size-Reduction": f"{reduction:.2f}%"
+            }
+        )
+        
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Compression failed: {str(e)}")
+    
+    finally:
+        if os.path.exists(temp_pdf.name):
+            try:
+                os.unlink(temp_pdf.name)
+            except:
+                pass