Update custom-entrypoint.sh

custom-entrypoint.sh

@@ -13,20 +13,24 @@ echo "Waiting for ngrok TCP tunnel..."
 PUBLIC_TCP=""
 while [[ -z "$PUBLIC_TCP" ]]; do
     sleep 2
-    PUBLIC_TCP=$(curl -s http://127.0.0.1:7860/api/tunnels \
-        | python3 -c "
+
+    # Get tunnels JSON safely
+    JSON=$(curl -s http://127.0.0.1:4040/api/tunnels || echo "{}")
+
+    # Try parsing TCP tunnel
+    PUBLIC_TCP=$(echo "$JSON" | python3 -c "
 import sys, json
-tunnels=json.load(sys.stdin).get('tunnels', [])
-for t in tunnels:
-    if t.get('proto')=='tcp':
-        print(t.get('public_url','').replace('tcp://',''))
-        break
+try:
+    data = json.load(sys.stdin)
+    tunnels = data.get('tunnels', [])
+    for t in tunnels:
+        if t.get('proto')=='tcp':
+            print(t.get('public_url','').replace('tcp://',''))
+            break
+except Exception:
+    pass
 " || echo "")
 
-    if [[ -n "$PUBLIC_TCP" ]]; then
-        echo "Ngrok public TCP address: $PUBLIC_TCP"
-        break
-    fi
 done
 
 # Extract host and host:port