limit max seed value to max

Author: mythz

MyApp.ServiceInterface/AppExtensions.cs

@@ -350,4 +350,17 @@ public static AgentInfo ToAgentInfo(this ComfyAgent from)
         }
         return to;
     }
+
+    public static long GetMaxSeedValue(this WorkflowInfo info)
+    {
+        return info.Inputs?.Where(x => x.Name is "seed" or "noise_seed")
+            .Select(x => (long)(x.Max ?? long.MaxValue))
+            .DefaultIfEmpty(long.MaxValue)
+            .Min() ?? long.MaxValue;
+    }
+
+    public static long GetNextSeedValue(this WorkflowInfo info)
+    {
+        return Random.Shared.NextInt64(0, GetMaxSeedValue(info));
+    }
 }

MyApp.ServiceInterface/ComfyServices.cs

@@ -399,16 +399,17 @@ public async Task<object> Post(RequeueGeneration request)
         // If we're retrying an existing generation we need to regenerate the seeds
         if (gen.Args?.Count > 0 && (gen.Args.ContainsKey("seed") || gen.Args.ContainsKey("noise_seed")))
         {
+            var workflowVersion = GetWorkflowVersion(Db, gen.WorkflowId, gen.VersionId);
+
             if (gen.Args.TryGetValue("seed", out var seed))
             {
-                gen.Args["seed"] = Random.Shared.NextInt64(0, long.MaxValue);
+                gen.Args["seed"] = workflowVersion.Info.GetNextSeedValue();
             }
             if (gen.Args.TryGetValue("noise_seed", out var noiseSeed))
             {
-                gen.Args["noise_seed"] = Random.Shared.NextInt64(0, long.MaxValue);
+                gen.Args["noise_seed"] = workflowVersion.Info.GetNextSeedValue();
             }
 
-            var workflowVersion = GetWorkflowVersion(Db, gen.WorkflowId, gen.VersionId);
             var (apiPrompt, newWorkflow, _) = await comfyConverter.CreateApiPromptAsync(workflowVersion, gen.Args, agent:null, gen.Id);
 
             gen.Workflow = newWorkflow;