Spaces:
Running
Running
| """Run the public Node API and private Python worker as one service.""" | |
| import json | |
| import os | |
| from pathlib import Path | |
| import signal | |
| import subprocess | |
| import sys | |
| import time | |
| from urllib.request import urlopen | |
| from dotenv import load_dotenv | |
| def wait_for_analysis(process, should_stop, timeout_seconds=25 * 60): | |
| deadline = time.monotonic() + timeout_seconds | |
| while not should_stop(): | |
| exit_code = process.poll() | |
| if exit_code is not None: | |
| print(f"Analysis process exited before it was ready ({exit_code})", flush=True) | |
| return False | |
| try: | |
| with urlopen("http://127.0.0.1:8000/healthz", timeout=1) as response: | |
| payload = json.load(response) if response.status == 200 else None | |
| if isinstance(payload, dict) and payload.get("ok") is True: | |
| return process.poll() is None | |
| except (OSError, ValueError): | |
| pass | |
| if time.monotonic() >= deadline: | |
| print(f"Analysis process did not become ready within {timeout_seconds} seconds", flush=True) | |
| return False | |
| time.sleep(0.5) | |
| return False | |
| def main(): | |
| root = Path(__file__).resolve().parent | |
| load_dotenv(root / ".env") | |
| port = int(os.environ.get("PORT", "4000")) | |
| if not 1 <= port <= 65535 or port == 8000: | |
| raise ValueError("PORT must be between 1 and 65535 and differ from AI port 8000") | |
| env = dict(os.environ, PORT=str(port), | |
| ANALYSIS_API_URL="http://127.0.0.1:8000", | |
| BACKEND_INTERNAL_URL=f"http://127.0.0.1:{port}") | |
| children = [] | |
| stopping = False | |
| def stop(signum, frame): | |
| nonlocal stopping | |
| stopping = True | |
| for sig in (signal.SIGINT, signal.SIGTERM): | |
| signal.signal(sig, stop) | |
| try: | |
| analysis = subprocess.Popen([sys.executable, "-m", "uvicorn", "app:app", | |
| "--host", "127.0.0.1", "--port", "8000", "--workers", "1"], cwd=root, env=env) | |
| children.append(analysis) | |
| if not wait_for_analysis(analysis, lambda: stopping): | |
| return 0 if stopping else 1 | |
| children.append(subprocess.Popen(["node", "dist/server.js"], cwd=root, env=env)) | |
| while not stopping: | |
| for child in children: | |
| if child.poll() is not None: | |
| print(f"Backend process {child.pid} exited ({child.returncode}); stopping service", flush=True) | |
| return child.returncode or 1 | |
| time.sleep(0.25) | |
| return 0 | |
| finally: | |
| for child in children: | |
| if child.poll() is None: | |
| child.terminate() | |
| deadline = time.monotonic() + 15 | |
| for child in children: | |
| try: | |
| child.wait(timeout=max(0, deadline - time.monotonic())) | |
| except subprocess.TimeoutExpired: | |
| child.kill() | |
| child.wait() | |
| if __name__ == "__main__": | |
| sys.exit(main()) | |