修改clone地址

notebooks/EasyTranslate_Production.ipynb

@@ -67,22 +67,81 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "REPO_URL = \"https://github.com/UCAS-EasyTranslate/UCAS-EasyTranslate.git\"\n",
+    "# You can override this in Colab before running this cell:\n",
+    "# os.environ[\"EASYTRANSLATE_REPO_URL\"] = \"https://github.com/<org>/<repo>.git\"\n",
+    "REPO_URL = os.environ.get(\n",
+    "    \"EASYTRANSLATE_REPO_URL\",\n",
+    "    \"https://huggingface.co/sdfjliom/UCAS-EasyTranslate\",\n",
+    ")\n",
     "REPO_DIR = Path(\"/content/UCAS-EasyTranslate\") if IN_COLAB else Path(\".\").resolve()\n",
     "\n",
+    "\n",
+    "def _is_repo_root(path: Path) -> bool:\n",
+    "    return (path / \"src\" / \"easytranslate\").exists() and (path / \"setup.py\").exists()\n",
+    "\n",
+    "\n",
+    "def _find_repo_root(start_path: Path) -> Path | None:\n",
+    "    p = start_path.resolve()\n",
+    "    for candidate in [p] + list(p.parents):\n",
+    "        if _is_repo_root(candidate):\n",
+    "            return candidate\n",
+    "    return None\n",
+    "\n",
+    "\n",
+    "resolved_repo: Path | None = None\n",
+    "\n",
     "if IN_COLAB:\n",
-    "    if not REPO_DIR.exists():\n",
-    "        !git clone {REPO_URL} {REPO_DIR}\n",
+    "    # 1) If already present, use it directly\n",
+    "    if _is_repo_root(REPO_DIR):\n",
+    "        resolved_repo = REPO_DIR\n",
     "    else:\n",
-    "        %cd {REPO_DIR}\n",
-    "        !git pull origin main\n",
-    "    %cd {REPO_DIR}\n",
+    "        # 2) Try clone (public repo expected). If private/auth fails, continue to fallbacks.\n",
+    "        try:\n",
+    "            subprocess.run(\n",
+    "                [\"git\", \"clone\", REPO_URL, str(REPO_DIR)],\n",
+    "                check=True,\n",
+    "                stdout=subprocess.PIPE,\n",
+    "                stderr=subprocess.STDOUT,\n",
+    "                text=True,\n",
+    "            )\n",
+    "            resolved_repo = REPO_DIR\n",
+    "            print(f\"Cloned repository to: {REPO_DIR}\")\n",
+    "        except subprocess.CalledProcessError as e:\n",
+    "            print(\"Clone failed. This usually means repo is private or URL is incorrect.\")\n",
+    "            print(\"Git output:\")\n",
+    "            print(e.stdout)\n",
+    "\n",
+    "    # 3) Fallback: try to locate an existing copy (e.g., from Drive or current working dir)\n",
+    "    if resolved_repo is None:\n",
+    "        candidate_paths = [\n",
+    "            Path.cwd(),\n",
+    "            Path(\"/content\"),\n",
+    "            Path(\"/content/drive/MyDrive/UCAS-EasyTranslate\"),\n",
+    "            Path(\"/content/drive/MyDrive/Colab Notebooks/UCAS-EasyTranslate\"),\n",
+    "        ]\n",
+    "        for candidate in candidate_paths:\n",
+    "            root = _find_repo_root(candidate)\n",
+    "            if root is not None:\n",
+    "                resolved_repo = root\n",
+    "                print(f\"Using existing repository at: {resolved_repo}\")\n",
+    "                break\n",
+    "\n",
+    "    if resolved_repo is None:\n",
+    "        raise FileNotFoundError(\n",
+    "            \"Cannot locate UCAS-EasyTranslate repository. \"\n",
+    "            \"Please do one of these: \"\n",
+    "            \"(1) set a public/correct EASYTRANSLATE_REPO_URL, \"\n",
+    "            \"(2) clone repo to /content/UCAS-EasyTranslate, \"\n",
+    "            \"(3) place repo in Google Drive and mount Drive.\"\n",
+    "        )\n",
     "else:\n",
-    "    print(f\"Using local repository at: {REPO_DIR}\")\n",
+    "    # Local: use nearest project root from current path\n",
+    "    resolved_repo = _find_repo_root(Path.cwd()) or Path.cwd()\n",
+    "    print(f\"Using local repository at: {resolved_repo}\")\n",
     "\n",
-    "os.chdir(REPO_DIR)\n",
-    "sys.path.insert(0, str(REPO_DIR / \"src\"))\n",
-    "print(f\"Repository directory: {REPO_DIR}\")\n"
+    "os.chdir(resolved_repo)\n",
+    "sys.path.insert(0, str(Path(resolved_repo) / \"src\"))\n",
+    "print(f\"Repository directory: {resolved_repo}\")"
    ]
   },
   {