test_api_pdf.py

#!/usr/bin/env python3
"""Test the API server with the PDF URL to verify the fix."""

import requests
import json
import time

def test_api_pdf_extraction():
    """Test PDF extraction via the API server."""

    # API endpoint
    base_url = "http://localhost:8000"

    # Test data
    pdf_url = "https://daniel.com.pt/shrek.pdf"

    print("Testing PDF extraction via API server")
    print("=" * 50)
    print(f"URL: {pdf_url}")

    # Test the URL endpoint
    try:
        print("\nSending request to API...")
        start_time = time.time()

        response = requests.post(
            f"{base_url}/extract/url",
            json={"url": pdf_url},
            timeout=300  # 5 minute timeout
        )

        end_time = time.time()

        print(f"Response status: {response.status_code}")
        print(f"Total time: {end_time - start_time:.2f} seconds")

        if response.status_code == 200:
            result = response.json()

            print("\nAPI Response:")
            print(f"Success: {result.get('success', 'Unknown')}")
            print(f"Processor used: {result.get('processor_used', 'Unknown')}")
            print(f"Processing time: {result.get('processing_time', 'Unknown')} seconds")
            print(f"Text length: {result.get('text_length', 0)} characters")

            if result.get('error'):
                print(f"Error: {result['error']}")

            if result.get('success') and result.get('extracted_text'):
                print(f"\nExtracted text preview (first 300 characters):")
                print("-" * 30)
                print(result['extracted_text'][:300])
                if len(result['extracted_text']) > 300:
                    print("...")

            # Show file info
            if result.get('file_info'):
                print(f"\nFile info:")
                file_info = result['file_info']
                for key, value in file_info.items():
                    print(f"  {key}: {value}")

        else:
            print(f"API Error: {response.status_code}")
            print(f"Response: {response.text}")

    except requests.exceptions.ConnectionError:
        print("❌ API server is not running!")
        print("Start it with: python run_api.py")
        return False
    except requests.exceptions.Timeout:
        print("❌ Request timed out (took longer than 5 minutes)")
        return False
    except Exception as e:
        print(f"❌ Error: {e}")
        return False

    return True

if __name__ == "__main__":
    print("PDF API Test")
    print("=" * 30)

    success = test_api_pdf_extraction()

    if success:
        print("\n✅ Test completed successfully!")
    else:
        print("\n❌ Test failed!")