Spaces:
Sleeping
Sleeping
| #!/usr/bin/env python3 | |
| """ | |
| Simple test script for the deployed API | |
| Update BASE_URL to your Hugging Face Space URL or local server | |
| """ | |
| import httpx | |
| import json | |
| import os | |
| import sys | |
| from dotenv import load_dotenv | |
| load_dotenv() # Load environment variables from .env file | |
| # Possible Hugging Face Space URLs to try - Common formats for HF Spaces | |
| POSSIBLE_URLS = [ | |
| "https://snvskiit-content-processor.hf.space", # Standard format: username-spacename | |
| "https://snvskiit-content_processor.hf.space", # With underscore | |
| "https://content-processor-snvskiit.hf.space", # Reverse order with dash | |
| "https://content_processor-snvskiit.hf.space", # Reverse order with underscore | |
| "https://huggingface.co/spaces/snvskiit/content-processor", # Direct HF URL with dash | |
| "https://huggingface.co/spaces/snvskiit/content_processor", # Direct HF URL with underscore | |
| ] | |
| # Update this to your actual Hugging Face Space URL or use "local" for localhost:7860 | |
| BASE_URL = "https://snvskiit-content-processor.hf.space" # Working endpoint! | |
| # BASE_URL = "http://localhost:7860" # Uncomment for local testing | |
| def find_working_hf_endpoint(): | |
| """Try different HF Space URL formats to find the working one""" | |
| print("🔍 Searching for working Hugging Face Space endpoint...") | |
| print("=" * 60) | |
| for i, url in enumerate(POSSIBLE_URLS, 1): | |
| print(f"\n{i}. Testing: {url}") | |
| try: | |
| with httpx.Client(timeout=10.0, verify=False) as client: | |
| response = client.get(f"{url}/health") | |
| if response.status_code == 200: | |
| print(f" ✅ SUCCESS! Found working endpoint") | |
| try: | |
| data = response.json() | |
| print(f" 📊 Response: {data}") | |
| return url | |
| except: | |
| print(f" 📝 Response (non-JSON): {response.text[:100]}...") | |
| return url | |
| elif response.status_code == 404: | |
| print(f" ❌ 404 - Space not found at this URL") | |
| else: | |
| print(f" ⚠️ Status {response.status_code} - {response.text[:50]}...") | |
| except Exception as e: | |
| print(f" ❌ Connection failed: {str(e)[:50]}...") | |
| print(f"\n🚫 No working endpoint found. Please check:") | |
| print(f" • Your Hugging Face username") | |
| print(f" • Your Space name") | |
| print(f" • That the Space is deployed and running") | |
| return None | |
| def test_api(): | |
| """Test the deployed API""" | |
| global BASE_URL | |
| # Handle local testing | |
| if len(sys.argv) > 1 and sys.argv[1] == "local": | |
| BASE_URL = "http://localhost:7860" | |
| print("🏠 Testing local server...") | |
| elif len(sys.argv) > 1 and sys.argv[1] == "find": | |
| # Find working endpoint mode | |
| working_url = find_working_hf_endpoint() | |
| if working_url: | |
| BASE_URL = working_url | |
| print(f"\n🎯 Using found endpoint: {BASE_URL}") | |
| else: | |
| print(f"\n⚠️ Falling back to default: {BASE_URL}") | |
| print("\n" + "=" * 60) | |
| else: | |
| print("🌐 Testing Hugging Face Space deployment...") | |
| print("🔍 If 404 errors occur, try: python test.py find") | |
| print(" This will test all possible URL formats") | |
| # Get OpenAI API key from environment | |
| api_key = os.getenv("OPENAI_API_KEY") | |
| if not api_key: | |
| print("❌ Please set OPENAI_API_KEY environment variable") | |
| print(" export OPENAI_API_KEY='your_actual_key'") | |
| return | |
| headers = { | |
| "Content-Type": "application/json", | |
| "x-api-key": api_key | |
| } | |
| print(f"🧪 Testing API at: {BASE_URL}") | |
| print("=" * 50) | |
| # Test root endpoint first | |
| try: | |
| with httpx.Client(timeout=30.0, verify=False) as client: | |
| response = client.get(f"{BASE_URL}/") | |
| print(f"✅ Root Endpoint: {response.status_code}") | |
| if response.status_code == 200: | |
| try: | |
| print(f" {response.json()}") | |
| except: | |
| print(f" Response: {response.text[:200]}...") | |
| else: | |
| print(f" Error response: {response.text[:200]}...") | |
| except Exception as e: | |
| print(f"❌ Root endpoint failed: {e}") | |
| # Test health endpoint | |
| try: | |
| with httpx.Client(timeout=30.0, verify=False) as client: | |
| response = client.get(f"{BASE_URL}/health") | |
| print(f"✅ Health Check: {response.status_code}") | |
| if response.status_code == 200: | |
| try: | |
| print(f" {response.json()}") | |
| except: | |
| print(f" Response: {response.text[:200]}...") | |
| else: | |
| print(f" Error response: {response.text[:200]}...") | |
| except Exception as e: | |
| print(f"❌ Health check failed: {e}") | |
| # Test address detection | |
| try: | |
| with httpx.Client(timeout=30.0, verify=False) as client: | |
| data = { | |
| "entity_urn": "test_001", | |
| "content": "Send package to 123 Main Street, New York, NY 10001" | |
| } | |
| response = client.post(f"{BASE_URL}/address-detection", json=data, headers=headers) | |
| print(f"✅ Address Detection: {response.status_code}") | |
| if response.status_code == 200: | |
| try: | |
| result = response.json() | |
| print(f" Addresses: {result.get('result', 'N/A')}") | |
| except: | |
| print(f" Response: {response.text[:200]}...") | |
| else: | |
| print(f" Error response: {response.text[:200]}...") | |
| except Exception as e: | |
| print(f"❌ Address detection failed: {e}") | |
| # Test summarization | |
| try: | |
| with httpx.Client(timeout=30.0, verify=False) as client: | |
| data = { | |
| "entity_urn": "test_002", | |
| "content": "I love this product! It's amazing and exceeded my expectations." | |
| } | |
| response = client.post(f"{BASE_URL}/summarize", json=data, headers=headers) | |
| print(f"✅ Summarization: {response.status_code}") | |
| if response.status_code == 200: | |
| try: | |
| result = response.json() | |
| print(f" Summary: {result.get('result', 'N/A')}") | |
| print(f" Sentiment: {result.get('sentiment', 'N/A')}") | |
| except: | |
| print(f" Response: {response.text[:200]}...") | |
| else: | |
| print(f" Error response: {response.text[:200]}...") | |
| except Exception as e: | |
| print(f"❌ Summarization failed: {e}") | |
| print(f"\n📖 View docs at: {BASE_URL}/docs") | |
| print(f"\n💡 Usage:") | |
| print(f" python test.py # Test default Hugging Face deployment") | |
| print(f" python test.py find # Auto-discover working HF endpoint") | |
| print(f" python test.py local # Test local server") | |
| if "hf.space" in BASE_URL: | |
| print(f"\n🔧 Troubleshooting Hugging Face Space:") | |
| print(f" • Check if Space is active: {BASE_URL}") | |
| print(f" • Try: python test.py find") | |
| print(f" • Check Space logs for deployment errors") | |
| print(f" • Verify your username and space name are correct") | |
| if __name__ == "__main__": | |
| test_api() | |