Building new logic

screenshot.py

@@ -4,9 +4,14 @@ from playwright.async_api import async_playwright
 import asyncio
 import base64
 import time
-from typing import Optional ,List
+from typing import Optional, List
+import uvicorn
+import logging
+
 app = FastAPI()
 
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger("analyzer")
 
 class AnalysisResult(BaseModel):
     url: str
@@ -17,22 +22,21 @@ class AnalysisResult(BaseModel):
     seo_flags: List[str]
     accessibility_flags: List[str]
     screenshot_base64: str
-
+    status_code: Optional[int] = None
 
 @app.get("/analyze", response_model=AnalysisResult)
 async def analyze_website(url: str):
     try:
         async with async_playwright() as p:
             browser = await p.chromium.launch(headless=True)
-            page = await browser.new_page()
+            context = await browser.new_context()
+            page = await context.new_page()
 
             # Start timing
             start_time = time.time()
-            response = await page.goto(url, timeout=30000)
-            load_time = round(time.time() - start_time, 2)
-
-            # Wait for content
+            response = await page.goto(url, timeout=60000, wait_until='domcontentloaded')
             await page.wait_for_load_state("networkidle")
+            load_time = round(time.time() - start_time, 2)
 
             # Screenshot
             screenshot = await page.screenshot(full_page=True)
@@ -63,6 +67,8 @@ async def analyze_website(url: str):
                     accessibility_flags.append("Image without alt attribute")
                     break
 
+            status_code = response.status if response else None
+
             await browser.close()
 
             return AnalysisResult(
@@ -73,7 +79,12 @@ async def analyze_website(url: str):
                 og_image=og_image,
                 seo_flags=seo_flags,
                 accessibility_flags=accessibility_flags,
-                screenshot_base64=screenshot_base64
+                screenshot_base64=screenshot_base64,
+                status_code=status_code
             )
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        logger.error(f"Analysis failed for {url}: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Error analyzing {url}: {str(e)}")
+
+if __name__ == "__main__":
+    uvicorn.run("main:app", host="0.0.0.0", port=8000, reload=True)