import json
import requests
import time
import subprocess

print("Starting ComfyUI")

command = ["python", "main.py", "--disable-auto-launch", "--disable-metadata", "--cpu"]
# Start the server
server_process = subprocess.Popen(command, cwd="/comfyui")

def check_server(url, retries=50, delay=500):
    for i in range(retries):
        try:
            response = requests.head(url)

            # If the response status code is 200, the server is up and running
            if response.status_code == 200:
                print(f"builder - API is reachable")
                return True
        except requests.RequestException as e:
            # If an exception occurs, the server may not be ready
            pass

        # Wait for the specified delay before retrying
        time.sleep(delay / 1000)

    print(
        f"builder- Failed to connect to server at {url} after {retries} attempts."
    )
    return False

root_url = "http://127.0.0.1:8188"

check_server(root_url)

headers = {"Content-Type": "application/json"}

# Load JSON array from deps.json
with open('deps.json') as f:
    packages = json.load(f)

# Make a POST request for each package
for package in packages:
    response = requests.request("POST", f"{root_url}/customnode/install", json=package, headers=headers)
    print(response.text)

with open('models.json') as f:
    models = json.load(f)

for model in models:
    response = requests.request("POST", f"{root_url}/model/install", json=model, headers=headers)
    print(response.text)

# Close the server
server_process.terminate()
print("Finished installing dependencies.")