Update api.py

api.py

@@ -98,7 +98,8 @@ async def root():
         "endpoints": {
             "POST /convert": "Convert HTML to PDF (file upload)",
             "POST /convert-text": "Convert HTML text to PDF",
-            "GET /health": "Health check"
+            "GET /health": "Health check",
+            "GET /docs": "API documentation (Swagger UI)"
         }
     }
 
@@ -136,203 +137,15 @@ async def convert_file(
         pdf_bytes = convert_html_to_pdf(html_content, aspect_ratio)
         
         # Return PDF file
+        filename = file.filename.replace('.html', '.pdf').replace('.htm', '.pdf')
+        if not filename.endswith('.pdf'):
+            filename += '.pdf'
+            
         return Response(
             content=pdf_bytes,
             media_type="application/pdf",
             headers={
-                "Content-Disposition": f"attachment; filename={file.filename.replace('.html', '.pdf').replace('.htm', '.pdf')}"
-            }
-        )
-        
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Conversion failed: {str(e)}")
-
-@app.post("/convert-text")
-async def convert_text(
-    html: str = Form(...),
-    aspect_ratio: str = Form(default="9:16"),
-    return_base64: bool = Form(default=False)
-):
-    """
-    Convert HTML text to PDF
-    
-    - **html**: HTML content as string
-    - **aspect_ratio**: Page orientation (16:9, 1:1, or 9:16)
-    - **return_base64**: If true, returns base64 encoded PDF in JSON
-    """
-    if aspect_ratio not in ["16:9", "1:1", "9:16"]:
-        raise HTTPException(status_code=400, detail="Invalid aspect ratio. Use: 16:9, 1:1, or 9:16")
-    
-    try:
-        # Convert to PDF
-        pdf_bytes = convert_html_to_pdf(html, aspect_ratio)
-        
-        if return_base64:
-            # Return as JSON with base64 encoded PDF
-            pdf_base64 = base64.b64encode(pdf_bytes).decode('utf-8')
-            return JSONResponse(content={
-                "success": True,
-                "pdf_base64": pdf_base64,
-                "size_bytes": len(pdf_bytes)
-            })
-        else:
-            # Return PDF file directly
-            return Response(
-                content=pdf_bytes,
-                media_type="application/pdf",
-                headers={
-                    "Content-Disposition": "attachment; filename=converted.pdf"
-                }
-            )
-        
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Conversion failed: {str(e)}")
-
-if __name__ == "__main__":
-    import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=7860)from fastapi import FastAPI, File, UploadFile, Form, HTTPException
-from fastapi.responses import Response, JSONResponse
-from fastapi.middleware.cors import CORSMiddleware
-import tempfile
-import shutil
-import os
-import subprocess
-import base64
-from pathlib import Path
-
-app = FastAPI(
-    title="HTML to PDF API",
-    description="Convert HTML to PDF using Puppeteer",
-    version="1.0.0"
-)
-
-# Enable CORS
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-def convert_html_to_pdf(html_content: str, aspect_ratio: str):
-    """Convert HTML content to PDF"""
-    temp_dir = None
-    try:
-        temp_dir = tempfile.mkdtemp()
-        
-        # Style injection for better PDF rendering
-        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>
-        """
-        
-        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 temp 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 conversion
-        result = subprocess.run(
-            ['node', puppeteer_script, html_file, aspect_ratio],
-            capture_output=True,
-            text=True,
-            timeout=60,
-            cwd=script_dir
-        )
-        
-        if result.returncode != 0:
-            raise Exception(f"PDF conversion failed: {result.stderr}")
-        
-        pdf_file = html_file.replace('.html', '.pdf')
-        
-        if not os.path.exists(pdf_file):
-            raise Exception("PDF file was not generated")
-        
-        with open(pdf_file, 'rb') as f:
-            pdf_bytes = f.read()
-        
-        shutil.rmtree(temp_dir, ignore_errors=True)
-        return pdf_bytes
-        
-    except Exception as e:
-        if temp_dir:
-            shutil.rmtree(temp_dir, ignore_errors=True)
-        raise e
-
-@app.get("/")
-async def root():
-    """API root endpoint"""
-    return {
-        "message": "HTML to PDF Conversion API",
-        "version": "1.0.0",
-        "endpoints": {
-            "POST /convert": "Convert HTML to PDF (file upload)",
-            "POST /convert-text": "Convert HTML text to PDF",
-            "GET /health": "Health check"
-        }
-    }
-
-@app.get("/health")
-async def health_check():
-    """Health check endpoint"""
-    return {"status": "healthy", "service": "html-to-pdf-api"}
-
-@app.post("/convert")
-async def convert_file(
-    file: UploadFile = File(...),
-    aspect_ratio: str = Form(default="9:16")
-):
-    """
-    Convert uploaded HTML file to PDF
-    
-    - **file**: HTML file to convert
-    - **aspect_ratio**: Page orientation (16:9, 1:1, or 9:16)
-    """
-    if not file.filename.lower().endswith(('.html', '.htm')):
-        raise HTTPException(status_code=400, detail="File must be HTML (.html or .htm)")
-    
-    if aspect_ratio not in ["16:9", "1:1", "9:16"]:
-        raise HTTPException(status_code=400, detail="Invalid aspect ratio. Use: 16:9, 1:1, or 9:16")
-    
-    try:
-        # Read HTML content
-        content = await file.read()
-        try:
-            html_content = content.decode('utf-8')
-        except UnicodeDecodeError:
-            html_content = content.decode('latin-1')
-        
-        # Convert to PDF
-        pdf_bytes = convert_html_to_pdf(html_content, aspect_ratio)
-        
-        # Return PDF file
-        return Response(
-            content=pdf_bytes,
-            media_type="application/pdf",
-            headers={
-                "Content-Disposition": f"attachment; filename={file.filename.replace('.html', '.pdf').replace('.htm', '.pdf')}"
+                "Content-Disposition": f"attachment; filename={filename}"
             }
         )