Improve ISO installer non-GPU and bundled setup flow

autoinstall/websetup/server.py

@@ -36,7 +36,30 @@ def get_interfaces():
     except:
         return ["eth0"]
 
+def has_nvidia_gpu():
+    """Detect NVIDIA PCI devices before the driver or nvidia-smi exists."""
+    try:
+        for root, _, files in os.walk("/sys/bus/pci/devices"):
+            if "vendor" not in files:
+                continue
+            with open(os.path.join(root, "vendor")) as f:
+                if f.read().strip().lower() == "0x10de":
+                    return True
+    except Exception:
+        pass
+    return False
+
+def validate_static_network(ip, prefix, gateway, dns):
+    ipaddress.ip_address(ip)
+    ipaddress.ip_address(gateway)
+    ipaddress.ip_address(dns)
+    prefix_int = int(prefix)
+    if prefix_int < 1 or prefix_int > 32:
+        raise ValueError("CIDR prefix must be between 1 and 32")
+    return str(prefix_int)
+
 def apply_static_ip(iface, ip, prefix, gateway, dns):
+    prefix = validate_static_network(ip, prefix, gateway, dns)
     config = f"""network:
   version: 2
   ethernets:
@@ -69,9 +92,10 @@ def run_install(tier, skip_tools):
         open(SETUP_DONE_FILE, "w").close()
 
         env = os.environ.copy()
-        # Phase 1: installs NVIDIA drivers, registers cezen-phase2 systemd service,
-        # then reboots. Phase 2 (full stack) runs automatically after reboot.
-        cmd = ["bash", f"{AIPACKAGE_DIR}/install.sh", "--phase=1", f"--tier={tier}"]
+        # Fresh NVIDIA servers do not have nvidia-smi yet, so detect the PCI
+        # device and run phase 1 to install drivers before the AI stack.
+        phase = "1" if has_nvidia_gpu() else "2"
+        cmd = ["bash", f"{AIPACKAGE_DIR}/install.sh", f"--phase={phase}", f"--tier={tier}"]
         with open(INSTALL_LOG, "w") as log:
             proc = subprocess.Popen(cmd, stdout=log, stderr=log, env=env)
             proc.wait()
@@ -525,11 +549,18 @@ function streamLog() {
     }
   };
 
+  let reconnectAttempts = 0;
   es.onerror = () => {
     es.close();
-    if (!installDone) {
-      // Connection lost — most likely the server rebooted
+    if (installDone) return;
+    reconnectAttempts++;
+    lbl.textContent = `Connection lost — reconnecting... (${reconnectAttempts})`;
+    if (reconnectAttempts >= 5) {
+      // After 5 failed reconnects assume it's a real reboot
       showRebootNotice();
+    } else {
+      // Try reconnecting after a delay
+      setTimeout(() => { if (!installDone) streamLog(); }, 4000);
     }
   };
 }
@@ -653,14 +684,46 @@ class Handler(BaseHTTPRequestHandler):
             self.end_headers()
 
 # ─── Main ─────────────────────────────────────────────────
+def show_console_banner(ip):
+    """Write the setup URL banner to /dev/tty1 so it appears on the physical console."""
+    banner = f"""
+
+\033[1;36m╔══════════════════════════════════════════════════════╗
+║                                                      ║
+║          CEZEN AI SUITE — SERVER SETUP               ║
+║                                                      ║
+║   Open a browser on any computer on this network:   ║
+║                                                      ║
+║   \033[1;33m➜  http://{ip:<42}\033[1;36m║
+║   \033[1;33m➜  http://cezenai.local\033[1;36m                              ║
+║                                                      ║
+║   Complete setup from your browser — no keyboard    ║
+║   input needed here.                                 ║
+║                                                      ║
+╚══════════════════════════════════════════════════════╝\033[0m
+
+"""
+    # Write to tty1 (physical console) and stdout (journalctl)
+    print(banner)
+    try:
+        with open("/dev/tty1", "w") as tty:
+            tty.write(banner)
+    except Exception:
+        pass  # tty1 may not be accessible in all environments
+
+    # Also update /etc/issue so the URL appears above the login prompt
+    try:
+        with open("/etc/issue", "w") as f:
+            f.write(f"Ubuntu 22.04.5 LTS \\n \\l\n\n")
+            f.write(f"  \033[1;36mCezen AI Suite Setup:\033[0m http://{ip}  |  http://cezenai.local\n\n")
+    except Exception:
+        pass
+
+
 if __name__ == "__main__":
     # Ensure log file exists
     open(INSTALL_LOG, "a").close()
     ip = get_ip()
-    print(f"\n{'='*50}")
-    print(f"  Cezen AI Suite — Setup Server")
-    print(f"  Open in browser: http://{ip}")
-    print(f"  Or: http://cezenai.local")
-    print(f"{'='*50}\n")
+    show_console_banner(ip)
     server = HTTPServer(("0.0.0.0", 80), Handler)
     server.serve_forever()