Update app_gradio.py

app_gradio.py

@@ -155,7 +155,7 @@ def process_video(num_frames, num_seeds, generator, exp_dir, load_name, caption,
 
     return gifs
 
-def generate_output(image, prompt: str, num_seeds: int = 3, lambda_value: float = 0.5, progress=gr.Progress(track_tqdm=True)) -> List[str]:
+def generate_output(image, apply_filter, prompt: str, num_seeds: int = 3, lambda_value: float = 0.5, progress=gr.Progress(track_tqdm=True)) -> List[str]:
     """Main function to generate output GIFs"""
     exp_dir = "static/app_tmp"
     os.makedirs(exp_dir, exist_ok=True)
@@ -178,12 +178,17 @@ def generate_output(image, prompt: str, num_seeds: int = 3, lambda_value: float
     # Apply filtering (assuming filter function is imported)
     print("APPLYING FILTERS")
     filtered_gifs = filter(generated_gifs, temp_image_path)   
-    
+
+    if apply_filter == True:
+        print("APPLYING FILTER")
+    else:
+        print("NOT APPLYING FILTER")
+        
     return filtered_gifs
 
-def generate_output_from_sketchpad(image, prompt: str, num_seeds: int = 3, lambda_value: float = 0.5, progress=gr.Progress(track_tqdm=True)):
+def generate_output_from_sketchpad(image, apply_filter, prompt: str, num_seeds: int = 3, lambda_value: float = 0.5, progress=gr.Progress(track_tqdm=True)):
     image = image['composite']
-    results = generate_output(image, prompt, num_seeds, lambda_value)
+    results = generate_output(image, apply_filter, prompt, num_seeds, lambda_value)
     return results
 
 css=""" """
@@ -236,7 +241,11 @@ with gr.Blocks(css=css) as demo:
                         value=0.5, 
                         step=0.1, 
                         label="Motion Strength"
-                    )                     
+                    )          
+                    apply_filter = gr.Checkbox(
+                        value=True,
+                        visible=False
+                    )
 
                 with gr.Column():
                     gr.Examples(
@@ -304,6 +313,10 @@ with gr.Blocks(css=css) as demo:
                         step=0.1, 
                         label="Motion Strength"
                     )
+                    draw_apply_filter = gr.Checkbox(
+                        label="Appl Filters",
+                        value=False
+                    )
                     
                     sketchpad_generate_btn = gr.Button(
                         "Generate Animation",
@@ -332,6 +345,7 @@ with gr.Blocks(css=css) as demo:
             fn=generate_output,
             inputs=[
                 input_sketch,
+                apply_filter,
                 motion_prompt,
                 num_seeds,
                 lambda_
@@ -356,6 +370,7 @@ with gr.Blocks(css=css) as demo:
             fn=generate_output_from_sketchpad,
             inputs=[
                 draw_sketchpad,
+                draw_apply_filter,
                 draw_motion_prompt,
                 draw_num_seeds,
                 draw_lambda_,