lib.py

@@ -199,22 +199,15 @@ def generate_image(
 
         # Conditionally set model if node and param are provided
         if model_node and model_param:
-            if user_config["comfyui"].get("FLUX"):
-                available_model_list = user_config["comfyui"]["confyui:flux"]["models"].split(",")
-            else:
-                available_model_list = user_config["comfyui"]["models"].split(",")
-
             valid_models = list(
-                set(get_available_models()) & set(available_model_list)
+                set(get_available_models())
+                & set(user_config["comfyui"]["models"].split(","))
             )
-
             if not valid_models:
                 raise Exception("No valid models available.")
-
             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)
@@ -251,8 +244,8 @@ def create_image(prompt: str | None = None) -> None:
                 seed_param="seed",
                 save_node="CivitAI Image Saver",
                 save_param="filename",
-                model_node="CivitAI Image Saver",
-                model_param="modelname",
+                model_node=None,  # FLUX doesn't use model selection
+                model_param=None,
             )
         else:
             generate_image("image", prompt)

publish.sh

@@ -1,22 +0,0 @@
-#!/bin/bash
-
-# Set variables
-IMAGE_NAME="ai-frame-image-server"
-REGISTRY="kithub.k-world.me.uk"
-USERNAME="karl"
-TAG="latest"
-FULL_IMAGE="$REGISTRY/$USERNAME/$IMAGE_NAME:$TAG"
-
-# Build the image
-echo "🛠️  Building Docker image..."
-docker build -t $IMAGE_NAME .
-
-# Tag the image
-echo "🏷️  Tagging image as $FULL_IMAGE"
-docker tag $IMAGE_NAME $FULL_IMAGE
-
-# Push the image
-echo "📤  Pushing $FULL_IMAGE to $REGISTRY..."
-docker push $FULL_IMAGE
-
-echo "✅ Done!"

user_config.cfg.sample

@@ -13,9 +13,6 @@ width = 1568
 height = 672
 FLUX = False
 
-[comfyui:flux]
-models = flux1-dev-Q4_0.gguf,flux1-schnell-Q4_0.gguf
-
 [openwebui]
 base_url = https://openwebui
 api_key = sk-