import requests
import os
import json

def test_webui_upload():
    base_url = "http://localhost:3015"
    
    # Check if server is running
    try:
        response = requests.get(f"{base_url}/")
        print(f"Server status: {response.status_code}")
    except Exception as e:
        print(f"Cannot connect to server: {e}")
        return
    
    # Test file upload
    test_file = "ocr.pdf"
    if not os.path.exists(test_file):
        print(f"Test file {test_file} not found")
        return
    
    # Upload the file
    files = {'file': (test_file, open(test_file, 'rb'), 'application/pdf')}
    
    try:
        print(f"Uploading {test_file} to web UI...")
        response = requests.post(f"{base_url}/upload", files=files)
        
        print(f"Upload response status: {response.status_code}")
        print(f"Upload response: {response.text}")
        
        if response.status_code == 200:
            result = response.json()
            print(f"Upload result: {json.dumps(result, indent=2)}")
        else:
            print(f"Upload failed with status {response.status_code}")
            
    except Exception as e:
        print(f"Upload error: {e}")
    
    # Also test the documents endpoint to see current documents
    try:
        print("\nChecking current documents...")
        response = requests.get(f"{base_url}/documents")
        print(f"Documents status: {response.status_code}")
        if response.status_code == 200:
            docs = response.json()
            print(f"Current documents: {json.dumps(docs, indent=2)}")
    except Exception as e:
        print(f"Documents check error: {e}")

if __name__ == "__main__":
    test_webui_upload()