Create check_server.py

check_server.py

@@ -0,0 +1,68 @@
+import requests
+import sys
+import time
+
+def check_server(url="http://localhost:7860", max_retries=5, retry_delay=2):
+    """
+    Check if the server is running by making a request to the root URL.
+    
+    Args:
+        url (str): The URL to check
+        max_retries (int): Maximum number of retry attempts
+        retry_delay (int): Delay between retries in seconds
+        
+    Returns:
+        bool: True if the server is running, False otherwise
+    """
+    print(f"Checking if server is running at {url}...")
+    
+    for i in range(max_retries):
+        try:
+            response = requests.get(url)
+            if response.status_code == 200:
+                print(f"Server is running! Status code: {response.status_code}")
+                return True
+            else:
+                print(f"Server returned status code {response.status_code}, retrying ({i+1}/{max_retries})...")
+        except requests.exceptions.ConnectionError:
+            print(f"Connection error, retrying ({i+1}/{max_retries})...")
+        
+        # Wait before retrying
+        time.sleep(retry_delay)
+    
+    print(f"Server is not running after {max_retries} attempts.")
+    return False
+
+def check_health(url="http://localhost:7860/health"):
+    """
+    Check the health endpoint of the server.
+    
+    Args:
+        url (str): The health endpoint URL
+        
+    Returns:
+        dict: The health check response or None if failed
+    """
+    try:
+        response = requests.get(url)
+        if response.status_code == 200:
+            health_data = response.json()
+            print("Health check successful!")
+            print(f"Python version: {health_data.get('python_version', 'unknown')}")
+            print(f"System: {health_data.get('system', 'unknown')}")
+            print("Dependencies:")
+            for dep, status in health_data.get('dependencies', {}).items():
+                print(f"  - {dep}: {status}")
+            return health_data
+        else:
+            print(f"Health check failed with status code: {response.status_code}")
+            return None
+    except Exception as e:
+        print(f"Error checking health: {str(e)}")
+        return None
+
+if __name__ == "__main__":
+    base_url = sys.argv[1] if len(sys.argv) > 1 else "http://localhost:7860"
+    
+    if check_server(base_url):
+        check_health(f"{base_url}/health")