Update app.py

app.py

@@ -141,6 +141,7 @@ def infer(
     guidance_scale,
     num_inference_steps,
     expanded,
+    expanded_only,
     latent_file,  # Add latents file input
     progress=gr.Progress(track_tqdm=True),
 ):
@@ -199,7 +200,7 @@ def infer(
         print('-- filtered prompt 2 --')
         print(enhanced_prompt_2)
         if expanded_only:
-            return enhanced_prompt_2
+            return None, seed, enhanced_prompt
     else:
         enhanced_prompt = prompt
         enhanced_prompt_2 = prompt
@@ -426,6 +427,7 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
             guidance_scale,
             num_inference_steps,
             expanded,
+            expanded_only,
             latent_file,  # Add latent_file to the inputs
         ],
         outputs=[result, seed, expanded_prompt_output],