From a78bb82b442ea4cd630f2e3ca40b0e93985711a1 Mon Sep 17 00:00:00 2001
From: city-unit <140349364+city-unit@users.noreply.github.com>
Date: Sun, 17 Sep 2023 23:53:11 -0400
Subject: [PATCH] Fix collab to generate an api key

---
 colab/GPU.ipynb | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/colab/GPU.ipynb b/colab/GPU.ipynb
index a1ab0207d..6084690a3 100644
--- a/colab/GPU.ipynb
+++ b/colab/GPU.ipynb
@@ -73,6 +73,7 @@
         "#@markdown Enables ChromaDB for Infinity Context plugin\n",
         "\n",
         "import subprocess\n",
+        "import secrets\n",
         "\n",
         "# ---\n",
         "# SillyTavern extras\n",
@@ -119,6 +120,13 @@
         "!wget https://github.com/cloudflare/cloudflared/releases/download/2023.5.0/cloudflared-linux-amd64 -O /tmp/cloudflared-linux-amd64\n",
         "!chmod +x /tmp/cloudflared-linux-amd64\n",
         "\n",
+        "# Generate a random API key\n",
+        "api_key = secrets.token_hex(5)\n",
+        "\n",
+        "# Write the API key to api_key.txt\n",
+        "with open('./api_key.txt', 'w') as f:\n",
+        "    f.write(api_key)\n",
+        "print(f\"API Key generated: {api_key}\")\n",
         "\n",
         "cmd = f\"python server.py {' '.join(params)}\"\n",
         "print(cmd)\n",