fix workflows

Author: mythz

AiServer.ServiceInterface/Comfy/ComfyClient.cs

@@ -180,21 +180,17 @@ public async Task<ComfyWorkflowResponse> PromptGenerationAsync(ComfyWorkflowRequ
         // Handle any file uploads required before processing the workflow
         await HandleAssetUploadsAsync(comfyRequest, promptId, token);
 
-        // Read template from file for Text to Image
-        var workflowArgs = comfyRequest.ToObjectDictionary();
-        foreach (var key in workflowArgs.Keys.ToList())
-        {
-            workflowArgs[key.ToCamelCase()] = workflowArgs[key];
-        }
-
         var mediaModel = AppData.Instance.MediaModels.FirstOrDefault(x =>
             x.ApiModels?.TryGetValue("ComfyUI", out var comfyModel) == true && comfyModel == comfyRequest.Model);
-        if (mediaModel?.WorkflowVars != null)
+        var workflowArgs = mediaModel?.WorkflowVars != null
+            ? new(mediaModel.WorkflowVars)
+            : new Dictionary<string, object>();
+
+        // Read template from file for Text to Image
+        var requestArgs = comfyRequest.ToObjectDictionary();
+        foreach (var entry in requestArgs)
         {
-            foreach (var entry in mediaModel.WorkflowVars)
-            {
-                workflowArgs[entry.Key] = entry.Value;
-            }
+            workflowArgs[entry.Key.ToCamelCase()] = entry.Value;
         }
 
         var workflowJson = await PopulateWorkflowAsync(workflowArgs, templatePath, token);

AiServer/workflows/text-to-image/flux1.json

@@ -251,7 +251,7 @@
         "Node name for S&R": "CheckpointLoaderSimple"
       },
       "widgets_values": [
-        "flux1-schnell.safetensors"
+        "{{model}}"
       ]
     },
     {

AiServer/workflows/text-to-image/hidream.json

@@ -339,7 +339,7 @@
         "Node name for S&R": "CheckpointLoaderSimple"
       },
       "widgets_values": [
-        "{{Model}}"
+        "{{model}}"
       ]
     },
     {

AiServer/workflows/text-to-image/sd35-large.json

@@ -443,7 +443,7 @@
         "Node name for S&R": "CheckpointLoaderSimple"
       },
       "widgets_values": [
-        "sd3.5_large.safetensors"
+        "{{model}}"
       ]
     },
     {

AiServer/workflows/text-to-image/sd35-turbo.json

@@ -419,7 +419,7 @@
         "Node name for S&R": "CheckpointLoaderSimple"
       },
       "widgets_values": [
-        "sd3.5_large_turbo.safetensors"
+        "{{model}}"
       ]
     },
     {

AiServer/workflows/text-to-image/sdxl-lightning.json

@@ -281,7 +281,7 @@
         "Node name for S&R": "CheckpointLoaderSimple"
       },
       "widgets_values": [
-        "sdxl_lightning_4step.safetensors"
+        "{{model}}"
       ]
     }
   ],

AiServer/workflows/text-to-image/sdxl.json

@@ -213,7 +213,7 @@
         "Node name for S&R": "CheckpointLoaderSimple"
       },
       "widgets_values": [
-        "{{Model}}"
+        "{{model}}"
       ]
     },
     {

AiServer/workflows/text_to_image.json

@@ -213,7 +213,7 @@
         "Node name for S&R": "CheckpointLoaderSimple"
       },
       "widgets_values": [
-        "{{Model}}"
+        "{{model}}"
       ]
     },
     {