Update main.py

main.py

@@ -290,7 +290,28 @@ async def convert_html_to_image(request: ImageRequest):
         temp_image_path = f"temp/{unique_id}.png"
         
         # 使用DrissionPage进行截图
-        page = ChromiumPage()
+        from drivisionpage import ChromiumPage
+        import os
+        import time
+        
+        # 简单的重试逻辑避免浏览器启动问题
+        max_retries = 3
+        retry_count = 0
+        page = None
+        
+        while retry_count < max_retries:
+            try:
+                # 创建页面实例（使用简单的参数避免环境兼容性问题）
+                page = ChromiumPage()
+                break
+            except Exception as e:
+                retry_count += 1
+                if retry_count >= max_retries:
+                    raise e
+                time.sleep(0.5)  # 等待后重试
+        
+        if page is None:
+            raise HTTPException(status_code=500, detail="浏览器启动失败，请检查环境配置")
         
         try:
             # 设置页面视口大小
@@ -326,7 +347,8 @@ async def convert_html_to_image(request: ImageRequest):
                 page.screenshot(path=temp_image_path)
                 
         finally:
-            page.quit()
+            if page:
+                page.quit()
             
         return ImageResponse(
             success=True,