Update app_gradio.py

app_gradio.py

@@ -220,14 +220,24 @@ def create_gradio_interface():
         
         with gr.Row():
             with gr.Column():        
-                
-                input_sketch = gr.Image(
-                    type="pil",
-                    label="Selected Sketch",
-                    scale=1,
-                    interactive=True,
-                    height=300  # Fixed height for consistency
-                )
+                with gr.Tab("Main"):
+                    input_sketch = gr.Image(
+                        type="pil",
+                        label="Selected Sketch",
+                        scale=1,
+                        interactive=True,
+                        height=300  # Fixed height for consistency
+                    )
+                with gr.Tab("Draw"):
+                    draw_sketchpad = gr.Sketchpad(
+                        type="pil",
+                        height=300
+                    )
+
+                def predict(im):
+                    return im["composite"]
+
+                draw_sketchpad.change(load_drawing, outputs=input_sketch, inputs=draw_sketchpad, show_progress="hidden")
                
                 motion_prompt = gr.Textbox(
                     label="Prompt",