| """ |
| Test script for InklyAI Web UI |
| """ |
|
|
| import requests |
| import json |
| import os |
| import time |
| from pathlib import Path |
|
|
|
|
| def test_web_ui(): |
| """Test the web UI functionality.""" |
| print("🧪 Testing InklyAI Web UI") |
| print("=" * 50) |
| |
| base_url = "http://localhost:8080" |
| |
| |
| print("\n1. Testing health check...") |
| try: |
| response = requests.get(f"{base_url}/api/health") |
| health = response.json() |
| print(f" ✅ Health status: {health['status']}") |
| print(f" ✅ Agents registered: {health['agents_registered']}") |
| except Exception as e: |
| print(f" ❌ Health check failed: {e}") |
| return False |
| |
| |
| print("\n2. Testing agent listing...") |
| try: |
| response = requests.get(f"{base_url}/api/agents") |
| agents = response.json() |
| print(f" ✅ Found {agents['total_agents']} agents") |
| for agent in agents['agents']: |
| print(f" - {agent['agent_id']} ({'Active' if agent['is_active'] else 'Inactive'})") |
| except Exception as e: |
| print(f" ❌ Agent listing failed: {e}") |
| |
| |
| print("\n3. Testing signature verification...") |
| if os.path.exists('data/samples/john_doe_1.png') and os.path.exists('data/samples/john_doe_2.png'): |
| try: |
| |
| with open('data/samples/john_doe_1.png', 'rb') as f1, open('data/samples/john_doe_2.png', 'rb') as f2: |
| files = { |
| 'signature1': ('john_doe_1.png', f1, 'image/png'), |
| 'signature2': ('john_doe_2.png', f2, 'image/png') |
| } |
| data = {'agent_id': 'Agent_01'} |
| |
| response = requests.post(f"{base_url}/api/verify", files=files, data=data) |
| result = response.json() |
| |
| if result['success']: |
| print(f" ✅ Verification successful") |
| print(f" ✅ Similarity: {result['similarity_score']:.3f}") |
| print(f" ✅ Verified: {result['is_verified']}") |
| else: |
| print(f" ❌ Verification failed: {result['error']}") |
| except Exception as e: |
| print(f" ❌ Signature verification failed: {e}") |
| else: |
| print(" ⚠️ Sample data not found. Run demo.py first to create sample signatures.") |
| |
| |
| print("\n4. Testing agent statistics...") |
| try: |
| response = requests.get(f"{base_url}/api/stats") |
| stats = response.json() |
| |
| if stats['success']: |
| print(f" ✅ Statistics loaded successfully") |
| for agent_id, agent_stats in stats['stats'].items(): |
| print(f" - {agent_id}: {agent_stats['total_verifications']} verifications, " |
| f"success rate: {agent_stats['success_rate']:.1%}") |
| else: |
| print(f" ❌ Stats failed: {stats['error']}") |
| except Exception as e: |
| print(f" ❌ Agent stats failed: {e}") |
| |
| print("\n🎉 Web UI tests completed!") |
| print(f"\n🌐 Access the web UI at: {base_url}") |
| print(f"📊 Agent management at: {base_url}/agents") |
| |
| return True |
|
|
|
|
| def check_web_server(): |
| """Check if the web server is running.""" |
| try: |
| response = requests.get("http://localhost:8080/api/health", timeout=5) |
| return response.status_code == 200 |
| except: |
| return False |
|
|
|
|
| if __name__ == "__main__": |
| print("InklyAI Web UI Test Suite") |
| print("=" * 50) |
| |
| |
| if not check_web_server(): |
| print("❌ Web server is not running!") |
| print("Please start the web server with:") |
| print(" python web_app.py") |
| print("\nThen run this test script again.") |
| exit(1) |
| |
| print("✅ Web server is running") |
| |
| |
| test_web_ui() |
| |
| print("\n" + "=" * 50) |
| print("Web UI Test Summary:") |
| print("✅ Health monitoring") |
| print("✅ Agent management") |
| print("✅ Signature verification") |
| print("✅ Statistics and reporting") |
| print("✅ Error handling") |
| print("\n🚀 Web UI is ready for use!") |
|
|
