Update app.py

app.py

@@ -45,12 +45,12 @@ def get_seed(seed):
         return random.randint(0, MAX_SEED)
 
 @spaces.GPU(duration=30)
-def generate(input=DEFAULT_INPUT, negative_input=DEFAULT_NEGATIVE_INPUT, model=DEFAULT_MODEL, height=DEFAULT_HEIGHT, width=DEFAULT_WIDTH, steps=1, guidance=0, number=1, seed=None):
+def generate(input=DEFAULT_INPUT, filter_input="", negative_input=DEFAULT_NEGATIVE_INPUT, model=DEFAULT_MODEL, height=DEFAULT_HEIGHT, width=DEFAULT_WIDTH, steps=1, guidance=0, number=1, seed=None):
 
     repo = None
     seed = get_seed(seed)
 
-    print(input, negative_input, model, height, width, steps, guidance, number, seed)
+    print(input, filter_input, negative_input, model, height, width, steps, guidance, number, seed)
 
     if model == "Anime":   
         vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
@@ -71,7 +71,7 @@ def generate(input=DEFAULT_INPUT, negative_input=DEFAULT_NEGATIVE_INPUT, model=D
     
     parameters  = {
         "prompt": input,
-        "negative_prompt": negative_input,
+        "negative_prompt": filter_input + negative_input,
         "height": height,
         "width": width,
         "num_inference_steps": steps,
@@ -100,6 +100,7 @@ with gr.Blocks(css=css) as main:
         
     with gr.Column():
         input = gr.Textbox(lines=1, value=DEFAULT_INPUT, label="Input")
+        filter_input = gr.Textbox(lines=1, value="", label="Input Filter")
         negative_input = gr.Textbox(lines=1, value=DEFAULT_NEGATIVE_INPUT, label="Input Negative")
         model = gr.Dropdown(label="Models", choices=["Default", "Anime"], value="Default")
         height = gr.Slider(minimum=1, maximum=2160, step=1, value=DEFAULT_HEIGHT, label="Height")
@@ -114,7 +115,7 @@ with gr.Blocks(css=css) as main:
     with gr.Column():
         images = gr.Gallery(columns=1, label="Image")
             
-    submit.click(generate, inputs=[input, negative_input, model, height, width, steps, guidance, number, seed], outputs=[images], queue=False)
+    submit.click(generate, inputs=[input, filter_input, negative_input, model, height, width, steps, guidance, number, seed], outputs=[images], queue=False)
     maintain.click(cloud, inputs=[], outputs=[], queue=False)
 
 main.launch(show_api=True)