{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "import threading\n",
    "import subprocess\n",
    "import time\n",
    "import os\n",
    "#/content/esay\n",
    "%cd /content/esay/sd-ui-files\n",
    "use_remote = True # @param {type:\"boolean\"}\n",
    "use_cloudflare = True # @param {type:\"boolean\"}\n",
    "\n",
    "def start_tunnle():\n",
    "  open(\"/content/output.log\", \"w\").close()\n",
    "  # 循环检测文件内容，直到存在\"http://127.0.0.1:9090\"。\n",
    "  while not \"http://localhost:9000/\" in open(\"/content/output.log\", \"r\").read():\n",
    "    time.sleep(1)\n",
    "  clear_output()  \n",
    "  if ngrok_token:\n",
    "    try:\n",
    "        from pyngrok import ngrok\n",
    "        ngrok_tunnel = ngrok.connect(9000, \"http\")\n",
    "        print(\"ngrok_tunnel:\", ngrok_tunnel)\n",
    "    except Exception as e:\n",
    "        print(\"ngrok 连接失败：\", e)\n",
    "\n",
    "  if use_cloudflare:\n",
    "    try:\n",
    "        from pycloudflared import try_cloudflare\n",
    "        cloudflare_url = try_cloudflare(9000, verbose=False)\n",
    "        print(\"cloudflare_tunnel:\", cloudflare_url)\n",
    "    except Exception as e:\n",
    "        print(\"cloudflare 连接失败：\", e)\n",
    "\n",
    "  if use_remote:\n",
    "    try:\n",
    "         !ssh -R 80:127.0.0.1:9000 -o StrictHostKeyChecking=no -i /content/id_rsa remote.moe 2>&1 | tee -a /content/tunnel.log\n",
    "    except Exception as e:\n",
    "        print(\"cloudflare 连接失败：\", e)\n",
    "\n",
    "threading.Thread(target=start_tunnle, daemon=True).start()\n",
    "\n",
    "!python ./scripts/check_modules.py --launch-uvicorn 2>&1 | tee /content/output.log\n",
    "#!python ./ui/main.py"
   ]
  }
 ],
 "metadata": {
  "language_info": {
   "name": "python"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}