Update app_gradio.py

app_gradio.py

@@ -179,15 +179,15 @@ def generate_output(image, apply_filter, prompt: str, num_seeds: int = 3, lambda
         print("APPLYING FILTER")
         # Apply filtering (assuming filter function is imported)
         filtered_gifs = filter(generated_gifs, temp_image_path)   
-        return filtered_gifs
+        return filtered_gifs, filtered_gifs
     else:
         print("NOT APPLYING FILTER")
-        return generated_gifs
+        return generated_gifs, generated_gifs
 
 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, apply_filter, prompt, num_seeds, lambda_value)
-    return results
+    results, results_to_download= generate_output(image, apply_filter, prompt, num_seeds, lambda_value)
+    return results, results_to_download
 
 css=""" """
 
@@ -349,7 +349,7 @@ with gr.Blocks(css=css) as demo:
                 num_seeds,
                 lambda_
             ],
-            outputs=output_gallery
+            outputs=[output_gallery, download_gifs]
     )
     def reload_pad():
          blank_pad ={
@@ -374,7 +374,7 @@ with gr.Blocks(css=css) as demo:
                 draw_num_seeds,
                 draw_lambda_,
             ],
-            outputs=output_gallery
+            outputs=[output_gallery, download_gifs]
     )