“Update”

colab.ipynb

@@ -6,22 +6,21 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# colab.ipynb\n",
-    "import sys\n",
     "import os\n",
+    "import sys\n",
+    "\n",
+    "# Set PYTHONPATH to the project root \n",
+    "# Solves all problems w subfolders\n",
+    "os.environ[\"PYTHONPATH\"] = os.path.abspath(os.path.join(\"..\"))\n",
     "\n",
-    "# Add the root folder to the Python path\n",
-    "root_path = os.path.abspath(\"\")\n",
-    "if root_path not in sys.path:\n",
-    "    sys.path.append(root_path)\n",
     "\n",
     "# Import necessary libraries\n",
     "import random\n",
     "import ipywidgets as widgets\n",
     "from huggingface_hub import InferenceClient\n",
     "from IPython.display import display, clear_output\n",
-    "from img_gen_logic_colab import generate_image, save_image\n",
-    "from config_colab import models, prompts, api_token  # Import from config folder\n",
+    "from src.img_gen_logic_colab import generate_image, save_image\n",
+    "from config.config_colab import models, prompts, api_token  # Import from config folder\n",
     "from PIL import Image\n",
     "from google.colab import userdata\n",
     "from datetime import datetime\n",