Update main.py

main.py

@@ -87,4 +87,24 @@ async def get_realtime_trending_searches(pn: str = Query('US', description="Coun
 
 @app.get("/", tags=["Home"])
 def api_home():
-    return {'detail': 'Welcome to Web-Scraping API! Visit https://pvanand-web-scraping.hf.space/docs to test'}    
+    return {'detail': 'Welcome to Web-Scraping API! Visit https://pvanand-web-scraping.hf.space/docs to test'}    
+
+class HTMLRequest(BaseModel):
+    html_content: str
+
+@app.post("/convert-to-pdf")
+async def convert_to_pdf(request: HTMLRequest):
+    try:
+        options = {
+            'page-size': 'A4',
+            'margin-top': '0.75in',
+            'margin-right': '0.75in',
+            'margin-bottom': '0.75in',
+            'margin-left': '0.75in',
+            'encoding': "UTF-8",
+        }
+        
+        pdf = pdfkit.from_string(request.html_content, False, options=options)
+        return Response(content=pdf, media_type="application/pdf")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))