Update app_gradio.py

app_gradio.py

@@ -175,7 +175,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) -> List[str]:
+def generate_output(image, 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)
@@ -200,6 +200,10 @@ def generate_output(image, prompt: str, num_seeds: int = 3, lambda_value: float
     
     return filtered_gifs
 
+def generate_output_from_sketchpad(image, prompt, num_seed, lambda_value, progress=gr.Progress(track_tqdm=True)):
+    results = generate_output(image['composite'], prompt, num_seed, lambda_value)
+    return results
+
 css=""" """
 
 def create_gradio_interface():
@@ -229,6 +233,13 @@ def create_gradio_interface():
                         interactive=True,
                         height=300  # Fixed height for consistency
                     )
+                    generate_btn = gr.Button(
+                        "Generate Animation",
+                        variant="primary",
+                        elem_classes="generate-btn",
+                        interactive=True,
+                    )
+                
                 with gr.Tab("Draw"):
                     draw_sketchpad = gr.Sketchpad(
                         value={
@@ -241,11 +252,13 @@ def create_gradio_interface():
                         layers=False,
                         height=300
                     )
+                    sketchpad_generate_btn = gr.Button(
+                        "Generate Animation",
+                        variant="primary",
+                        elem_classes="generate-btn",
+                        interactive=True,
+                    )
 
-                def load_drawing(im):
-                    return im["composite"]
-
-                draw_sketchpad.change(load_drawing, outputs=input_sketch, inputs=draw_sketchpad, show_progress="hidden")
                 
                
                 motion_prompt = gr.Textbox(
@@ -270,9 +283,9 @@ def create_gradio_interface():
                         label="Motion Strength"
                     )         
                 
+            
                 
-                
-        
+            with gr.Column():
                 gr.Examples(
                     examples=[
                         ['./static/examples/sketch1.png', 'The camel walks slowly'],
@@ -291,16 +304,9 @@ def create_gradio_interface():
                    inputs=[input_sketch, motion_prompt],
                    examples_per_page=4
                )
-            
                 
-            with gr.Column():
-                generate_btn = gr.Button(
-                    "Generate Animation",
-                    variant="primary",
-                    elem_classes="generate-btn",
-                    interactive=True,
-                )
-                output_gallery = gr.Gallery(
+                
+        output_gallery = gr.Gallery(
                     label="Results",
                     elem_classes="output-gallery",
                     columns=3,
@@ -310,7 +316,7 @@ def create_gradio_interface():
                     show_share_button=False,
                     object_fit="cover",
                     preview=False
-                )
+        )
             
            
         
@@ -325,6 +331,16 @@ def create_gradio_interface():
             ],
             outputs=output_gallery
         )
+        sketchpad_generate_btn.click(
+            fn=generate_output_from_sketchpad,
+            inputs=[
+                draw_sketchpad,
+                motion_prompt,
+                num_seeds,
+                lambda_
+            ],
+            outputs=output_gallery
+        )
     
     return demo