Create api_service.py

api_service.py

@@ -0,0 +1,200 @@
+from fastapi import FastAPI, HTTPException, Request
+from fastapi.responses import Response
+from pydantic import BaseModel
+import subprocess
+import os
+import tempfile
+import shutil
+import re
+from typing import Optional
+
+app = FastAPI(
+    title="HTML to PDF API",
+    description="Convert HTML to PDF using Puppeteer",
+    version="1.0.0"
+)
+
+class HTMLRequest(BaseModel):
+    html: str
+    aspect_ratio: Optional[str] = "auto"  # "16:9", "1:1", "9:16", or "auto"
+
+def detect_aspect_ratio(html_content):
+    """
+    Detect aspect ratio from HTML content
+    Returns: "16:9", "1:1", or "9:16"
+    """
+    # Check for viewport meta tag
+    viewport_match = re.search(r'<meta[^>]*viewport[^>]*content=["\']([^"\']*)["\']', html_content, re.IGNORECASE)
+    if viewport_match:
+        viewport = viewport_match.group(1).lower()
+        if 'width=device-width' in viewport or 'width=100%' in viewport:
+            if 'orientation=portrait' in viewport:
+                return "9:16"
+            elif 'orientation=landscape' in viewport:
+                return "16:9"
+    
+    # Check for CSS aspect-ratio property
+    aspect_match = re.search(r'aspect-ratio\s*:\s*(\d+)\s*/\s*(\d+)', html_content, re.IGNORECASE)
+    if aspect_match:
+        width = int(aspect_match.group(1))
+        height = int(aspect_match.group(2))
+        ratio = width / height
+        if ratio > 1.5:
+            return "16:9"
+        elif ratio < 0.7:
+            return "9:16"
+        else:
+            return "1:1"
+    
+    # Check for common presentation frameworks
+    if any(keyword in html_content.lower() for keyword in ['reveal.js', 'impress.js', 'slide', 'presentation']):
+        return "16:9"
+    
+    # Default to A4 portrait
+    return "9:16"
+
+def convert_html_to_pdf(html_content, aspect_ratio):
+    """
+    Convert HTML content to PDF using Puppeteer
+    
+    Args:
+        html_content: String containing HTML content
+        aspect_ratio: One of "16:9", "1:1", or "9:16"
+    
+    Returns:
+        Tuple of (pdf_bytes, error_message)
+    """
+    temp_dir = None
+    try:
+        # Create temporary directory
+        temp_dir = tempfile.mkdtemp()
+        
+        # Inject CSS to preserve styles
+        style_injection = """
+        <style>
+            @page {
+                margin: 0;
+            }
+            * {
+                -webkit-print-color-adjust: exact !important;
+                print-color-adjust: exact !important;
+                color-adjust: exact !important;
+            }
+            body {
+                -webkit-print-color-adjust: exact !important;
+                print-color-adjust: exact !important;
+            }
+        </style>
+        """
+        
+        # Insert style injection
+        if '</head>' in html_content:
+            html_content = html_content.replace('</head>', style_injection + '</head>')
+        elif '<body' in html_content:
+            html_content = html_content.replace('<body', style_injection + '<body', 1)
+        else:
+            html_content = style_injection + html_content
+        
+        # Save HTML to temporary file
+        html_file = os.path.join(temp_dir, "input.html")
+        with open(html_file, 'w', encoding='utf-8') as f:
+            f.write(html_content)
+        
+        # Get puppeteer script path
+        script_dir = os.path.dirname(os.path.abspath(__file__))
+        puppeteer_script = os.path.join(script_dir, 'puppeteer_pdf.js')
+        
+        # Run Node.js script
+        result = subprocess.run(
+            ['node', puppeteer_script, html_file, aspect_ratio],
+            capture_output=True,
+            text=True,
+            timeout=60
+        )
+        
+        if result.returncode != 0:
+            return None, f"PDF conversion failed: {result.stderr}"
+        
+        # Read generated PDF
+        pdf_file = html_file.replace('.html', '.pdf')
+        
+        if not os.path.exists(pdf_file):
+            return None, "PDF file was not generated"
+        
+        with open(pdf_file, 'rb') as f:
+            pdf_bytes = f.read()
+        
+        # Cleanup
+        shutil.rmtree(temp_dir, ignore_errors=True)
+        
+        return pdf_bytes, None
+        
+    except subprocess.TimeoutExpired:
+        if temp_dir:
+            shutil.rmtree(temp_dir, ignore_errors=True)
+        return None, "PDF conversion timed out (60 seconds)"
+    except Exception as e:
+        if temp_dir:
+            shutil.rmtree(temp_dir, ignore_errors=True)
+        return None, f"Error: {str(e)}"
+
+@app.get("/")
+async def root():
+    return {
+        "message": "HTML to PDF API",
+        "endpoints": {
+            "POST /convert": "Convert HTML to PDF",
+            "GET /health": "Health check"
+        }
+    }
+
+@app.get("/health")
+async def health():
+    return {"status": "healthy"}
+
+@app.post("/convert")
+async def convert_html(request: HTMLRequest):
+    """
+    Convert HTML to PDF
+    
+    Request body:
+    {
+        "html": "<html>...</html>",
+        "aspect_ratio": "auto"  // Optional: "auto", "16:9", "1:1", or "9:16"
+    }
+    
+    Returns: PDF file as binary response
+    """
+    if not request.html or not request.html.strip():
+        raise HTTPException(status_code=400, detail="HTML content is required")
+    
+    # Determine aspect ratio
+    if request.aspect_ratio == "auto":
+        aspect_ratio = detect_aspect_ratio(request.html)
+    elif request.aspect_ratio in ["16:9", "1:1", "9:16"]:
+        aspect_ratio = request.aspect_ratio
+    else:
+        raise HTTPException(
+            status_code=400, 
+            detail="Invalid aspect_ratio. Must be 'auto', '16:9', '1:1', or '9:16'"
+        )
+    
+    # Convert to PDF
+    pdf_bytes, error = convert_html_to_pdf(request.html, aspect_ratio)
+    
+    if error:
+        raise HTTPException(status_code=500, detail=error)
+    
+    # Return PDF as binary response
+    return Response(
+        content=pdf_bytes,
+        media_type="application/pdf",
+        headers={
+            "Content-Disposition": "attachment; filename=converted.pdf",
+            "X-Aspect-Ratio": aspect_ratio
+        }
+    )
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=8000)