Spaces:
Running
Running
| #!/usr/bin/env python3 | |
| """ | |
| 🧪 Test Ziwei API locally | |
| """ | |
| import requests | |
| import json | |
| # Test data | |
| test_request = { | |
| "birth_date": "2000-1-1", | |
| "birth_time": "14:30", | |
| "gender": "男", | |
| "is_leap_month": False, | |
| "fix_leap": True | |
| } | |
| print("🔮 Testing Ziwei API...") | |
| print(f"Request: {json.dumps(test_request, ensure_ascii=False)}\n") | |
| try: | |
| # Note: This test requires the API to be running | |
| # Run: uvicorn app:app --host 127.0.0.1 --port 8765 | |
| response = requests.post( | |
| "http://127.0.0.1:8765/calculate-ziwei", | |
| json=test_request, | |
| timeout=10 | |
| ) | |
| if response.status_code == 200: | |
| result = response.json() | |
| print("✅ API Test Successful!\n") | |
| print(f"Status: {result['status']}") | |
| print(f"Solar Date: {result['data']['solar_date']}") | |
| print(f"Lunar Date: {result['data']['lunar_date']}") | |
| print(f"Sign: {result['data']['chart']['sign']}") | |
| print(f"Zodiac: {result['data']['chart']['zodiac']}") | |
| print(f"\n命宫: {result['data']['palaces'][3]['name']}") | |
| print(f"Major Stars: {', '.join(result['data']['palaces'][3]['major_stars'])}") | |
| print("\n🎯 All tests passed! ✨") | |
| else: | |
| print(f"❌ API returned status {response.status_code}") | |
| print(response.text) | |
| except requests.exceptions.ConnectionError: | |
| print("⚠️ API not running. Please start with:") | |
| print(" cd /Volumes/sparksverse/huggingface-spaces/astrology-ziwei-api") | |
| print(" uvicorn app:app --host 127.0.0.1 --port 8765") | |
| except Exception as e: | |
| print(f"❌ Test failed: {e}") | |