Update app.py

app.py

@@ -198,6 +198,8 @@ def infer(
         print(enhanced_prompt)
         print('-- filtered prompt 2 --')
         print(enhanced_prompt_2)
+        if expanded_only:
+            return enhanced_prompt_2
     else:
         enhanced_prompt = prompt
         enhanced_prompt_2 = prompt
@@ -335,6 +337,14 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
                 value=True,
                 label="Use expanded prompt: ",
             )
+            expanded_only = gr.Radio(
+                show_label=True,
+                container=True,
+                interactive=True,
+                choices=options,
+                value=False,
+                label="Only get expanded prompt: ",
+            )
             run_button = gr.Button("Run", scale=0, variant="primary")
         result = gr.Image(label="Result", show_label=False)
         with gr.Accordion("Advanced Settings", open=True):