Update app.py

app.py

@@ -13,31 +13,22 @@ class HtmlInput(BaseModel):
     message: str
 
 
-
 async def convert_html_to_image(html, output_file, width=612, height=800, device_scale_factor=2):
-    browser = await launch(
-        headless=True,
-        args=[
-            '--no-sandbox',
-            '--disable-setuid-sandbox',
-            '--disable-gpu',
-            '--disable-dev-shm-usage',  # Try to disable /dev/shm usage
-            '--disable-software-rasterizer',
-            '--disable-accelerated-2d-canvas',
-            '--disable-infobars',
-            '--disable-extensions',
-            '--disable-features=NetworkService',
-            '--disable-features=VizDisplayCompositor',
-            '--window-size=612,800'
-        ]
-    )
-    page = await browser.newPage()
-    await page.setViewport({'width': width, 'height': height, 'deviceScaleFactor': device_scale_factor})
-    await page.setContent(html)
-    # Capture only the viewport without extra white space
-    await page.screenshot({'path': output_file})
-    await browser.close()
-    return output_file  # Return the path of the captured image
+    try:
+        browser = await launch({
+            'headless': True,
+            'args': ['--no-sandbox', '--disable-setuid-sandbox'],
+            'timeout': 60000  # Increase timeout if needed
+        })
+        page = await browser.newPage()
+        await page.setViewport({'width': width, 'height': height, 'deviceScaleFactor': device_scale_factor})
+        await page.setContent(html)
+        await page.screenshot({'path': output_file})
+        await browser.close()
+        return output_file  # Return the path of the captured image
+    except Exception as e:
+        print(f"Error in convert_html_to_image: {e}")
+        raise
 
 # Function to tilt the content randomly
 def tilt_content(html_content):