Update setup.py

setup.py

@@ -136,28 +136,18 @@ def main():
     download_helper_script(helper_script_path)
     inject_keys(civitai_key, hf_read_token, helper_script_path)
 
+    # --- THE FIX: Execute the helper script to load its magic commands ---
     print(f"{ARROW} Loading downloader functions...")
     get_ipython().run_line_magic('run', str(helper_script_path))
     print(f"{SUCCESS} Downloader is now ready!")
 
     setup_directory_structure(webui_path, model_path)
 
-    # --- THE FIX: Export paths as environment variables ---
-    print(f"{ARROW} Creating environment variables for paths...")
-    os.environ['WEBUI_PATH'] = str(webui_path)
-    os.environ['CKPT'] = str(model_path / 'Stable-diffusion')
-    os.environ['VAE'] = str(model_path / 'VAE')
-    os.environ['LORA'] = str(model_path / 'Lora')
-    os.environ['EMB'] = str(model_path / 'embeddings')
-    os.environ['CTRL'] = str(model_path / 'ControlNet')
-    os.environ['AD'] = str(model_path / 'adetailer')
-    print(f"{SUCCESS} Environment variables created!")
-
     print("\n" + "="*50)
     print(f"{SUCCESS} Setup complete!")
     print(f"{INFO} Your main folder is: {webui_path}")
-    print(f"{INFO} You can now use the %download magic command and path variables directly.")
-    print(f"{INFO} Example: %cd -q $CKPT")
+    print(f"{INFO} You can now use the %download magic command directly in your cells.")
+    print(f"{INFO} Models will be saved in the 'models' subdirectory of your main folder.")
     print("="*50)