Spaces:
Sleeping
Sleeping
| #!/usr/bin/env python3 | |
| """ | |
| Monitor HF Space deployment status. | |
| Checks if the space is running and provides status updates. | |
| """ | |
| import requests | |
| import time | |
| import sys | |
| from datetime import datetime | |
| def check_space_status(base_url: str) -> dict: | |
| """Check if the HF Space is running and responding.""" | |
| try: | |
| # Try the root endpoint first | |
| response = requests.get(f"{base_url}/", timeout=10) | |
| if response.status_code == 200: | |
| return {"status": "running", "message": "Space is active and responding"} | |
| elif response.status_code == 404: | |
| # Check if it's the HF "space not found" 404 or our app's 404 | |
| if "huggingface" in response.text.lower(): | |
| return {"status": "not_found", "message": "Space not found or not public"} | |
| else: | |
| return {"status": "app_404", "message": "App is running but endpoint not found"} | |
| else: | |
| return {"status": "error", "message": f"Unexpected status code: {response.status_code}"} | |
| except requests.exceptions.Timeout: | |
| return {"status": "timeout", "message": "Space is not responding (timeout)"} | |
| except requests.exceptions.ConnectionError: | |
| return {"status": "connection_error", "message": "Cannot connect to space"} | |
| except Exception as e: | |
| return {"status": "error", "message": f"Error: {str(e)}"} | |
| def monitor_space(base_url: str, check_interval: int = 30, max_checks: int = 20): | |
| """Monitor the space status over time.""" | |
| print(f"🔍 Monitoring HF Space: {base_url}") | |
| print(f"⏱️ Check interval: {check_interval} seconds") | |
| print(f"🔢 Max checks: {max_checks}") | |
| print("-" * 60) | |
| for i in range(max_checks): | |
| timestamp = datetime.now().strftime("%H:%M:%S") | |
| print(f"[{timestamp}] Check {i+1}/{max_checks}:", end=" ") | |
| status = check_space_status(base_url) | |
| if status["status"] == "running": | |
| print(f"✅ {status['message']}") | |
| print(f"🎉 Space is running! You can now test the endpoints.") | |
| return True | |
| elif status["status"] == "app_404": | |
| print(f"⚠️ {status['message']}") | |
| print(f"💡 App seems to be running, try the API endpoints directly.") | |
| return True | |
| else: | |
| print(f"❌ {status['message']}") | |
| if i < max_checks - 1: | |
| print(f" Waiting {check_interval} seconds before next check...") | |
| time.sleep(check_interval) | |
| print(f"\n💥 Space did not start after {max_checks} checks.") | |
| print("🔧 Check the HF Space logs for deployment errors.") | |
| return False | |
| if __name__ == "__main__": | |
| import argparse | |
| parser = argparse.ArgumentParser(description='Monitor HF Space deployment') | |
| parser.add_argument( | |
| '--url', | |
| default='https://ch404-cardserver.hf.space', | |
| help='Base URL of the HF Space' | |
| ) | |
| parser.add_argument( | |
| '--interval', | |
| type=int, | |
| default=30, | |
| help='Check interval in seconds' | |
| ) | |
| parser.add_argument( | |
| '--max-checks', | |
| type=int, | |
| default=20, | |
| help='Maximum number of checks before giving up' | |
| ) | |
| args = parser.parse_args() | |
| success = monitor_space(args.url, args.interval, args.max_checks) | |
| sys.exit(0 if success else 1) | |