Update app_gradio.py

app_gradio.py

@@ -222,11 +222,12 @@ def create_gradio_interface():
 
             """
             )
+
+        with gr.Tab("Main"):
         
-        with gr.Row():
-            with gr.Column():        
+            with gr.Row():       
+                with gr.Column():
 
-                with gr.Tab("Main") as main_tab:
                     input_sketch = gr.Image(
                         type="pil",
                         label="Selected Sketch",
@@ -234,41 +235,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,
+                    motion_prompt = gr.Textbox(
+                        label="Prompt",
+                        placeholder="Describe the motion...",
+                        lines=3
                     )
                 
-                with gr.Tab("Draw"):
-                    draw_sketchpad = gr.Sketchpad(
-                        value={
-                            "background": "./static/examples/background.jpeg",
-                            "layers": None,
-                            "composite": None
-                        },
-                        type="pil",
-                        image_mode="RGB",
-                        layers=False,
-                        
-                    )
-                    sketchpad_generate_btn = gr.Button(
-                        "Generate Animation",
-                        variant="primary",
-                        elem_classes="generate-btn",
-                        interactive=True,
-                    )
-
                 
-               
-                motion_prompt = gr.Textbox(
-                    label="Prompt",
-                    placeholder="Describe the motion...",
-                    lines=3
-                )
-                
-                with gr.Row():
                     num_seeds = gr.Slider(
                         minimum=1, 
                         maximum=10, 
@@ -282,15 +255,10 @@ def create_gradio_interface():
                         value=0.5, 
                         step=0.1, 
                         label="Motion Strength"
-                    )         
-                
-            
-                
-            with gr.Column():
-                def load_examples(input_sketch, motion_prompt):
-                    return gr.update(open=True)
-                
-                gr.Examples(
+                    )    
+
+                with gr.Column():
+                    gr.Examples(
                     examples=[
                         ['./static/examples/sketch1.png', 'The camel walks slowly'],
                         ['./static/examples/sketch2.png', 'The wine in the wine glass sways from side to side'],
@@ -305,14 +273,54 @@ def create_gradio_interface():
                         ['./static/examples/sketch11.png', 'The jazz saxophonist performs on stage with a rhythmic sway, his upper body sways subtly to the rhythm of the music.'],
                         ['./static/examples/sketch12.png', 'The biker rides on the road']
                     ],
-                    fn=load_examples,
                     inputs=[input_sketch, motion_prompt],
-                    outputs=[main_tab],
-                    examples_per_page=4,
-                    cache_examples=False
-                    
+                    examples_per_page=4,   
                 )
                 
+        with gr.Tab("Draw"):
+            with gr.Row():
+                with gr.Column():
+                    draw_sketchpad = gr.Sketchpad(
+                        value={
+                            "background": "./static/examples/background.jpeg",
+                            "layers": None,
+                            "composite": None
+                        },
+                        type="pil",
+                        image_mode="RGB",
+                        layers=False,
+                        
+                    )
+                with gr.Column():
+                    draw_motion_prompt = gr.Textbox(
+                        label="Prompt",
+                        placeholder="Describe the motion...",
+                        lines=3
+                    )
+                
+                
+                    draw_num_seeds = gr.Slider(
+                        minimum=1, 
+                        maximum=10, 
+                        value=5, 
+                        step=1, 
+                        label="Seeds"
+                    )
+                    draw_lambda_ = gr.Slider(
+                        minimum=0, 
+                        maximum=1, 
+                        value=0.5, 
+                        step=0.1, 
+                        label="Motion Strength"
+                    )    
+
+                    sketchpad_generate_btn = gr.Button(
+                        "Generate Animation",
+                        variant="primary",
+                        elem_classes="generate-btn",
+                        interactive=True,
+                    )
+              
                 
         output_gallery = gr.Gallery(
                     label="Results",
@@ -339,13 +347,27 @@ def create_gradio_interface():
             ],
             outputs=output_gallery
         )
+        def reload_pad():
+            blank_pad ={
+                            "background": "./static/examples/background.jpeg",
+                            "layers": None,
+                            "composite": None
+                        }
+            return blank_pad
+        
+        draw_sketchpad.delete(
+            fn=reload_pad,
+            inputs = None,
+            outputs = [draw_sketchpad],
+            queue=False
+        )
         sketchpad_generate_btn.click(
             fn=generate_output_from_sketchpad,
             inputs=[
                 draw_sketchpad,
-                motion_prompt,
-                num_seeds,
-                lambda_
+                draw_motion_prompt,
+                draw_num_seeds,
+                draw_lambda_
             ],
             outputs=output_gallery
         )