wrong key name

.github/workflows/docker-publish.yml

@@ -0,0 +1,47 @@
+name: Build and Publish Docker Image
+
+on:
+  push:
+    branches: [main]
+    tags: ['*']  # triggers on any tag push
+  workflow_dispatch:
+
+jobs:
+  build-and-push:
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+
+      - name: Log in to Docker Registry
+        run: echo "${{ secrets.PASSWORD }}" | docker login ${{ secrets.REGISTRY }} -u "${{ secrets.USERNAME }}" --password-stdin
+
+      - name: Build and Push Docker Images
+        run: |
+          IMAGE_NAME="ai-frame-image-server"
+          REGISTRY="${{ secrets.REGISTRY }}"
+          USERNAME="${{ secrets.USERNAME }}"
+          IMAGE_LATEST="$REGISTRY/$USERNAME/$IMAGE_NAME:latest"
+
+          # Always build and tag as latest
+          echo "🔧 Building $IMAGE_LATEST"
+          docker build -t $IMAGE_LATEST .
+
+          echo "📤 Pushing $IMAGE_LATEST"
+          docker push $IMAGE_LATEST
+
+          # If this is a tag push, tag the image accordingly
+          if [[ "${GITHUB_REF}" == refs/tags/* ]]; then
+            GIT_TAG="${GITHUB_REF#refs/tags/}"
+            IMAGE_TAGGED="$REGISTRY/$USERNAME/$IMAGE_NAME:$GIT_TAG"
+
+            echo "🏷️ Also tagging as $IMAGE_TAGGED"
+            docker tag $IMAGE_LATEST $IMAGE_TAGGED
+
+            echo "📤 Pushing $IMAGE_TAGGED"
+            docker push $IMAGE_TAGGED
+          fi

.gitignore

@@ -1,4 +1,4 @@
-venv/*
+.venv/*
 script.log
 **/*.pyc
 *.rtf

ai_frame_image_server.py

@@ -14,6 +14,8 @@ from libs.generic import load_config, load_recent_prompts, get_prompt_from_png
 from libs.comfyui import cancel_current_job, create_image
 from libs.ollama import create_prompt_on_openwebui
 
+#workflow test commit
+
 user_config = load_config()
 app = Flask(__name__)
 

lib.py

@@ -1,300 +1,302 @@
-import random
-import configparser
-import logging
-import sys
-import litellm
-import time
-import os
-import requests
-from PIL import Image
-from typing import Optional
-from comfy_api_simplified import ComfyApiWrapper, ComfyWorkflowWrapper
-from tenacity import (
-    retry,
-    stop_after_attempt,
-    wait_fixed,
-    before_log,
-    retry_if_exception_type,
-)
-import nest_asyncio
-import json
-from datetime import datetime
-from libs.create_thumbnail import generate_thumbnail
-nest_asyncio.apply()
+# import random
+# import configparser
+# import logging
+# import sys
+# import litellm
+# import time
+# import os
+# import requests
+# from PIL import Image
+# from typing import Optional
+# from comfy_api_simplified import ComfyApiWrapper, ComfyWorkflowWrapper
+# from tenacity import (
+#     retry,
+#     stop_after_attempt,
+#     wait_fixed,
+#     before_log,
+#     retry_if_exception_type,
+# )
+# import nest_asyncio
+# import json
+# from datetime import datetime
+# from libs.create_thumbnail import generate_thumbnail
+# nest_asyncio.apply()
 
-logging.basicConfig(level=logging.INFO)
+# logging.basicConfig(level=logging.INFO)
 
-LOG_FILE = "./prompts_log.jsonl"
+# LOG_FILE = "./prompts_log.jsonl"
 
 
-def load_recent_prompts(count=7):
-    recent_prompts = []
+# def load_recent_prompts(count=7):
+#     recent_prompts = []
 
-    try:
-        with open(LOG_FILE, "r") as f:
-            lines = f.readlines()
-            for line in lines[-count:]:
-                data = json.loads(line.strip())
-                recent_prompts.append(data["prompt"])
-    except FileNotFoundError:
-        pass  # No prompts yet
+#     try:
+#         with open(LOG_FILE, "r") as f:
+#             lines = f.readlines()
+#             for line in lines[-count:]:
+#                 data = json.loads(line.strip())
+#                 recent_prompts.append(data["prompt"])
+#     except FileNotFoundError:
+#         pass  # No prompts yet
 
-    return recent_prompts
+#     return recent_prompts
 
 
-def save_prompt(prompt):
-    entry = {"date": datetime.now().strftime("%Y-%m-%d"), "prompt": prompt}
-    with open(LOG_FILE, "a") as f:
-        f.write(json.dumps(entry) + "\n")
+# def save_prompt(prompt):
+#     entry = {"date": datetime.now().strftime("%Y-%m-%d"), "prompt": prompt}
+#     with open(LOG_FILE, "a") as f:
+#         f.write(json.dumps(entry) + "\n")
 
 
 
-def get_available_models() -> list:
-    """Fetches available models from ComfyUI."""
-    url = user_config["comfyui"]["comfyui_url"] + "/object_info"
-    response = requests.get(url)
-    if response.status_code == 200:
-        data = response.json()
-        return (
-            data.get("CheckpointLoaderSimple", {})
-            .get("input", {})
-            .get("required", {})
-            .get("ckpt_name", [])[0]
-        )
-    else:
-        print(f"Failed to fetch models: {response.status_code}")
-        return []
+# def get_available_models() -> list:
+#     """Fetches available models from ComfyUI."""
+#     url = user_config["comfyui"]["comfyui_url"] + "/object_info"
+#     response = requests.get(url)
+#     if response.status_code == 200:
+#         data = response.json()
+#         return (
+#             data.get("CheckpointLoaderSimple", {})
+#             .get("input", {})
+#             .get("required", {})
+#             .get("ckpt_name", [])[0]
+#         )
+#     else:
+#         print(f"Failed to fetch models: {response.status_code}")
+#         return []
 
 
-def cancel_current_job() -> list:
-    """Fetches available models from ComfyUI."""
-    url = user_config["comfyui"]["comfyui_url"] + "/interrupt"
-    response = requests.post(url)
-    if response.status_code == 200:
-        return "Cancelled"
-    else:
-        return "Failed to cancel"
+# def cancel_current_job() -> list:
+#     """Fetches available models from ComfyUI."""
+#     url = user_config["comfyui"]["comfyui_url"] + "/interrupt"
+#     response = requests.post(url)
+#     if response.status_code == 200:
+#         return "Cancelled"
+#     else:
+#         return "Failed to cancel"
 
 
-def load_config() -> configparser.ConfigParser:
-    """Loads user configuration from ./user_config.cfg."""
-    user_config = configparser.ConfigParser()
-    try:
-        user_config.read("./user_config.cfg")
-        logging.debug("Configuration loaded successfully.")
-        return user_config
-    except KeyError as e:
-        logging.error(f"Missing configuration key: {e}")
-        sys.exit(1)
+# def load_config() -> configparser.ConfigParser:
+#     """Loads user configuration from ./user_config.cfg."""
+#     user_config = configparser.ConfigParser()
+#     try:
+#         user_config.read("./user_config.cfg")
+#         logging.debug("Configuration loaded successfully.")
+#         return user_config
+#     except KeyError as e:
+#         logging.error(f"Missing configuration key: {e}")
+#         sys.exit(1)
 
 
-def rename_image() -> str | None:
-    """Renames 'image.png' in the output folder to a timestamped filename if it exists."""
-    old_path = os.path.join(user_config["comfyui"]["output_dir"], "image.png")
+# def rename_image() -> str | None:
+#     """Renames 'image.png' in the output folder to a timestamped filename if it exists."""
+#     old_path = os.path.join(user_config["comfyui"]["output_dir"], "image.png")
 
-    if os.path.exists(old_path):
-        new_filename = f"{str(time.time())}.png"
-        new_path = os.path.join(user_config["comfyui"]["output_dir"], new_filename)
-        os.rename(old_path, new_path)
-        generate_thumbnail(new_path)
-        print(f"Renamed 'image.png' to '{new_filename}'")
-        return new_filename
-    else:
-        print("No image.png found.")
-        return None
+#     if os.path.exists(old_path):
+#         new_filename = f"{str(time.time())}.png"
+#         new_path = os.path.join(user_config["comfyui"]["output_dir"], new_filename)
+#         os.rename(old_path, new_path)
+#         generate_thumbnail(new_path)
+#         print(f"Renamed 'image.png' to '{new_filename}'")
+#         return new_filename
+#     else:
+#         print("No image.png found.")
+#         return None
 
 
-def create_prompt_on_openwebui(prompt: str) -> str:
-    """Sends prompt to OpenWebui and returns the generated response."""
-    # Unique list of recent prompts
-    recent_prompts = list(set(load_recent_prompts()))
-    # Decide on whether to include a topic (e.g., 30% chance to include)
-    topics = [t.strip() for t in user_config["comfyui"]["topics"].split(",") if t.strip()]
-    topic_instruction = ""
-    if random.random() < 0.3 and topics:
-        selected_topic = random.choice(topics)
-        topic_instruction = f" Incorporate the theme of '{selected_topic}' into the new prompt."
+# def create_prompt_on_openwebui(prompt: str) -> str:
+#     """Sends prompt to OpenWebui and returns the generated response."""
+#     # Unique list of recent prompts
+#     recent_prompts = list(set(load_recent_prompts()))
+#     # Decide on whether to include a topic (e.g., 30% chance to include)
+#     topics = [t.strip() for t in user_config["comfyui"]["topics"].split(",") if t.strip()]
+#     topic_instruction = ""
+#     if random.random() < 0.3 and topics:
+#         selected_topic = random.choice(topics)
+#         topic_instruction = f" Incorporate the theme of '{selected_topic}' into the new prompt."
 
-    user_content = (
-        "Here are the prompts from the last 7 days:\n\n"
-        + "\n".join(f"{i+1}. {p}" for i, p in enumerate(recent_prompts))
-        + "\n\nDo not repeat ideas, themes, or settings from the above. "
-        "Now generate a new, completely original Stable Diffusion prompt that hasn't been done yet."
-        + topic_instruction
-    )
+#     user_content = (
+#         "Here are the prompts from the last 7 days:\n\n"
+#         + "\n".join(f"{i+1}. {p}" for i, p in enumerate(recent_prompts))
+#         + "\n\nDo not repeat ideas, themes, or settings from the above. "
+#         "Now generate a new, completely original Stable Diffusion prompt that hasn't been done yet."
+#         + topic_instruction
+#     )
 
-    model = random.choice(user_config["openwebui"]["models"].split(","))
-    response = litellm.completion(
-        api_base=user_config["openwebui"]["base_url"],
-        model="openai/" + model,
-        messages=[
-            {
-                "role": "system",
-                "content": (
-                    "You are a prompt generator for Stable Diffusion. "
-                    "Generate a detailed and imaginative prompt with a strong visual theme. "
-                    "Focus on lighting, atmosphere, and artistic style. "
-                    "Keep the prompt concise, no extra commentary or formatting."
-                ),
-            },
-            {
-                "role": "user",
-                "content": user_content,
-            },
-        ],
-        api_key=user_config["openwebui"]["api_key"],
-    )
+#     model = random.choice(user_config["openwebui"]["models"].split(","))
+#     response = litellm.completion(
+#         api_base=user_config["openwebui"]["base_url"],
+#         model="openai/" + model,
+#         messages=[
+#             {
+#                 "role": "system",
+#                 "content": (
+#                     "You are a prompt generator for Stable Diffusion. "
+#                     "Generate a detailed and imaginative prompt with a strong visual theme. "
+#                     "Focus on lighting, atmosphere, and artistic style. "
+#                     "Keep the prompt concise, no extra commentary or formatting."
+#                 ),
+#             },
+#             {
+#                 "role": "user",
+#                 "content": user_content,
+#             },
+#         ],
+#         api_key=user_config["openwebui"]["api_key"],
+#     )
 
-    prompt = response["choices"][0]["message"]["content"].strip('"')
-    # response = litellm.completion(
-    #     api_base=user_config["openwebui"]["base_url"],
-    #     model="openai/brxce/stable-diffusion-prompt-generator:latest",
-    #     messages=[
-    #         {
-    #             "role": "user",
-    #             "content": prompt,
-    #         },
-    #     ],
-    #     api_key=user_config["openwebui"]["api_key"],
-    # )
-    # prompt = response["choices"][0]["message"]["content"].strip('"')
-    logging.debug(prompt)
-    return prompt
+#     prompt = response["choices"][0]["message"]["content"].strip('"')
+#     # response = litellm.completion(
+#     #     api_base=user_config["openwebui"]["base_url"],
+#     #     model="openai/brxce/stable-diffusion-prompt-generator:latest",
+#     #     messages=[
+#     #         {
+#     #             "role": "user",
+#     #             "content": prompt,
+#     #         },
+#     #     ],
+#     #     api_key=user_config["openwebui"]["api_key"],
+#     # )
+#     # prompt = response["choices"][0]["message"]["content"].strip('"')
+#     logging.debug(prompt)
+#     return prompt
 
 
-# Define the retry logic using Tenacity
-@retry(
-    stop=stop_after_attempt(3),
-    wait=wait_fixed(5),
-    before=before_log(logging.getLogger(), logging.DEBUG),
-    retry=retry_if_exception_type(Exception),
-)
-def generate_image(
-    file_name: str,
-    comfy_prompt: str,
-    workflow_path: str = "./workflow_api.json",
-    prompt_node: str = "CLIP Text Encode (Prompt)",
-    seed_node: str = "KSampler",
-    seed_param: str = "seed",
-    save_node: str = "Save Image",
-    save_param: str = "filename_prefix",
-    model_node: Optional[str] = "Load Checkpoint",
-    model_param: Optional[str] = "ckpt_name",
-) -> None:
-    """Generates an image using the Comfy API with configurable workflow settings."""
-    try:
-        api = ComfyApiWrapper(user_config["comfyui"]["comfyui_url"])
-        wf = ComfyWorkflowWrapper(workflow_path)
+# # Define the retry logic using Tenacity
+# @retry(
+#     stop=stop_after_attempt(3),
+#     wait=wait_fixed(5),
+#     before=before_log(logging.getLogger(), logging.DEBUG),
+#     retry=retry_if_exception_type(Exception),
+# )
+# def generate_image(
+#     file_name: str,
+#     comfy_prompt: str,
+#     workflow_path: str = "./workflow_api.json",
+#     prompt_node: str = "CLIP Text Encode (Prompt)",
+#     seed_node: str = "KSampler",
+#     seed_param: str = "seed",
+#     save_node: str = "Save Image",
+#     save_param: str = "filename_prefix",
+#     model_node: Optional[str] = "Load Checkpoint",
+#     model_param: Optional[str] = "ckpt_name",
+# ) -> None:
+#     """Generates an image using the Comfy API with configurable workflow settings."""
+#     try:
+#         api = ComfyApiWrapper(user_config["comfyui"]["comfyui_url"])
+#         wf = ComfyWorkflowWrapper(workflow_path)
 
-        # Set workflow parameters
-        wf.set_node_param(seed_node, seed_param, random.getrandbits(32))
-        wf.set_node_param(prompt_node, "text", comfy_prompt)
-        wf.set_node_param(save_node, save_param, file_name)
-        wf.set_node_param(
-            (
-                "Empty Latent Image"
-                if workflow_path.endswith("workflow_api.json")
-                else "CR Aspect Ratio"
-            ),
-            "width",
-            user_config["comfyui"]["width"],
-        )
-        wf.set_node_param(
-            (
-                "Empty Latent Image"
-                if workflow_path.endswith("workflow_api.json")
-                else "CR Aspect Ratio"
-            ),
-            "height",
-            user_config["comfyui"]["height"],
-        )
+#         # Set workflow parameters
+#         wf.set_node_param(seed_node, seed_param, random.getrandbits(32))
+#         wf.set_node_param(prompt_node, "text", comfy_prompt)
+#         wf.set_node_param(save_node, save_param, file_name)
+#         wf.set_node_param(
+#             (
+#                 "Empty Latent Image"
+#                 if workflow_path.endswith("workflow_api.json")
+#                 else "CR Aspect Ratio"
+#             ),
+#             "width",
+#             user_config["comfyui"]["width"],
+#         )
+#         wf.set_node_param(
+#             (
+#                 "Empty Latent Image"
+#                 if workflow_path.endswith("workflow_api.json")
+#                 else "CR Aspect Ratio"
+#             ),
+#             "height",
+#             user_config["comfyui"]["height"],
+#         )
 
-        # Conditionally set model if node and param are provided
-        if model_node and model_param:
-            if user_config["comfyui"].get("FLUX"):
-                valid_models = user_config["comfyui:flux"]["models"].split(",")
-            else:
-                available_model_list = user_config["comfyui"]["models"].split(",")
-                valid_models = list(
-                    set(get_available_models()) & set(available_model_list)
-                )
+#         # Conditionally set model if node and param are provided
+#         if model_node and model_param:
+#             if user_config["comfyui"].get("FLUX"):
+#                 valid_models = user_config["comfyui:flux"]["models"].split(",")
+#             else:
+#                 available_model_list = user_config["comfyui"]["models"].split(",")
+#                 valid_models = list(
+#                     set(get_available_models()) & set(available_model_list)
+#                 )
 
-                if not valid_models:
-                    raise Exception("No valid models available.")
+#                 if not valid_models:
+#                     raise Exception("No valid models available.")
 
-            model = random.choice(valid_models)
-            wf.set_node_param(model_node, model_param, model)
+#             model = random.choice(valid_models)
+#             wf.set_node_param(model_node, model_param, model)
 
 
-        # Generate image
-        logging.debug(f"Generating image: {file_name}")
-        results = api.queue_and_wait_images(wf, save_node)
-        rename_image()
+#         # Generate image
+#         logging.debug(f"Generating image: {file_name}")
+#         results = api.queue_and_wait_images(wf, save_node)
+#         rename_image()
 
-        for _, image_data in results.items():
-            output_path = os.path.join(
-                user_config["comfyui"]["output_dir"], f"{file_name}.png"
-            )
-            with open(output_path, "wb+") as f:
-                f.write(image_data)
+#         for _, image_data in results.items():
+#             output_path = os.path.join(
+#                 user_config["comfyui"]["output_dir"], f"{file_name}.png"
+#             )
+#             with open(output_path, "wb+") as f:
+#                 f.write(image_data)
 
-        logging.debug(f"Image generated successfully for UID: {file_name}")
+#         logging.debug(f"Image generated successfully for UID: {file_name}")
 
-    except Exception as e:
-        logging.error(f"Failed to generate image for UID: {file_name}. Error: {e}")
-        raise
+#     except Exception as e:
+#         logging.error(f"Failed to generate image for UID: {file_name}. Error: {e}")
+#         raise
 
 
-def create_image(prompt: str | None = None) -> None:
-    """Main function for generating images."""
-    if prompt is None:
-        prompt = create_prompt_on_openwebui(user_config["comfyui"]["prompt"])
+# def create_image(prompt: str | None = None) -> None:
+#     """Main function for generating images."""
+#     if prompt is None:
+#         prompt = create_prompt_on_openwebui(user_config["comfyui"]["prompt"])
 
-    if not prompt:
-        logging.error("No prompt generated.")
-        return
-    save_prompt(prompt)
+#     if not prompt:
+#         logging.error("No prompt generated.")
+#         return
+#     save_prompt(prompt)
 
-    use_flux = user_config["comfyui"].get("USE_FLUX", False)
-    only_flux = user_config["comfyui"].get("ONLY_FLUX", False)
+#     use_flux = user_config["comfyui"].get("USE_FLUX", False)
+#     only_flux = user_config["comfyui"].get("ONLY_FLUX", False)
 
-    selected_workflow = "SDXL"
-    if use_flux:
-        selected_workflow = "FLUX" if only_flux else random.choice(["FLUX", "SDXL"])
+#     selected_workflow = "SDXL"
+#     if use_flux:
+#         selected_workflow = "FLUX" if only_flux else random.choice(["FLUX", "SDXL"])
 
-    if selected_workflow == "FLUX":
-        generate_image(
-            file_name="image",
-            comfy_prompt=prompt,
-            workflow_path="./FLUX.json",
-            prompt_node="Positive Prompt T5",
-            seed_node="Seed",
-            seed_param="seed",
-            save_node="CivitAI Image Saver",
-            save_param="filename",
-            model_node="CivitAI Image Saver",
-            model_param="modelname",
-        )
-    else:
-        generate_image("image", prompt)
+#     if selected_workflow == "FLUX":
+#         # generate_image(
+#         #     file_name="image",
+#         #     comfy_prompt=prompt,
+#         #     workflow_path="./FLUX.json",
+#         #     prompt_node="Positive Prompt T5",
+#         #     seed_node="Seed",
+#         #     seed_param="seed",
+#         #     save_node="CivitAI Image Saver",
+#         #     save_param="filename",
+#         #     model_node="CivitAI Image Saver",
+#         #     model_param="modelname",
+#         # )
+#         print("flux")
+#     else:
+#         print("sdxl")
+#         # generate_image("image", prompt)
 
-    logging.info(f"{selected_workflow} generation started with prompt: {prompt}")
+#     logging.info(f"{selected_workflow} generation started with prompt: {prompt}")
 
 
-def get_prompt_from_png(path):
-    try:
-        with Image.open(path) as img:
-            try:
-                # Flux workflow
-                meta = json.loads(img.info["prompt"])['44']['inputs']['text']
-            except KeyError:
-                # SDXL workflow
-                meta = json.loads(img.info["prompt"])['6']['inputs']['text']
-            return meta or ""
-    except Exception as e:
-        print(f"Error reading metadata from {path}: {e}")
-        return ""
+# def get_prompt_from_png(path):
+#     try:
+#         with Image.open(path) as img:
+#             try:
+#                 # Flux workflow
+#                 meta = json.loads(img.info["prompt"])['44']['inputs']['text']
+#             except KeyError:
+#                 # SDXL workflow
+#                 meta = json.loads(img.info["prompt"])['6']['inputs']['text']
+#             return meta or ""
+#     except Exception as e:
+#         print(f"Error reading metadata from {path}: {e}")
+#         return ""
 
-user_config = load_config()
-output_folder = user_config["comfyui"]["output_dir"]
+# user_config = load_config()
+# output_folder = user_config["comfyui"]["output_dir"]
 

libs/comfyui.py

@@ -1,6 +1,7 @@
 import random
 import logging
 import os
+import json
 import requests
 from typing import Optional
 from comfy_api_simplified import ComfyApiWrapper, ComfyWorkflowWrapper
@@ -139,25 +140,35 @@ def create_image(prompt: str | None = None) -> None:
     """Main function for generating images."""
     if prompt is None:
         prompt = create_prompt_on_openwebui(user_config["comfyui"]["prompt"])
-    if prompt:
-        logging.info(f"Generated prompt: {prompt}")  # Log generated prompt
-        save_prompt(prompt)
-        if user_config["comfyui"]["FLUX"]:
-            generate_image(
-                file_name="image",
-                comfy_prompt=prompt,
-                workflow_path="./FLUX.json",
-                prompt_node="Positive Prompt T5",
-                seed_node="Seed",
-                seed_param="seed",
-                save_node="CivitAI Image Saver",
-                save_param="filename",
-                model_node="CivitAI Image Saver",
-                model_param="modelname",
-            )
-        else:
-            generate_image("image", prompt)
-        print(f"Image generation started with prompt: {prompt}")
-    else:
+
+    if not prompt:
         logging.error("No prompt generated.")
+        return
+    save_prompt(prompt)
+
+    use_flux = json.loads((user_config["comfyui"].get("FLUX", False)).lower())
+    only_flux = json.loads((user_config["comfyui"].get("ONLY_FLUX", False)).lower())
+
+    selected_workflow = "SDXL"
+    if use_flux:
+        selected_workflow = "FLUX" if only_flux else random.choice(["FLUX", "SDXL"])
+
+    if selected_workflow == "FLUX":
+        generate_image(
+            file_name="image",
+            comfy_prompt=prompt,
+            workflow_path="./FLUX.json",
+            prompt_node="Positive Prompt T5",
+            seed_node="Seed",
+            seed_param="seed",
+            save_node="CivitAI Image Saver",
+            save_param="filename",
+            model_node="CivitAI Image Saver",
+            model_param="modelname",
+        )
+    else:
+        generate_image("image", prompt)
+
+    logging.info(f"{selected_workflow} generation started with prompt: {prompt}")
+
     

libs/ollama.py

@@ -19,7 +19,7 @@ def create_prompt_on_openwebui(prompt: str) -> str:
     # Decide on whether to include a topic (e.g., 30% chance to include)
     topics = [t.strip() for t in user_config["comfyui"]["topics"].split(",") if t.strip()]
     topic_instruction = ""
-    if random.random() < 0.3 and topics:
+    if random.random() < 0.5 and topics:
         selected_topic = random.choice(topics)
         topic_instruction = f" Incorporate the theme of '{selected_topic}' into the new prompt."
 

user_config.cfg.sample

@@ -13,7 +13,7 @@ width = 1568
 height = 672
 topics = 
 
-USE_FLUX = False
+FLUX = False
 ONLY_FLUX = False
 
 [comfyui:flux]