Update app.py

app.py

@@ -12,8 +12,6 @@ def get_custom_headers(url: str):
         "Connection": "keep-alive",
         "Referer": url,
     }
-    if "jannyai.com" in url:
-        headers["Host"] = "jannyai.com"
     return headers
 
 @app.get("/fetch")
@@ -28,9 +26,7 @@ def fetch_html(url: str):
             page.set_extra_http_headers(headers)
             
             # Modify the goto call with options
-            page.goto(url)
-            if "jannyai.com" in url:
-                page.wait_for_timeout(5000)
+            page.goto(url, timeout=15000, wait_until="networkidle")
             html = page.content()
             browser.close()
         return {"html": html}