Update app.py

app.py

@@ -149,7 +149,7 @@ cliptextencode = CLIPTextEncode()
 unetloader = NODE_CLASS_MAPPINGS["ScaledFP8HybridUNetLoader"]()
 vaeloader = VAELoader()
 modelsamplingauraflow = NODE_CLASS_MAPPINGS["ModelSamplingAuraFlow"]()
-basicguider = NODE_CLASS_MAPPINGS["BasicGuider"]()
+cfgguider = NODE_CLASS_MAPPINGS["CFGGuider"]()
 betascheduler = NODE_CLASS_MAPPINGS["BetaSamplingScheduler"]()
 samplercustomadvanced = NODE_CLASS_MAPPINGS["SamplerCustomAdvanced"]()
 vaedecode = VAEDecode()
@@ -181,7 +181,7 @@ valid_models = [
 model_management.load_models_gpu(valid_models)
 
 @spaces.GPU
-def generate_image(prompt, width, height, steps, seed):
+def generate_image(prompt, negative_prompt, width, height, steps, cfg, seed):
     with torch.inference_mode():
         # Set random seed if provided
         if seed == -1:
@@ -198,6 +198,10 @@ def generate_image(prompt, width, height, steps, seed):
             text=prompt,
             clip=get_value_at_index(t5tokenizeroptions_82, 0),
         )
+        cliptextencode_75 = cliptextencode.encode(
+            text=negative_prompt,
+            clip=get_value_at_index(t5tokenizeroptions_82, 0),
+        )
 
 
         modelsamplingauraflow_85 = modelsamplingauraflow.patch(
@@ -205,9 +209,11 @@ def generate_image(prompt, width, height, steps, seed):
             model=get_value_at_index(unetloader_76, 0),
         )
 
-        basicguider_73 = basicguider.get_guider(
+        cfgguider_73 = cfgguider.get_guider(
+            cfg=cfg,
             model=get_value_at_index(unetloader_76, 0),
-            conditioning=get_value_at_index(cliptextencode_74, 0),
+            positive=get_value_at_index(cliptextencode_74, 0),
+            negative=get_value_at_index(cliptextencode_75, 0),
         )
 
         betascheduler_84 = betascheduler.get_sigmas(
@@ -219,7 +225,7 @@ def generate_image(prompt, width, height, steps, seed):
 
         samplercustomadvanced_67 = samplercustomadvanced.sample(
             noise=get_value_at_index(randomnoise_68, 0),
-            guider=get_value_at_index(basicguider_73, 0),
+            guider=get_value_at_index(cfgguider_73, 0),
             sampler=get_value_at_index(ksamplerselect_72, 0),
             sigmas=get_value_at_index(betascheduler_84, 0),
             latent_image=get_value_at_index(emptysd3latentimage_69, 0),
@@ -259,8 +265,15 @@ Space Author: [GitHub](https://github.com/gokayfem) | [X.com](https://x.com/goka
             prompt = gr.Textbox(
                 label="Prompt",
                 placeholder="Enter your prompt here...",
+                value="Overlaid at the center of the image is a title text that says \"CHROMA1-FLASH-HEUN\" in a large white 3D letters. This is a close-up photograph from a nature documentary capturing the right side of the face of a tiger. The photograph is centered on its highly detailed and speckled eye surrounded by intricately detailed fur. Amateur photography. unfiltered. natural lighting. anatomically correct. subtle shadows. perfect composition. highest quality. detailed. sharp focus",
                 lines=3
             )
+            negative_prompt = gr.Textbox(
+                label="Negative Prompt",
+                placeholder="Enter negative prompt here...",
+                value="low quality, ugly, unfinished, out of focus, deformed, disfigure, blurry, smudged, restricted palette, flat colors",
+                lines=2
+            )
             
             with gr.Row():
                 width = gr.Slider(
@@ -282,10 +295,17 @@ Space Author: [GitHub](https://github.com/gokayfem) | [X.com](https://x.com/goka
                 steps = gr.Slider(
                     minimum=1,
                     maximum=50,
-                    value=24,
+                    value=12,
                     step=1,
                     label="Steps"
                 )
+                cfg = gr.Slider(
+                    minimum=1,
+                    maximum=20,
+                    value=1,
+                    step=0.01,
+                    label="CFG Scale"
+                )
                 seed = gr.Number(
                     value=-1,
                     label="Seed (-1 for random)"
@@ -298,14 +318,14 @@ Space Author: [GitHub](https://github.com/gokayfem) | [X.com](https://x.com/goka
     
     generate_btn.click(
         fn=generate_image,
-        inputs=[prompt, width, height, steps, seed],
+        inputs=[prompt, negative_prompt, width, height, steps, cfg, seed],
         outputs=[output_image]
     )
     
     # Add examples section
     gr.Examples(
         examples=EXAMPLES,
-        inputs=[prompt, width, height, steps, seed],
+        inputs=[prompt, negative_prompt, width, height, steps, cfg, seed],
         outputs=[output_image],
         fn=generate_image,
         cache_examples=True,