Create save_report_to_s3.py

report_generation/save_report_to_s3.py

@@ -0,0 +1,84 @@
+import os
+import base64
+from fastapi import APIRouter, HTTPException
+from pydantic import BaseModel
+from playwright.async_api import async_playwright
+
+
+# ------------------------------
+# FastAPI Models
+# ------------------------------
+
+class ReportRequest(BaseModel):
+    html_code: str   # <-- input from API
+
+
+class ReportResponse(BaseModel):
+    report_generation_status: str
+    report_url: str | None
+    pdf_base64: str  # <-- output base64
+
+
+# ------------------------------
+# Router
+# ------------------------------
+Report_save_router = APIRouter(prefix="/report_save_pdf", tags=["report_save_pdf"])
+
+
+# ------------------------------
+# Playwright: Convert HTML → PDF
+# ------------------------------
+async def html_to_pdf(html_file_path, output_pdf_path):
+    async with async_playwright() as p:
+        browser = await p.chromium.launch()
+        page = await browser.new_page()
+
+        abs_path = os.path.abspath(html_file_path)
+        await page.goto(f"file:///{abs_path}")
+
+        await page.wait_for_load_state("networkidle")
+
+        await page.pdf(
+            path=output_pdf_path,
+            format="A4",
+            print_background=True,
+            margin={"top": "1cm", "right": "1cm", "bottom": "1cm", "left": "1cm"}
+        )
+
+        await browser.close()
+
+
+# ------------------------------
+# API: HTML → PDF (Base64 Output)
+# ------------------------------
+@Report_save_router.post("/save_generated_report", response_model=ReportResponse)
+async def save_generated_report(report_request: ReportRequest):
+
+    try:
+        # 1. Get HTML from request
+        html_content = report_request.html_code
+
+        # 2. Save HTML to temporary file
+        temp_html_file = "temp_report.html"
+        with open(temp_html_file, "w", encoding="utf-8") as f:
+            f.write(html_content)
+
+        # 3. Output PDF path
+        output_pdf_path = "generated_report.pdf"
+
+        # 4. Convert HTML → PDF via Playwright
+        await html_to_pdf(temp_html_file, output_pdf_path)
+
+        # 5. Convert PDF to Base64
+        with open(output_pdf_path, "rb") as pdf:
+            pdf_base64 = base64.b64encode(pdf.read()).decode()
+
+        # 6. Send result
+        return ReportResponse(
+            report_generation_status="success",
+            report_url=None,
+            pdf_base64=pdf_base64
+        )
+
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"PDF generation failed: {str(e)}")