Feat/docker PDF strategy

crawl4ai/__init__.py

@@ -10,6 +10,9 @@
     LXMLWebScrapingStrategy,
     WebScrapingStrategy,  # Backward compatibility alias
 )
+
+from .processors.pdf import PDFCrawlerStrategy, PDFContentScrapingStrategy
+
 from .async_logger import (
     AsyncLoggerBase,
     AsyncLogger,
@@ -128,6 +131,8 @@
     "BFSDeepCrawlStrategy",
     "BestFirstCrawlingStrategy",
     "DFSDeepCrawlStrategy",
+    "PDFCrawlerStrategy",
+    "PDFContentScrapingStrategy",
     "FilterChain",
     "URLPatternFilter",
     "ContentTypeFilter",

deploy/docker/api.py

@@ -13,8 +13,12 @@
 from fastapi import HTTPException, Request, status
 from fastapi.background import BackgroundTasks
 from fastapi.responses import JSONResponse
+from fastapi.encoders import jsonable_encoder
+
 from redis import asyncio as aioredis
 
+from utils import is_pdf_url
+
 from crawl4ai import (
     AsyncWebCrawler,
     CrawlerRunConfig,
@@ -31,6 +35,10 @@
     BM25ContentFilter,
     LLMContentFilter
 )
+
+from crawl4ai.processors.pdf import PDFCrawlerStrategy, PDFContentScrapingStrategy
+from crawl4ai.async_configs import to_serializable_dict
+
 from crawl4ai.markdown_generation_strategy import DefaultMarkdownGenerator
 from crawl4ai.content_scraping_strategy import LXMLWebScrapingStrategy
 
@@ -431,6 +439,18 @@ async def handle_crawl_request(
         urls = [('https://' + url) if not url.startswith(('http://', 'https://')) and not url.startswith(("raw:", "raw://")) else url for url in urls]
         browser_config = BrowserConfig.load(browser_config)
         crawler_config = CrawlerRunConfig.load(crawler_config)
+
+        is_pdf_flags = await asyncio.gather(*(is_pdf_url(url) for url in urls))
+        is_pdf = any(is_pdf_flags)
+        crawler_strategy = PDFCrawlerStrategy() if is_pdf else None
+
+        if is_pdf and crawler_config.scraping_strategy is None:
+            # Default strategy if not set.
+            crawler_config.scraping_strategy = PDFContentScrapingStrategy(
+                extract_images=False,
+                save_images_locally=False,
+                batch_size=2
+            )
 
         dispatcher = MemoryAdaptiveDispatcher(
             memory_threshold_percent=config["crawler"]["memory_threshold_percent"],
@@ -440,7 +460,7 @@ async def handle_crawl_request(
         )
 
         from crawler_pool import get_crawler
-        crawler = await get_crawler(browser_config)
+        crawler = await get_crawler(browser_config, crawler_strategy)
 
         # crawler: AsyncWebCrawler = AsyncWebCrawler(config=browser_config)
         # await crawler.start()
@@ -476,7 +496,8 @@ async def handle_crawl_request(
             # If PDF exists, encode it to base64
             if result_dict.get('pdf') is not None:
                 result_dict['pdf'] = b64encode(result_dict['pdf']).decode('utf-8')
-            processed_results.append(result_dict)
+
+            processed_results.append(to_serializable_dict(result_dict))
 
         return {
             "success": True,
@@ -521,8 +542,19 @@ async def handle_stream_crawl_request(
         # browser_config.verbose = True # Set to False or remove for production stress testing
         browser_config.verbose = False
         crawler_config = CrawlerRunConfig.load(crawler_config)
-        crawler_config.scraping_strategy = LXMLWebScrapingStrategy()
         crawler_config.stream = True
+
+        is_pdf_flags = await asyncio.gather(*(is_pdf_url(url) for url in urls))
+        is_pdf = any(is_pdf_flags)
+        crawler_strategy = PDFCrawlerStrategy() if is_pdf else None
+
+        if is_pdf and crawler_config.scraping_strategy is None:
+            # Default strategy if not set
+            crawler_config.scraping_strategy = PDFContentScrapingStrategy(
+                extract_images=True,
+                save_images_locally=False,
+                batch_size=2
+            )
 
         dispatcher = MemoryAdaptiveDispatcher(
             memory_threshold_percent=config["crawler"]["memory_threshold_percent"],
@@ -532,7 +564,7 @@ async def handle_stream_crawl_request(
         )
 
         from crawler_pool import get_crawler
-        crawler = await get_crawler(browser_config)
+        crawler = await get_crawler(browser_config, crawler_strategy)
 
         # crawler = AsyncWebCrawler(config=browser_config)
         # await crawler.start()

deploy/docker/crawler_pool.py

@@ -1,11 +1,11 @@
 # crawler_pool.py  (new file)
 import asyncio, json, hashlib, time, psutil
 from contextlib import suppress
-from typing import Dict
+from typing import Dict, Optional
 from crawl4ai import AsyncWebCrawler, BrowserConfig
-from typing import Dict
 from utils import load_config 
 
+
 CONFIG = load_config()
 
 POOL: Dict[str, AsyncWebCrawler] = {}
@@ -15,20 +15,31 @@
 MEM_LIMIT  = CONFIG.get("crawler", {}).get("memory_threshold_percent", 95.0)   # % RAM – refuse new browsers above this
 IDLE_TTL  = CONFIG.get("crawler", {}).get("pool", {}).get("idle_ttl_sec", 1800)   # close if unused for 30 min
 
-def _sig(cfg: BrowserConfig) -> str:
-    payload = json.dumps(cfg.to_dict(), sort_keys=True, separators=(",",":"))
-    return hashlib.sha1(payload.encode()).hexdigest()
+def _sig(cfg: BrowserConfig, crawler_strategy: Optional[object]  = None) -> str:
+    """
+    Generate a unique signature for a crawler based on browser config
+    and optional crawler strategy. This ensures that crawlers with
+    different strategies (e.g., PDF) are stored separately in the pool.
+    """
+    payload = cfg.to_dict()
+
+    if crawler_strategy is not None:
+        payload["strategy"] = crawler_strategy.__class__.__name__
+
+    json_payload = json.dumps(payload, sort_keys=True, separators=(",", ":"))
+    return hashlib.sha1(json_payload.encode()).hexdigest()
+
 
-async def get_crawler(cfg: BrowserConfig) -> AsyncWebCrawler:
+async def get_crawler(cfg: BrowserConfig, crawler_strategy: Optional[object] = None) -> AsyncWebCrawler:
     try:
-        sig = _sig(cfg)
+        sig = _sig(cfg, crawler_strategy=crawler_strategy)
         async with LOCK:
             if sig in POOL:
                 LAST_USED[sig] = time.time();  
                 return POOL[sig]
             if psutil.virtual_memory().percent >= MEM_LIMIT:
                 raise MemoryError("RAM pressure – new browser denied")
-            crawler = AsyncWebCrawler(config=cfg, thread_safe=False)
+            crawler = AsyncWebCrawler(config=cfg, thread_safe=False, crawler_strategy=crawler_strategy)
             await crawler.start()
             POOL[sig] = crawler; LAST_USED[sig] = time.time()
             return crawler
@@ -44,6 +55,7 @@ async def get_crawler(cfg: BrowserConfig) -> AsyncWebCrawler:
             POOL.pop(sig, None)
             LAST_USED.pop(sig, None)
         # If we failed to start the browser, we should remove it from the pool
+
 async def close_all():
     async with LOCK:
         await asyncio.gather(*(c.close() for c in POOL.values()), return_exceptions=True)

deploy/docker/utils.py

@@ -2,6 +2,7 @@
 import logging
 import yaml
 import os
+import httpx
 from datetime import datetime
 from enum import Enum
 from pathlib import Path
@@ -124,4 +125,32 @@ def verify_email_domain(email: str) -> bool:
         records = dns.resolver.resolve(domain, 'MX')
         return True if records else False
     except Exception as e:
-        return False
+        return False
+
+async def is_pdf_url(url: str) -> bool:
+    """
+    Check if a URL points to a PDF using httpx:
+    - Check extension
+    - Check Content-Type via HEAD request
+    - Check first 5 bytes (magic number) if needed
+    """
+    if url.lower().endswith(".pdf"):
+        return True
+
+    try:
+        async with httpx.AsyncClient(follow_redirects=True) as client:
+            # HEAD request to check Content-Type
+            head_resp = await client.head(url)
+            content_type = head_resp.headers.get("content-type", "").lower()
+            if "application/pdf" in content_type:
+                return True
+
+            # Fallback: GET first 5 bytes to check PDF magic number
+            get_resp = await client.get(url, headers={"Range": "bytes=0-4"})
+            if get_resp.status_code in (200, 206):  # 206 Partial Content
+                return get_resp.content.startswith(b"%PDF-")
+    except Exception:
+        return False
+
+    return False
+

docs/examples/docker/demo_docker_api.py

@@ -27,6 +27,7 @@
 FORMS_URL = "https://httpbin.org/forms/post"  # For JS demo
 BOOKS_URL = "http://books.toscrape.com/"  # For CSS extraction
 PYTHON_URL = "https://python.org"  # For deeper crawl
+PDF_URL = "https://arxiv.org/pdf/2310.06825" # For PDF demo
 # Use the same sample site as deep crawl tests for consistency
 DEEP_CRAWL_BASE_URL = os.getenv(
     "DEEP_CRAWL_TEST_SITE", "https://docs.crawl4ai.com/samples/deepcrawl/")
@@ -1261,6 +1262,71 @@ async def demo_config_dump_invalid(client: httpx.AsyncClient):
         console.print(
             f"[bold red]Unexpected error during invalid test:[/] {e}")
 
+# 10. Crawl PDF
+
+async def demo_pdf_crawl(client: httpx.AsyncClient):
+    payload = {
+        "urls": [PDF_URL],
+        "crawler_config": {
+            "type": "CrawlerRunConfig",
+            "params": {
+                "cache_mode": "BYPASS",
+                "scraping_strategy": {
+                    "type": "PDFContentScrapingStrategy",
+                    "params": {
+                        "extract_images": False,
+                        "save_images_locally": False,
+                        "batch_size": 2
+                    }
+                }
+            }
+        }
+    }
+
+    resp = await client.post("/crawl", json=payload)
+    resp.raise_for_status()
+    data = resp.json()
+    print("=== Demo: PDF Crawl ===")
+    print("Success:", data.get("success"))
+    print("Number of results:", len(data.get("results", [])))
+    if data.get("results"):
+        first = data["results"][0]
+        text_snippet = (first.get("text") or "")[:500]
+        print("Extracted text (first 500 chars):")
+        print(text_snippet)
+
+# 11. Crawl PDF stream
+
+async def demo_pdf_crawl_stream(client: httpx.AsyncClient):
+    """
+    Demo: Crawl a PDF and stream the extracted text content.
+    """
+    payload = {
+        "urls": [PDF_URL],
+        "crawler_config": {
+            "type": "CrawlerRunConfig",
+            "params": {
+                "stream": True,
+                "cache_mode": "BYPASS",
+                "scraping_strategy": {  # <-- Default strategy if not set
+                    "type": "PDFContentScrapingStrategy",
+                    "params": {
+                        "extract_images": False,     
+                        "save_images_locally": False,
+                        "batch_size": 2              
+                    }
+                }
+            }
+        }
+    }
+
+    await stream_request(
+        client,
+        "/crawl/stream",
+        payload,
+        "Demo PDF: Streaming PDF Crawl"
+    )
+
 
 # --- Update Main Runner to include new demo ---
 async def main_demo():
@@ -1294,6 +1360,9 @@ async def main_demo():
         # await demo_deep_with_llm_extraction(client)
         # await demo_deep_with_proxy(client)  # Skips if no PROXIES env var
         # await demo_deep_with_ssl(client)   # Added the new demo
+
+        # await demo_pdf_crawl_stream(client)
+        # await demo_pdf_crawl(client)
 
         # --- Helper endpoints ---
         await demo_markdown_endpoint(client)