Manage secrets in Colab and HF

Fashion_MVP_v0_0_2.ipynb

@@ -164,7 +164,12 @@
       "source": [
         "#|export\n",
         "import openai\n",
-        "from google.colab import userdata\n",
+        "\n",
+        "try:\n",
+        "  from google.colab import userdata\n",
+        "  IN_COLAB = True\n",
+        "except:\n",
+        "  IN_COLAB = False\n",
         "from openai import OpenAI\n",
         "from IPython.display import Image\n",
         "\n",
@@ -257,6 +262,14 @@
         "#|export\n",
         "os.makedirs('images', exist_ok=True)\n",
         "\n",
+        "\n",
+        "if IN_COLAB:\n",
+        "    OPEN_AI_KEY = userdata.get('OPEN_AI_KEY')\n",
+        "    MJ_AUTH = userdata.get('MJ_AUTH')\n",
+        "else:\n",
+        "    OPEN_AI_KEY = os.environ['API_TOKEN']\n",
+        "    MJ_AUTH = os.environ['MJ_AUTH']\n",
+        "\n",
         "client = OpenAI(\n",
         "        api_key=userdata.get('OPEN_AI_KEY'), # 'OPENAI_API_KEY'\n",
         "        base_url=\"https://api.openai.com/v1/\"\n",
@@ -461,7 +474,7 @@
         "            channel_id = \"1222929433682378787\",\n",
         "            version = \"1166847114203123795\",\n",
         "            id = \"938956540159881230\",\n",
-        "            authorization = \"MTIxOTk1NjY3MTI2MzE1MDE4NA.Gy7YpP.EJ0XxXJ8f7E8GFAaMU_1wk0SJlzpn9sZckbYN0\"\n",
+        "            authorization = MJ_AUTH\n",
         "        )\n",
         "        path = get_latest_file_path(\"./images/\")\n",
         "        gen_image_type = \"mj\"\n",

app.py

@@ -9,7 +9,12 @@ __all__ = ['client', 'vision_model', 'chat_model', 'text2img_model', 'initial_pr
 
 # %% Fashion_MVP_v0_0_2.ipynb 8
 import openai
-from google.colab import userdata
+
+try:
+  from google.colab import userdata
+  IN_COLAB = True
+except:
+  IN_COLAB = False
 from openai import OpenAI
 from IPython.display import Image
 
@@ -84,6 +89,14 @@ def get_latest_file_path(directory):
 # %% Fashion_MVP_v0_0_2.ipynb 10
 os.makedirs('images', exist_ok=True)
 
+
+if IN_COLAB:
+    OPEN_AI_KEY = userdata.get('OPEN_AI_KEY')
+    MJ_AUTH = userdata.get('MJ_AUTH')
+else:
+    OPEN_AI_KEY = os.environ['API_TOKEN']
+    MJ_AUTH = os.environ['MJ_AUTH']
+
 client = OpenAI(
         api_key=userdata.get('OPEN_AI_KEY'), # 'OPENAI_API_KEY'
         base_url="https://api.openai.com/v1/"
@@ -252,7 +265,7 @@ def generate_image(brand_name, style_name, initial_prompt, brand_prompt, brand_r
             channel_id = "1222929433682378787",
             version = "1166847114203123795",
             id = "938956540159881230",
-            authorization = "MTIxOTk1NjY3MTI2MzE1MDE4NA.Gy7YpP.EJ0XxXJ8f7E8GFAaMU_1wk0SJlzpn9sZckbYN0"
+            authorization = MJ_AUTH
         )
         path = get_latest_file_path("./images/")
         gen_image_type = "mj"