Update

colab.ipynb

@@ -13,8 +13,7 @@
     "from huggingface_hub import InferenceClient\n",
     "from IPython.display import display, clear_output\n",
     "from img_gen_logic import generate_image\n",
-    "from config_colab import models, api_token\n",
-    "from prompts import prompts  # Import prompts from prompts.py\n",
+    "from config_colab import models, prompts, api_token\n",
     "\n",
     "\n",
     "# Initialize the InferenceClient with the default model\n",
@@ -132,6 +131,18 @@
     "        print(f\"Seed: {seed}\")\n",
     "        print(f\"Custom Prompt: {custom_prompt}\")\n",
     "\n",
+    "            # Debug: Show selected parameters\n",
+    "        print(\"=== Debug: Selected Parameters ===\")\n",
+    "        print(f\"Selected Model: {selected_model} (Alias: {model_dropdown.label})\")\n",
+    "        print(f\"Selected Prompt: {selected_prompt} (Alias: {prompt_dropdown.label})\")\n",
+    "        print(f\"Selected Team: {selected_team}\")\n",
+    "        print(f\"Height: {height}\")\n",
+    "        print(f\"Width: {width}\")\n",
+    "        print(f\"Inference Steps: {num_inference_steps}\")\n",
+    "        print(f\"Guidance Scale: {guidance_scale}\")\n",
+    "        print(f\"Seed: {seed}\")\n",
+    "        print(f\"Custom Prompt: {custom_prompt}\")\n",
+    "        print(\"==================================\")\n",
     "        # Generate the image\n",
     "        image, message = generate_image(selected_prompt, selected_team, selected_model, height, width, num_inference_steps, guidance_scale, seed, custom_prompt)\n",
     "\n",

config_colab.py

@@ -7,18 +7,20 @@ api_token = userdata.get("HF_CTB_TOKEN")
 
 # Debugging: Check if the Hugging Face token is available
 if not api_token:
+    print("=== Debug: Error ===")
     print("ERROR: Hugging Face token (HF_CTB_TOKEN) is missing. Please set it in Colab secrets.")
 else:
+    print("=== Debug: Success ===")
     print("Hugging Face token loaded successfully.")
 
-
 # List of models with aliases
 models = [
     {"alias": "FLUX.1-dev", "name": "black-forest-labs/FLUX.1-dev"},
     {"alias": "Midjourney", "name": "strangerzonehf/Flux-Midjourney-Mix2-LoRA"}
 ]
 
-
 # Debugging: Print prompt and model options
+print("=== Debug: Available Options ===")
 print("Prompt Options:", [p["alias"] for p in prompts])
-print("Model Options:", [m["alias"] for m in models])
+print("Model Options:", [m["alias"] for m in models])
+print("=================================")