53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
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() |