randomly select dev/schnell when using flux

lib.py

@@ -199,15 +199,22 @@ 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(get_available_models()) & set(available_model_list)
-                & 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)
@@ -244,8 +251,8 @@ def create_image(prompt: str | None = None) -> None:
                 seed_param="seed",
                 save_node="CivitAI Image Saver",
                 save_param="filename",
-                model_node=None,  # FLUX doesn't use model selection
+                model_node="CivitAI Image Saver",
-                model_param=None,
+                model_param="modelname",
             )
         else:
             generate_image("image", prompt)

publish.sh

@@ -0,0 +1,22 @@
+#!/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,6 +13,9 @@ 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-