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, prompt: str, num_seeds: int = 3, lambda_value: float = 0.5, apply_filer: bool = True, 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)
@@ -174,15 +174,19 @@ def generate_output(image, prompt: str, num_seeds: int = 3, lambda_value: float
         caption=prompt,
         lambda_=1 - lambda_value
     )
+
+    if apply_filter:
+        # Apply filtering (assuming filter function is imported)
+        filtered_gifs = filter(generated_gifs, temp_image_path)
+        results = filtered_gifs
+    else:
+        results = generated_gifs
     
-    # Apply filtering (assuming filter function is imported)
-    filtered_gifs = filter(generated_gifs, temp_image_path)
-    
-    return filtered_gifs
+    return results
 
-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, prompt: str, num_seeds: int = 3, lambda_value: float = 0.5, apply_filer: bool = True, progress=gr.Progress(track_tqdm=True)):
     image = image['composite']
-    results = generate_output(image, prompt, num_seeds, lambda_value)
+    results = generate_output(image, prompt, num_seeds, lambda_value, apply_filter)
     return results
 
 css=""" """
@@ -303,6 +307,11 @@ with gr.Blocks(css=css) as demo:
                         step=0.1, 
                         label="Motion Strength"
                     )
+                    apply_filter = gr.Checkbox(
+                        label="Apply GIFs Filters",
+                        info="If checked, bad/off results will be filtered",
+                        value=True
+                    )
                     
                     sketchpad_generate_btn = gr.Button(
                         "Generate Animation",
@@ -357,7 +366,8 @@ with gr.Blocks(css=css) as demo:
                 draw_sketchpad,
                 draw_motion_prompt,
                 draw_num_seeds,
-                draw_lambda_
+                draw_lambda_,
+                apply_filter
             ],
             outputs=output_gallery
     )